From 12bc3ccf53b5893bee9292bfd9926f52d7f54a5a Mon Sep 17 00:00:00 2001 From: dimapihtar Date: Wed, 31 Jan 2024 14:49:31 -0800 Subject: [PATCH] update mcore Signed-off-by: dimapihtar --- Dockerfile | 2 +- Jenkinsfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 094c8fff408e..e39a96206975 100644 --- a/Dockerfile +++ b/Dockerfile @@ -66,7 +66,7 @@ WORKDIR /workspace/ # We leave it here in case we need to work off of a specific commit in main RUN git clone https://github.com/NVIDIA/Megatron-LM.git && \ cd Megatron-LM && \ - git checkout 27cbe46714a50c43ed290f1b1472db8d2780c55c && \ + git checkout 05342e7c95c26fd9f213f6f7ca85ed3373ca48a9 && \ pip install . # Apex bugfix for PyTorch 23.11 container: https://github.com/NVIDIA/apex/pull/1760 diff --git a/Jenkinsfile b/Jenkinsfile index 762751d454d4..d8a0d4db9197 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -85,7 +85,7 @@ pipeline { steps { sh 'git clone https://github.com/NVIDIA/Megatron-LM.git && \ cd Megatron-LM && \ - git checkout bed60a881f4b238b1c14b6c6a64997cc636e77b6 && \ + git checkout 05342e7c95c26fd9f213f6f7ca85ed3373ca48a9 && \ pip install .' } }