fix(torch-extras): Update xformers for CUDA 12.6 support

Compatibility fixed in v0.0.28 in the following commit: facebookresearch/xformers@926f410
coreweave · Oct 12, 2024 · c1bd46f · c1bd46f
1 parent 9362f03
commit c1bd46f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch-extras/Dockerfile b/torch-extras/Dockerfile
@@ -3,7 +3,7 @@
 ARG BASE_IMAGE
 ARG DEEPSPEED_VERSION="0.14.4"
 ARG APEX_COMMIT="23c1f86520e22b505e8fdfcf6298273dff2d93d8"
-ARG XFORMERS_VERSION="0.0.27.post2"
+ARG XFORMERS_VERSION="0.0.28.post1"
 
 FROM alpine/git:2.36.3 as apex-downloader
 WORKDIR /git