fix: treat SDXL hires fix like cascade (1024x longest dim)

Haidra-Org · Jul 8, 2024 · c0bab91 · c0bab91
1 parent 77f5b4d
commit c0bab91
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/hordelib/horde.py b/hordelib/horde.py
@@ -868,8 +868,10 @@ def _final_pipeline_adjustments(self, payload, pipeline_data) -> tuple[dict, lis
                     original_width, original_height = (1024, 1024)
 
             new_width, new_height = (None, None)
-
-            if model_details and model_details.get("baseline") == "stable_cascade":
+            baseline = None
+            if model_details:
+                baseline = model_details.get("baseline")
+            if baseline and (baseline == "stable_cascade" or baseline == "stable_diffusion_xl"):
                 new_width, new_height = ImageUtils.get_first_pass_image_resolution_max(
                     original_width,
                     original_height,

diff --git a/images_expected/sdxl_text_to_image_hires_fix.png b/images_expected/sdxl_text_to_image_hires_fix.png