22 lines
797 B
Python
22 lines
797 B
Python
import pytest
|
|
|
|
from mflux.models.common.config import ModelConfig
|
|
from tests.image_generation.helpers.image_generation_kontext_test_helper import ImageGeneratorKontextTestHelper
|
|
|
|
|
|
class TestImageGeneratorKontext:
|
|
@pytest.mark.slow
|
|
def test_image_generation_kontext(self):
|
|
ImageGeneratorKontextTestHelper.assert_matches_reference_image(
|
|
reference_image_path="reference_dev_kontext.png",
|
|
output_image_path="output_dev_kontext.png",
|
|
model_config=ModelConfig.dev_kontext(),
|
|
steps=20,
|
|
seed=4869845,
|
|
height=384,
|
|
width=640,
|
|
guidance=2.5,
|
|
prompt="Make the hand fistbump the camera instead of showing a flat palm",
|
|
kontext_image_path="reference_upscaled.png",
|
|
)
|