Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-05-18 13:15:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

update swift 4.1.3

Browse Source
This commit is contained in:
Jintao Huang
2026-04-25 21:26:56 +08:00
parent 07b0321e78
commit 5c160e97c4
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile.ubuntu
View File

@@ -66,7 +66,7 @@ RUN bash /tmp/install.sh {version_args} && \
RUN if [ "$IMAGE_TYPE" = "gpu" ]; then \
pip install --no-cache-dir math_verify "gradio<5.33" "deepspeed<0.19" ray mcore-bridge -U && \
pip install --no-cache-dir liger_kernel wandb swanlab nvitop pre-commit "transformers<5.6" "trl<0.30" "peft<0.19" huggingface-hub -U && \
pip install --no-cache-dir liger_kernel wandb swanlab nvitop pre-commit "transformers<5.7" "trl<0.30" "peft<0.20" huggingface-hub -U && \
pip install --no-cache-dir --no-build-isolation transformer_engine[pytorch]; \
pip install -U flash-linear-attention --no-build-isolation; \
pip install -U git+https://github.com/Dao-AILab/causal-conv1d --no-build-isolation; \

2
docker/build_image.py
View File

@@ -398,7 +398,7 @@ class LatestGPUImageBuilder(StableGPUImageBuilder):
args.torchaudio_version = '2.10.0'
args.torchvision_version = '0.25.0'
if not args.vllm_version:
args.vllm_version = '0.19.0'
args.vllm_version = '0.19.1'
if not args.base_image:
# A mirrored image of nvidia/cuda:12.4.0-devel-ubuntu22.04
args.base_image = 'nvidia/cuda:12.9.1-cudnn-devel-ubuntu22.04'