Updating linux requirements

bmaltais · Mar 2, 2024 · 32db52c · 32db52c
1 parent 30670f1
commit 32db52c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/requirements_linux.txt b/requirements_linux.txt
@@ -1,4 +1,5 @@
-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.21 bitsandbytes==0.41.1
+torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+xformers==0.0.23.post1+cu118 --index-url https://download.pytorch.org/whl/cu118
+bitsandbytes==0.41.1
 tensorboard==2.15.2 tensorflow==2.15.0.post1
 -r requirements.txt