fix pipeline args

2026-04-14 21:58:17 +00:00 · 2026-04-14 10:11:09 +08:00
parent 224060c2a0
commit 9654ec3c65
6 changed files with 102 additions and 102 deletions
--- a/diffsynth/pipelines/flux2_image.py
+++ b/diffsynth/pipelines/flux2_image.py
@@ -83,7 +83,7 @@ class Flux2ImagePipeline(BasePipeline):
        input_image: Image.Image = None,
        denoising_strength: float = 1.0,
        # Edit
-        edit_image: Union[Image.Image, List[Image.Image]] = None,
+        edit_image: List[Image.Image] = None,
        edit_image_auto_resize: bool = True,
        # Shape
        height: int = 1024,