Docker: use xformers 0.0.17

tdhd · May 1, 2023 · cac5e12 · cac5e12
1 parent 328d260
commit cac5e12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -21,7 +21,7 @@ RUN . ./venv/bin/activate && \
 
 # Upgrade to Torch 2.0
 RUN . ./venv/bin/activate && \
-    python3 -m pip install --use-pep517 --no-deps -U triton==2.0.0 torch>=2.0.0+cu121 xformers \
+    python3 -m pip install --use-pep517 --no-deps -U triton==2.0.0 torch>=2.0.0+cu121 xformers==0.0.17 \
 	                       --extra-index-url https://download.pytorch.org/whl/cu121
 
 # Fix missing libnvinfer7