Skip to content

Commit

Permalink
bump mcore commit in Dockerfile (#9766)
Browse files Browse the repository at this point in the history
Signed-off-by: ashors1 <[email protected]>
  • Loading branch information
ashors1 authored Jul 17, 2024
1 parent 8c04749 commit f4fa399
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 02871b4df8c69fac687ab6676c4246e936ce92d0 && \
git checkout c7a1f82d761577e6ca0338d3521eac82f2aa0904 && \
pip install .

# Performance optimizations for distributed optimizer: https://github.com/NVIDIA/apex/pull/1771
Expand Down

0 comments on commit f4fa399

Please sign in to comment.