Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix ppl builder with pipeline_name

Browse Source
This commit is contained in:
suluyana
2025-01-12 23:18:18 +08:00
parent 07d1d589cc
commit 9ebce138b4
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/pipelines/builder.py
View File

@@ -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