[diffusion] fix: fix missing backend argument in pipelines_core initialization (#17343)

This commit is contained in:
Prozac614
2026-01-26 15:47:10 +08:00
committed by GitHub
parent d4adff31aa
commit 12f794e516
4 changed files with 10 additions and 7 deletions

View File

@@ -180,7 +180,7 @@ async def available_models():
if not server_args:
raise HTTPException(status_code=500, detail="Server args not initialized")
model_info = get_model_info(server_args.model_path)
model_info = get_model_info(server_args.model_path, backend=server_args.backend)
card_kwargs = {
"id": server_args.model_path,
@@ -222,7 +222,7 @@ async def retrieve_model(model: str):
},
)
model_info = get_model_info(server_args.model_path)
model_info = get_model_info(server_args.model_path, backend=server_args.backend)
card_kwargs = {
"id": model,

View File

@@ -66,7 +66,7 @@ def build_pipeline(
)
else:
logger.info("No pipeline_class_name specified, using model_index.json")
model_info = get_model_info(model_path)
model_info = get_model_info(model_path, backend=server_args.backend)
pipeline_cls = model_info.pipeline_cls
logger.info(f"Using pipeline from model_index.json: {pipeline_cls.__name__}")