45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
import pytest
|
|
|
|
from mflux.models.common.config import ModelConfig
|
|
from mflux.models.qwen.variants.edit import QwenImageEdit
|
|
from tests.image_generation.helpers.image_generation_edit_test_helper import ImageGeneratorEditTestHelper
|
|
|
|
|
|
class TestImageGeneratorQwenImageEdit:
|
|
@pytest.mark.slow
|
|
def test_image_generation_qwen_edit(self):
|
|
ImageGeneratorEditTestHelper.assert_matches_reference_image(
|
|
reference_image_path="reference_qwen_edit.png",
|
|
output_image_path="output_qwen_edit.png",
|
|
model_class=QwenImageEdit,
|
|
model_config=ModelConfig.qwen_image_edit(),
|
|
steps=20,
|
|
seed=4869845,
|
|
height=384,
|
|
width=640,
|
|
guidance=2.5,
|
|
quantize=6, # We should probably use at least 8-bit, but it doesn't run on 32GB machines
|
|
prompt="Make the hand fistbump the camera instead of showing a flat palm",
|
|
image_path="reference_upscaled.png",
|
|
mismatch_threshold=0.25,
|
|
low_memory=True,
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.high_memory_requirement
|
|
def test_image_generation_qwen_edit_multiple_images(self):
|
|
ImageGeneratorEditTestHelper.assert_matches_reference_image(
|
|
reference_image_path="reference_qwen_edit_multiple_images.png",
|
|
output_image_path="output_qwen_edit_multiple_images.png",
|
|
model_class=QwenImageEdit,
|
|
model_config=ModelConfig.qwen_image_edit(),
|
|
steps=20,
|
|
seed=4869845,
|
|
height=384,
|
|
width=640,
|
|
guidance=2.5,
|
|
quantize=8, # We should probably use at least 8-bit, but it doesn't run on 32GB machines
|
|
prompt="Make the hand fistbump the camera instead of showing a flat palm, and the man should wear this shirt. Maintain the original pose, body position, and overall stance.",
|
|
image_paths=["reference_upscaled.png", "shirt.jpg"],
|
|
)
|