Updated

requirements.txt

@@ -1,6 +1,9 @@
---extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.6.0+cu121
-torchaudio==2.6.0+cu121
+--index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://pypi.org/simple
+
+torch==2.5.1+cu121
+torchaudio==2.5.1+cu121
+
 transformers>=4.43
 accelerate>=0.33
 diffusers>=0.31