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: