From de3399e8de2b60cd660f288404d612a90bbb81bf Mon Sep 17 00:00:00 2001 From: hjh0119 Date: Wed, 14 Jan 2026 10:35:49 +0800 Subject: [PATCH] pin optimum --- docker/install.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docker/install.sh b/docker/install.sh index 661dcc81..04a970f0 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -7,6 +7,7 @@ vllm_version=${4:-0.6.0} lmdeploy_version=${5:-0.6.1} autogptq_version=${6:-0.7.1} flashattn_version=${7:-2.7.1.post4} +optimum_version=${8:-2.0.0} pip uninstall -y torch torchvision torchaudio @@ -25,6 +26,8 @@ pip install --no-cache-dir flash_attn==$flashattn_version pip install --no-cache-dir triton auto-gptq==$autogptq_version -U && pip cache purge +pip install --no-cache-dir optimum==$optimum_version + if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0" ]]; then # vllm_version is >= 0.6.0 pip install --no-cache-dir vllm==$vllm_version && pip cache purge