Skip to content

Commit d67eb9f

Browse files
committed
Use dummy image for editing warmup
1 parent bcf1fbd commit d67eb9f

1 file changed

Lines changed: 16 additions & 9 deletions

File tree

src/exo/worker/engines/image/generate.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -33,15 +33,22 @@ def parse_size(size_str: str | None) -> tuple[int, int]:
3333

3434
def warmup_image_generator(model: ImageGenerator) -> Image.Image | None:
3535
"""Warmup the image generator with a small image."""
36-
for result in model.generate(
37-
prompt="Warmup",
38-
height=256,
39-
width=256,
40-
quality="low",
41-
seed=2,
42-
):
43-
if not isinstance(result, tuple):
44-
return result
36+
with tempfile.TemporaryDirectory() as tmpdir:
37+
# Create a small dummy image for warmup (needed for edit models)
38+
dummy_image = Image.new("RGB", (256, 256), color=(128, 128, 128))
39+
dummy_path = Path(tmpdir) / "warmup.png"
40+
dummy_image.save(dummy_path)
41+
42+
for result in model.generate(
43+
prompt="Warmup",
44+
height=256,
45+
width=256,
46+
quality="low",
47+
seed=2,
48+
image_path=dummy_path,
49+
):
50+
if not isinstance(result, tuple):
51+
return result
4552
return None
4653

4754

0 commit comments

Comments
 (0)