From 5c160e97c4054e87f57550547a5b0806f76f324f Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Sat, 25 Apr 2026 21:26:56 +0800 Subject: [PATCH] update swift 4.1.3 --- docker/Dockerfile.ubuntu | 2 +- docker/build_image.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 80ae8e0e..7baa82d5 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -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; \ diff --git a/docker/build_image.py b/docker/build_image.py index 502625eb..74af2c5e 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -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'