diff --git a/docker/Dockerfile.ascend b/docker/Dockerfile.ascend index c9f4e80e..442a08fe 100644 --- a/docker/Dockerfile.ascend +++ b/docker/Dockerfile.ascend @@ -28,8 +28,8 @@ RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple && \ # ---------- Install vllm + vllm-ascend ---------- RUN source /usr/local/Ascend/ascend-toolkit/set_env.sh && \ if [ -f /usr/local/Ascend/nnal/atb/set_env.sh ]; then source /usr/local/Ascend/nnal/atb/set_env.sh; fi && \ - git clone --depth 1 --branch v0.14.0 https://github.com/vllm-project/vllm && \ - git clone --depth 1 --branch v0.14.0rc1 https://github.com/vllm-project/vllm-ascend.git + git clone --depth 1 --branch v0.18.0 https://github.com/vllm-project/vllm && \ + git clone --depth 1 --branch v0.18.0 https://github.com/vllm-project/vllm-ascend.git RUN ARCH=$(uname -m) && \ source /usr/local/Ascend/ascend-toolkit/set_env.sh && \ diff --git a/modelscope/pipelines/builder.py b/modelscope/pipelines/builder.py index b0095b73..5704a4d9 100644 --- a/modelscope/pipelines/builder.py +++ b/modelscope/pipelines/builder.py @@ -245,7 +245,10 @@ def pipeline(task: str = None, if preprocessor is not None: cfg.preprocessor = preprocessor - return build_pipeline(cfg, task_name=task) + return build_pipeline( + cfg, + task_name=task, + default_args={'trust_remote_code': trust_remote_code}) def add_default_pipeline_info(task: str,