Update requirements.txt

requirements.txt

@@ -1,6 +1,9 @@
-# Core ML
-torch>=2.1.0
-torchaudio>=2.1.0
+# Force pip to use the CUDA 12.1 package index to match the Hugging Face Space environment
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+# Core ML (Pinning versions is highly recommended for Unsloth compatibility)
+torch==2.4.0+cu121
+torchaudio==2.4.0+cu121
 transformers==4.56.2
 accelerate>=0.27.0
 bitsandbytes>=0.43.0