Their release tagging uses x.x.x, but the brqnches use x.x.

coreweave · Aug 21, 2024 · ff2c28e · ff2c28e
1 parent bd0bb04
commit ff2c28e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch-extras/Dockerfile b/torch-extras/Dockerfile
@@ -5,7 +5,7 @@ ARG DEEPSPEED_VERSION="0.14.4"
 ARG FLASH_ATTN_VERSION="2.6.3"
 ARG APEX_COMMIT="23c1f86520e22b505e8fdfcf6298273dff2d93d8"
 ARG XFORMERS_VERSION="0.0.27.post2"
-ARG TE_VERSION="1.9.0"
+ARG TE_VERSION="1.9"
 
 FROM alpine/git:2.36.3 as flash-attn-downloader
 WORKDIR /git