apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): keep ControlNet metadata after lookup

This commit is contained in:
Sean Sube 2023-04-15 17:31:02 -05:00
parent 8d47b71183
commit 17e7b6aff2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/server/params.py
View File

@ -46,7 +46,12 @@ def pipeline_from_request(
# diffusion model
model = get_not_empty(request.args, "model", get_config_value("model"))
model_path = get_model_path(server, model)
control = get_from_list(request.args, "control", [m.name for m in get_network_models()])
control = None
control_name = request.args.get("control")
for network in get_network_models():
if network.name == control_name:
control = network
# pipeline stuff
pipeline = get_from_list(