[diffusion] fix: fix missing backend argument in pipelines_core initialization (#17343)
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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__}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user