Fix

app.py

@@ -43,6 +43,8 @@ def run(
     guidance_scale: float = 8.0,
 ) -> tuple[PIL.Image.Image | None, str]:
     generator = torch.Generator(device=device).manual_seed(seed)
+    if image is not None:
+        image = image.resize((512, 512))
     if mode == "t2i":
         pipe.set_text_to_image_mode()
         sample = pipe(prompt=prompt, num_inference_steps=num_steps, guidance_scale=guidance_scale, generator=generator)