From dd729efa359d8c20cde746b6673bf93afa098125 Mon Sep 17 00:00:00 2001 From: "xingjun.wxj" Date: Mon, 24 Mar 2025 17:21:28 +0800 Subject: [PATCH] set vllm==0.7.2 and xformers==0.0.27 --- docker/Dockerfile.ubuntu | 1 + docker/build_image.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index aede1070..883cfde5 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -56,6 +56,7 @@ ARG CUR_TIME={cur_time} RUN echo $CUR_TIME RUN sh /tmp/install.sh {version_args} && \ + pip install --no-cache-dir xformers==0.0.27 && \ curl -fsSL https://ollama.com/install.sh | sh && \ pip install --no-cache-dir -U funasr scikit-learn && \ pip install --no-cache-dir -U qwen_vl_utils pyav librosa timm transformers accelerate peft trl safetensors && \ diff --git a/docker/build_image.py b/docker/build_image.py index a2cfa83b..36e068a0 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -31,7 +31,7 @@ class Builder: if not args.cuda_version: args.cuda_version = '12.1.0' if not args.vllm_version: - args.vllm_version = '0.7.3' + args.vllm_version = '0.7.2' if not args.lmdeploy_version: args.lmdeploy_version = '0.6.2' if not args.autogptq_version: @@ -273,7 +273,7 @@ class LLMImageBuilder(Builder): if not args.cuda_version: args.cuda_version = '12.4.0' if not args.vllm_version: - args.vllm_version = '0.7.3' + args.vllm_version = '0.7.2' if not args.lmdeploy_version: args.lmdeploy_version = '0.7.0.post2' if not args.autogptq_version: