From 9ebce138b41ca28361a48c8550ab52b6fca967b5 Mon Sep 17 00:00:00 2001 From: suluyana Date: Sun, 12 Jan 2025 23:18:18 +0800 Subject: [PATCH] fix ppl builder with pipeline_name --- modelscope/pipelines/builder.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modelscope/pipelines/builder.py b/modelscope/pipelines/builder.py index cbf0a8e3..596d6d22 100644 --- a/modelscope/pipelines/builder.py +++ b/modelscope/pipelines/builder.py @@ -173,6 +173,8 @@ def pipeline(task: str = None, pipeline_name, default_model_repo = get_default_pipeline_info(task) model = normalize_model_input(default_model_repo, model_revision) pipeline_props = {'type': pipeline_name} + else: + pipeline_props = {'type': pipeline_name} pipeline_props['model'] = model pipeline_props['device'] = device