Update requirement runpod

bmaltais · Jun 26, 2023 · 85e3edf · 85e3edf
1 parent 43ccfe1
commit 85e3edf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/requirements_runpod.txt b/requirements_runpod.txt
@@ -1,5 +1,4 @@
-torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
 xformers==0.0.20 bitsandbytes==0.39.1
-accelerate==0.19.0 tensorboard==2.12.1 tensorflow==2.12.0
-tensorrt
+accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0
 -r requirements.txt