21 lines
1.1 KiB
Python
21 lines
1.1 KiB
Python
from mflux import ModelConfig
|
|
from tests.image_generation.helpers.image_generation_in_context_test_helper import ImageGeneratorInContextTestHelper
|
|
|
|
|
|
class TestImageGeneratorInContext:
|
|
OUTPUT_IMAGE_FILENAME = "output.png"
|
|
|
|
def test_in_context_lora_identity(self):
|
|
ImageGeneratorInContextTestHelper.assert_matches_reference_image(
|
|
reference_image_path="ic_lora_reference_in_context_identity.png",
|
|
output_image_path=TestImageGeneratorInContext.OUTPUT_IMAGE_FILENAME,
|
|
model_config=ModelConfig.dev(),
|
|
steps=25,
|
|
seed=42,
|
|
height=320,
|
|
width=320,
|
|
prompt="In this set of two images, a bold modern typeface with the brand name 'DEMA' is introduced and is shown on a company merchandise product photo; [IMAGE1] a simplistic black logo featuring a modern typeface with the brand name 'DEMA' on a bright light green/yellowish background; [IMAGE2] the design is printed on a green/yellowish hoodie as a company merchandise product photo with a plain white background.",
|
|
image_path="ic_lora_reference_logo.png",
|
|
lora_style="identity",
|
|
)
|