change dockerfile 1.14

Azure · Feb 4, 2024 · 1b3525b · 1b3525b
1 parent 1dac840
commit 1b3525b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dockerfile/torch1.14-cuda11.8.dockerfile b/dockerfile/torch1.14-cuda11.8.dockerfile
@@ -50,7 +50,7 @@ RUN cd third_party/msccl && \
 # cache TE build to save time in CI
 ENV MAX_JOBS=1
 RUN python3 -m pip install --upgrade pip && \
-    python3 -m pip install flash-attn==1.0.9 git+https://github.com/NVIDIA/TransformerEngine.git@v0.11
+    python3 -m pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable
 
 ADD . .
 RUN python3 -m pip install . && \

diff --git a/pyproject.toml b/pyproject.toml
@@ -38,7 +38,7 @@ classifiers=[
 ]
 dependencies = [
     "torch",
-    "transformer-engine@git+git+https://github.com/NVIDIA/TransformerEngine.git@stable",
+    "transformer-engine@git+https://github.com/NVIDIA/TransformerEngine.git@stable",
     "colorlog>=6.7.0",
     "deepspeed==0.13.1",
     "mpi4py",