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

fix lint

Browse Source
This commit is contained in:
suluyan
2025-01-10 15:32:53 +08:00
parent 0cb31263ab
commit 07d1d589cc
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/pipelines/builder.py
View File

@@ -120,7 +120,8 @@ def pipeline(task: str = None,
model, str) else read_config(
model[0], revision=model_revision)
if cfg:
pipeline_name = cfg.safe_get('pipeline', {}).get('type', None)
pipeline_name = cfg.safe_get('pipeline',
{}).get('type', None)
if pipeline_name is None:
prefer_llm_pipeline = kwargs.get('external_engine_for_llm')
@@ -150,7 +151,8 @@ def pipeline(task: str = None,
ignore_file_pattern=ignore_file_pattern)
register_plugins_repo(cfg.safe_get('plugins'))
register_modelhub_repo(model, cfg.get('allow_remote', False))
register_modelhub_repo(model,
cfg.get('allow_remote', False))
if pipeline_name:
pipeline_props = {'type': pipeline_name}