hires_fix pipeline support for layerdiffuse:

Haidra-Org · Jul 7, 2024 · db602a9 · db602a9
1 parent bf16d3e
commit db602a9
Show file tree

Hide file tree

Showing 5 changed files with 555 additions and 246 deletions.
diff --git a/hordelib/horde.py b/hordelib/horde.py
@@ -1041,6 +1041,8 @@ def _final_pipeline_adjustments(self, payload, pipeline_data) -> tuple[dict, lis
                     self.generator.reconnect_input(pipeline_data, "layer_diffuse_apply.model", "model_loader")
                     self.generator.reconnect_input(pipeline_data, "output_image.images", "layer_diffuse_decode_rgba")
                     self.generator.reconnect_input(pipeline_data, "layer_diffuse_decode_rgba.images", "vae_decode")
+                    if payload.get("hires_fix") is True:
+                        self.generator.reconnect_input(pipeline_data, "upscale_sampler.model", "layer_diffuse_apply")
                     if model_details.get("baseline") == "stable diffusion 1":
                         pipeline_params["layer_diffuse_apply.config"] = "SD15, Attention Injection, attn_sharing"
                         pipeline_params["layer_diffuse_decode_rgba.sd_version"] = "SD15"