Yan Bai
commited on
Commit
·
72f5bfa
1
Parent(s):
054e141
build
Browse files- Dockerfile +4 -4
Dockerfile
CHANGED
|
@@ -4,13 +4,13 @@ FROM whatcanyousee/verl:ngc-cu124-vllm0.8.5-sglang0.4.6-mcore0.12.0-te2.3
|
|
| 4 |
RUN pip install --no-cache-dir \
|
| 5 |
fastapi \
|
| 6 |
uvicorn[standard] \
|
| 7 |
-
mbridge \
|
| 8 |
-
termcolor
|
| 9 |
-
|
| 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 && \
|
|
|
|
| 4 |
RUN pip install --no-cache-dir \
|
| 5 |
fastapi \
|
| 6 |
uvicorn[standard] \
|
| 7 |
+
mbridge==0.13.1 \
|
| 8 |
+
termcolor
|
| 9 |
+
|
| 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 && \
|