feat: switch to the usual img2img

Haidra-Org · Mar 18, 2024 · 9c396b2 · 9c396b2
1 parent 363e604
commit 9c396b2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_horde_inference_cascade.py b/tests/test_horde_inference_cascade.py
@@ -120,11 +120,14 @@ def test_cascade_image_to_image(
             "control_type": None,
             "image_is_control": False,
             "return_control_map": False,
-            "prompt": "A predatory dinosaur sitting on its prey",
+            "prompt": (
+                "a medieval fantasy swashbuckler with a big floppy hat walking towards "
+                "a camera while there's an explosion in the background"
+            ),
             "ddim_steps": 20,
             "n_iter": 1,
             "model": stable_cascade_base_model_name,
-            "source_image": Image.open("images/test_inpaint_original.png"),
+            "source_image": Image.open("images/test_db0.jpg"),
             "source_processing": "img2img",
         }
         pil_image = hordelib_instance.basic_inference_single_image(data).image