Downgrade torch

jhj0517 · Oct 4, 2024 · b1b9396 · b1b9396
1 parent f3f351e
commit b1b9396
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,10 +2,11 @@
 # If you're using it, update url to your CUDA version (CUDA 12.1 is minimum requirement):
 # For CUDA 12.1, use : https://download.pytorch.org/whl/cu121
 # For CUDA 12.4, use : https://download.pytorch.org/whl/cu124
---extra-index-url https://download.pytorch.org/whl/cu124
+--extra-index-url https://download.pytorch.org/whl/cu121
 
 
-torch
+torch==2.3.1
+torchaudio==2.3.1
 git+https://github.com/jhj0517/jhj0517-whisper.git
 faster-whisper==1.0.3
 transformers==4.42.3