Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

update swift docker

Browse Source
This commit is contained in:
Jintao Huang
2025-09-16 00:07:44 +08:00
parent 0bcefbe08a
commit 4ebfecb6cb
4 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.ubuntu
View File

@@ -79,8 +79,8 @@ RUN if [ "$INSTALL_MS_DEPS" = "True" ]; then \
pip install --no-cache-dir huggingface-hub transformers peft -U; \
fi; \
if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \
pip install "sglang[all]<0.4.10" "math_verify==0.5.2" "gradio<5.33" -U && \
pip install liger_kernel nvitop pre-commit "transformers<4.55" huggingface-hub -U && \
pip install "sglang[all]<0.5" "math_verify==0.5.2" "gradio<5.33" -U && \
pip install liger_kernel wandb swanlab nvitop pre-commit "transformers<4.57" "trl<0.21" huggingface-hub -U && \
SITE_PACKAGES=$(python -c "import site; print(site.getsitepackages()[0])") && echo $SITE_PACKAGES && \
CUDNN_PATH=$SITE_PACKAGES/nvidia/cudnn CPLUS_INCLUDE_PATH=$SITE_PACKAGES/nvidia/cudnn/include \
pip install --no-build-isolation transformer_engine[pytorch]; \

4
docker/build_image.py
View File

@@ -354,9 +354,9 @@ class SwiftImageBuilder(LLMImageBuilder):
args.torchaudio_version = '2.7.1'
args.torchvision_version = '0.22.1'
if not args.vllm_version:
args.vllm_version = '0.10.0'
args.vllm_version = '0.10.1.1'
if not args.lmdeploy_version:
args.lmdeploy_version = '0.9.2'
args.lmdeploy_version = '0.9.2.post1'
if not args.flashattn_version:
args.flashattn_version = '2.7.4.post1'
return super().init_args(args)