From 94527f0053f0ef84ccb3c7d84ea1fdfcb71950b0 Mon Sep 17 00:00:00 2001 From: tastelikefeet Date: Fri, 13 Mar 2026 11:03:33 +0800 Subject: [PATCH] fix --- docker/build_image.py | 2 +- docker/install.sh | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/docker/build_image.py b/docker/build_image.py index accb9636..6c978972 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -34,7 +34,7 @@ class Builder: if not args.vllm_version: args.vllm_version = '0.15.1' if not args.lmdeploy_version: - args.lmdeploy_version = '0.10.1' + args.lmdeploy_version = '0.11.0' if not args.autogptq_version: args.autogptq_version = '0.7.1' if not args.flashattn_version: diff --git a/docker/install.sh b/docker/install.sh index 371ef927..14066872 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -12,8 +12,6 @@ pip uninstall -y torch torchvision torchaudio pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version -pip install --no-cache-dir -U autoawq lmdeploy==$lmdeploy_version - pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version pip install --no-cache-dir tiktoken transformers_stream_generator bitsandbytes deepspeed torchmetrics decord optimum openai-whisper @@ -23,7 +21,7 @@ pip install --no-cache-dir tiktoken transformers_stream_generator bitsandbytes d # cd /tmp && git clone https://github.com/Dao-AILab/flash-attention.git && cd flash-attention && python setup.py install && cd / && rm -fr /tmp/flash-attention && pip cache purge; MAX_JOBS=16 pip install --no-cache-dir flash_attn==$flashattn_version --no-build-isolation -pip install --no-cache-dir triton auto-gptq==$autogptq_version -U && pip cache purge +pip install --no-cache-dir triton -U && pip cache purge if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0" ]]; then # vllm_version is >= 0.6.0