Spaces:

ISEEKYAN
/

megatron_memory_estimator

Running

Yan Bai commited on 2 days ago

Commit

85d63c1

1 Parent(s): c74073d

0.13

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -8,7 +8,7 @@ RUN pip install --no-cache-dir \
     termcolor \
     ipdb
 # 添加 Megatron-LM core_v0.12.2
-RUN git clone -b core_v0.12.2 --depth 1 https://github.com/NVIDIA/Megatron-LM.git /opt/Megatron-LM
 RUN git clone -b estimator_mcore013 --depth 1 https://github.com/ISEEKYAN/mbridge.git /opt/mbridge

     termcolor \
     ipdb
 # 添加 Megatron-LM core_v0.12.2
+RUN git clone -b core_v0.13.0rc4 --depth 1 https://github.com/NVIDIA/Megatron-LM.git /opt/Megatron-LM
 RUN git clone -b estimator_mcore013 --depth 1 https://github.com/ISEEKYAN/mbridge.git /opt/mbridge