fix

Signed-off-by: Oliver Koenig <okoenig@nvidia.com>
NVIDIA · Nov 28, 2024 · 81d9959 · 81d9959
1 parent 2142e23
commit 81d9959
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/import-test.yml b/.github/workflows/import-test.yml
@@ -32,7 +32,7 @@ jobs:
         
         # Preserve wheel location
         DIST_FILE=$(find ./dist -name "*.whl" | head -n 1)
-        echo "::set-output name=DIST_FILE::${DIST_FILE}"
+        echo "DIST_FILE=${DIST_FILE}" | tee -a "$GITHUB_OUTPUT"
     
     - name: Install NeMo + test dependencies
       run: |

diff --git a/Dockerfile.ci b/Dockerfile.ci
@@ -68,7 +68,7 @@ pip install --no-cache-dir --no-build-isolation --extra-index-url https://pypi.n
 ".[all]"
 EOF
 
-ARG MCORE_TAG=452d52083081398e09fb5cdd753434aa6f1b2d1f
+ARG MCORE_TAG=c1728c12f1f1cdbb786e52f1ffe512295d76bef3
 RUN <<"EOF" bash -ex
 # Megatron-LM installation
 git clone https://github.com/NVIDIA/Megatron-LM.git

diff --git a/requirements/requirements_nlp.txt b/requirements/requirements_nlp.txt
@@ -10,7 +10,7 @@ ijson
 jieba
 markdown2
 matplotlib>=3.3.2
-megatron_core @ git+https://github.com/NVIDIA/Megatron-LM.git@452d52083081398e09fb5cdd753434aa6f1b2d1f
+megatron_core @ git+https://github.com/NVIDIA/Megatron-LM.git@c1728c12f1f1cdbb786e52f1ffe512295d76bef3
 nltk>=3.6.5
 numpy<2  # tensorstore has an implicit compiled dependency on numpy<2
 opencc