Hugging Face's logo

Spaces:
ISEEKYAN
/
megatron_memory_estimator
Running

App Files Community
Yan Bai commited on
Commit
054e141
·
1 Parent(s): 802abe8

build

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ RUN pip install --no-cache-dir \
10
  # 添加 Megatron-LM core_v0.12.2
11
  RUN git clone -b core_v0.13.0rc4 --depth 1 https://github.com/NVIDIA/Megatron-LM.git /opt/Megatron-LM
12
 
13
- # RUN git clone -b estimator_mcore013 --depth 1 https://github.com/ISEEKYAN/mbridge.git /opt/mbridge
14
 
15
  # 创建与 HF Spaces 运行时一致的 uid=1000 用户，避免 getpwuid() 错误
16
  RUN groupadd -g 1000 user && \
 
10
  # 添加 Megatron-LM core_v0.12.2
11
  RUN git clone -b core_v0.13.0rc4 --depth 1 https://github.com/NVIDIA/Megatron-LM.git /opt/Megatron-LM
12
 
13
+ RUN git clone -b main --depth 1 https://github.com/ISEEKYAN/mbridge.git /opt/mbridge
14
 
15
  # 创建与 HF Spaces 运行时一致的 uid=1000 用户，避免 getpwuid() 错误
16
  RUN groupadd -g 1000 user && \