From 9c396b2c389d5995d856b0aaa5eb12f94eb0c052 Mon Sep 17 00:00:00 2001 From: db0 Date: Mon, 18 Mar 2024 18:15:26 +0100 Subject: [PATCH] feat: switch to the usual img2img --- tests/test_horde_inference_cascade.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/test_horde_inference_cascade.py b/tests/test_horde_inference_cascade.py index ec857fb0..a69dec91 100644 --- 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