diff --git a/main.py b/main.py
index 02ae944..d412677 100644
--- a/main.py
+++ b/main.py
@@ -16,17 +16,6 @@ os.chdir(PROJECT_ROOT)
 # Add project root to path
 sys.path.insert(0, str(PROJECT_ROOT))
 
-# PyTorch 2.5+ compatibility patch for audiocraft
-# The _pytree API changed - add backwards compatibility
-try:
-    import torch.utils._pytree as _pytree
-    if not hasattr(_pytree, 'register_pytree_node') and hasattr(_pytree, '_register_pytree_node'):
-        _pytree.register_pytree_node = _pytree._register_pytree_node
-    elif not hasattr(_pytree, '_register_pytree_node') and hasattr(_pytree, 'register_pytree_node'):
-        _pytree._register_pytree_node = _pytree.register_pytree_node
-except Exception:
-    pass  # Ignore if patch fails
-
 from config.settings import get_settings
 from src.core.gpu_manager import GPUMemoryManager
 from src.core.model_registry import ModelRegistry
diff --git a/requirements.txt b/requirements.txt
index 8bc3e48..1130f5b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
-# Core ML
-torch>=2.1.0
-torchaudio>=2.1.0
+# Core ML - Pin to AudioCraft-compatible versions (2.4+ breaks pytree API)
+torch>=2.1.0,<2.4.0
+torchaudio>=2.1.0,<2.4.0
 audiocraft>=1.3.0
 xformers>=0.0.22