feat(torch-extras): Update Apex to a1df804

coreweave · Nov 26, 2024 · 56f06ed · 56f06ed
1 parent e6dac89
commit 56f06ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch-extras/Dockerfile b/torch-extras/Dockerfile
@@ -2,7 +2,7 @@
 
 ARG BASE_IMAGE
 ARG DEEPSPEED_VERSION="0.14.4"
-ARG APEX_COMMIT="23c1f86520e22b505e8fdfcf6298273dff2d93d8"
+ARG APEX_COMMIT="a1df80457ba67d60cbdb0d3ddfb08a2702c821a8"
 ARG DEEPSPEED_KERNELS_COMMIT="e77acc40b104696d4e73229b787d1ef29a9685b1"
 ARG DEEPSPEED_KERNELS_CUDA_ARCH_LIST="80;86;89;90"
 ARG XFORMERS_VERSION="0.0.28.post1"