monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change img2img test to be non-square

Browse Source
This commit is contained in:
filipstrand 2024-10-29 19:58:39 +01:00
parent 4e2a10b1b9
commit 8d93f3e24f
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/helpers/image_generation_test_helper.py
View File

@ -63,7 +63,7 @@ class ImageGeneratorTestHelper:
os.remove(output_image_path) os.remove(output_image_path)
@staticmethod @staticmethod
def resolve_path(path) -> Path: def resolve_path(path) -> Path | None:
if path is None: if path is None:
return None return None
return Path(__file__).parent.parent / "resources" / path return Path(__file__).parent.parent / "resources" / path

BIN
tests/resources/reference_dev_image_to_image_init.png
View File

Binary file not shown.
Side by Side

Before

Width:  |  Height:  |  Size: 658 KiB

BIN
tests/resources/reference_dev_image_to_image_result.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 630 KiB

After

Width:  |  Height:  |  Size: 351 KiB

10
tests/test_generate_image.py
View File

@ -46,13 +46,13 @@ class TestImageGenerator:
def test_image_generation_dev_image_to_image(self): def test_image_generation_dev_image_to_image(self):
ImageGeneratorTestHelper.assert_matches_reference_image( ImageGeneratorTestHelper.assert_matches_reference_image(
reference_image_path="reference_dev_image_to_image_result.png", reference_image_path="reference_dev_image_to_image_result.png",
init_image_path="reference_dev_image_to_image_init.png", init_image_path="reference_dev_lora.png",
init_image_strength=0.126, # fill 1/8 steps init_image_strength=0.4,
output_image_path=TestImageGenerator.OUTPUT_IMAGE_FILENAME, output_image_path=TestImageGenerator.OUTPUT_IMAGE_FILENAME,
model_config=ModelConfig.FLUX1_DEV, model_config=ModelConfig.FLUX1_DEV,
steps=8, steps=8,
seed=42, seed=44,
height=768, height=341,
width=768, width=768,
prompt="astronauts in a jungle", prompt="Luxury food photograph of a burger",
) )