Skip to content

Commit

Permalink
small fix to remove duplicate megatron-lm installation (#7864)
Browse files Browse the repository at this point in the history
Signed-off-by: David Mosallanezhad <[email protected]>
  • Loading branch information
Davood-M committed Nov 7, 2023
1 parent 86b198f commit b1bd2db
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,6 @@ RUN apt-get update && \
rm -rf /var/lib/apt/lists/*

WORKDIR /workspace/
# install megatron core, this can be removed once 0.3 pip package is released
RUN git clone https://github.com/NVIDIA/Megatron-LM.git && \
cd Megatron-LM && \
git checkout ab0336a5c8eab77aa74ae604ba1e73decbf6d560 && \
pip install -e .

# Install megatron core, this can be removed once 0.3 pip package is released
# 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 && \
Expand Down

0 comments on commit b1bd2db

Please sign in to comment.