fixed double

hordelib/horde.py

@@ -840,7 +840,7 @@ def _final_pipeline_adjustments(self, payload, pipeline_data) -> tuple[dict, lis
                 node_index = image_index + 1
                 pipeline_data[f"sc_image_loader_{node_index}"] = {
                     "inputs": {"image": extra_image, "upload": "image"},
-                    "class_type": "LoadImage",
+                    "class_type": "HordeImageLoader",
                 }
                 pipeline_data[f"clip_vision_encode_{node_index}"] = {
                     "inputs": {

tests/test_horde_inference_cascade.py

@@ -234,3 +234,93 @@ def test_cascade_image_remix_double(
         #     f"images_expected/{img_filename}",
         #     pil_image,
         # )
+
+    def test_cascade_image_remix_double_weak(
+        self,
+        stable_cascade_base_model_name: str,
+        hordelib_instance: HordeLib,
+    ):
+        data = {
+            "sampler_name": "k_euler",
+            "cfg_scale": 4,
+            "denoising_strength": 1,
+            "seed": 1312,
+            "height": 1024,
+            "width": 1024,
+            "karras": False,
+            "tiling": False,
+            "hires_fix": False,
+            "clip_skip": 1,
+            "control_type": None,
+            "image_is_control": False,
+            "return_control_map": False,
+            "prompt": "A herd of goats grazing",
+            "ddim_steps": 20,
+            "n_iter": 1,
+            "model": stable_cascade_base_model_name,
+            "source_image": Image.open("images/test_mountains.png"),
+            "source_processing": "remix",
+            "extra_source_images": [
+                {
+                    "image": Image.open("images/test_sunset.png"),
+                    "strength": 0.05,
+                },
+            ],
+        }
+        pil_image = hordelib_instance.basic_inference_single_image(data).image
+        assert pil_image is not None
+        assert isinstance(pil_image, Image.Image)
+
+        img_filename = "stable_cascade_image_remix_double_weak.png"
+        pil_image.save(f"images/{img_filename}", quality=100)
+
+        # assert check_single_inference_image_similarity(
+        #     f"images_expected/{img_filename}",
+        #     pil_image,
+        # )
+
+    def test_cascade_image_remix_triple(
+        self,
+        stable_cascade_base_model_name: str,
+        hordelib_instance: HordeLib,
+    ):
+        data = {
+            "sampler_name": "k_euler",
+            "cfg_scale": 4,
+            "denoising_strength": 1,
+            "seed": 1312,
+            "height": 1024,
+            "width": 1024,
+            "karras": False,
+            "tiling": False,
+            "hires_fix": False,
+            "clip_skip": 1,
+            "control_type": None,
+            "image_is_control": False,
+            "return_control_map": False,
+            "prompt": "Baking Sun",
+            "ddim_steps": 20,
+            "n_iter": 1,
+            "model": stable_cascade_base_model_name,
+            "source_image": Image.open("images/test_mountains.png"),
+            "source_processing": "remix",
+            "extra_source_images": [
+                {
+                    "image": Image.open("images/test_sunset.png"),
+                },
+                {
+                    "image": Image.open("images/test_db0.jpg"),
+                },
+            ],
+        }
+        pil_image = hordelib_instance.basic_inference_single_image(data).image
+        assert pil_image is not None
+        assert isinstance(pil_image, Image.Image)
+
+        img_filename = "stable_cascade_image_remix_triple.png"
+        pil_image.save(f"images/{img_filename}", quality=100)
+
+        # assert check_single_inference_image_similarity(
+        #     f"images_expected/{img_filename}",
+        #     pil_image,
+        # )