Pin torch and diffuser versions in requirements

riffusion · Nov 16, 2023 · df7703f · df7703f
1 parent a861b48
commit df7703f
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@ accelerate
 argh
 dacite
 demucs
-diffusers>=0.9.0
+diffusers==0.9.0
 flask
 flask_cors
 numpy
@@ -14,7 +14,8 @@ scipy
 soundfile
 sox
 streamlit>=1.18.0
-torch
-torchaudio
-torchvision
-transformers
+transformers==4.35.2
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.0.1+cu118
+torchaudio==2.0.2+cu118
+torchvision==0.15.2+cu118