diff --git a/docker/install.sh b/docker/install.sh index ee947f22..c2dfcd82 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -28,9 +28,6 @@ pip install --no-cache-dir triton auto-gptq==$autogptq_version -U && pip cache p 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 vllm==$vllm_version -else - # vllm_version is < 0.6.0 - pip install vllm==$vllm_version --no-dependencies fi # pip uninstall -y torch-scatter && TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9;9.0" pip install --no-cache-dir -U torch-scatter