no trans

mosaicml · Jan 13, 2024 · 408aa09 · 408aa09
1 parent 6b0b0ef
commit 408aa09
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -305,7 +305,6 @@ RUN if [ -n "$CUDA_VERSION" ] ; then \
         cd .. && \
         pip install --no-dependencies "git+ssh://[email protected]/mosaicml/megablocks-private.git@main#egg=megablocks[all]" && \
         pip install --no-dependencies git+https://github.com/tgale96/[email protected] && \
-        pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable && \
         pip uninstall -y torch && \
         pip uninstall -y pytorch-triton && \
         pip uninstall -y torchvision && \