fix(api): keep ControlNet metadata after lookup

ssube · Apr 15, 2023 · 17e7b6a · 17e7b6a
1 parent 8d47b71
commit 17e7b6a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/api/onnx_web/server/params.py b/api/onnx_web/server/params.py
@@ -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(