Merge branch 'gg_dckr' into 'main'

Update groupedgemm version in test dockerfile See merge request ADLR/megatron-lm!1517
llm-jp · May 30, 2024 · 7f616b0 · 7f616b0
2 parents 7b36131 + 0f2ce07
commit 7f616b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.test b/Dockerfile.test
@@ -8,4 +8,4 @@ RUN sed -i -e 's/^APT/# APT/' -e 's/^DPkg/# DPkg/' \
 
 RUN apt-get update && apt-get install -y --no-install-recommends
 
-RUN pip3 install --no-cache-dir einops flask-restful nltk pytest pytest-cov pytest_mock sentencepiece wrapt git+https://github.com/fanshiqing/[email protected].1
+RUN pip3 install --no-cache-dir einops flask-restful nltk pytest pytest-cov pytest_mock sentencepiece wrapt git+https://github.com/fanshiqing/[email protected].2