working sdxl tests

Haidra-Org · May 20, 2024 · 3858214 · 3858214
1 parent df16b58
commit 3858214
Show file tree

Hide file tree

Showing 5 changed files with 563 additions and 2,522 deletions.
diff --git a/hordelib/horde.py b/hordelib/horde.py
@@ -962,6 +962,12 @@ def _final_pipeline_adjustments(self, payload, pipeline_data) -> tuple[dict, lis
                 pipeline_params["qr_code_split.protocol"] = "None"
             if not pipeline_params.get("function_layer_prompt.text"):
                 pipeline_params["function_layer_prompt.text"] = payload["prompt"]
+            if SharedModelManager.manager.compvis:
+                model_details = SharedModelManager.manager.compvis.get_model_reference_info(payload["model_name"])
+                if model_details and model_details.get("baseline") == "stable diffusion 1":
+                    pipeline_params["controlnet_qr_model_loader.control_net_name"] = (
+                        "control_v1p_sd15_qrcode_monster_v2.safetensors"
+                    )
 
         return pipeline_params, faults