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