Yan Bai
commited on
Commit
·
d67c73d
1
Parent(s):
ae496b6
0.13
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
|
@@ -9,7 +9,8 @@ RUN pip install --no-cache-dir \
|
|
| 9 |
ipdb
|
| 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 |
|
| 14 |
# 创建与 HF Spaces 运行时一致的 uid=1000 用户，避免 getpwuid() 错误
|
| 15 |
RUN groupadd -g 1000 user && \
|
|
@@ -17,7 +18,7 @@ RUN groupadd -g 1000 user && \
|
|
| 17 |
|
| 18 |
# 复制代码至工作目录
|
| 19 |
WORKDIR $HOME/app
|
| 20 |
-
COPY --chown=user /opt/mbridge/memory_estimator
|
| 21 |
|
| 22 |
# HF Spaces 默认通过 $PORT 注入端口
|
| 23 |
ENV PYTHONPATH=/opt/Megatron-LM:$PYTHONPATH
|
|
|
|
| 9 |
ipdb
|
| 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 && \
|
|
|
|
| 18 |
|
| 19 |
# 复制代码至工作目录
|
| 20 |
WORKDIR $HOME/app
|
| 21 |
+
COPY --chown=user /opt/mbridge/memory_estimator $HOME/app
|
| 22 |
|
| 23 |
# HF Spaces 默认通过 $PORT 注入端口
|
| 24 |
ENV PYTHONPATH=/opt/Megatron-LM:$PYTHONPATH
|