bug fix (#81)

Author: tenderness-git

diffsynth_engine/tools/flux_outpainting_tool.py

@@ -1,5 +1,5 @@
 from diffsynth_engine import fetch_model, FluxControlNet, ControlNetParams, FluxImagePipeline
-from typing import List, Tuple, Optional
+from typing import List, Tuple, Optional, Callable
 from PIL import Image
 import torch
 
@@ -34,6 +34,7 @@ def __call__(
         inpainting_scale: float = 0.9,
         seed: int = 42,
         num_inference_steps: int = 20,
+        progress_callback: Optional[Callable] = None,
     ):
         assert scaling_factor >= 1.0, "scale must be >= 1.0"
         width, height = image.width, image.height
@@ -57,4 +58,5 @@ def __call__(
                 mask=mask,
                 scale=inpainting_scale,
             ),
+            progress_callback=progress_callback,
         )

diffsynth_engine/tools/flux_replace_tool.py

@@ -1,5 +1,5 @@
 from diffsynth_engine import ControlNetParams, FluxControlNet, FluxImagePipeline, FluxRedux, fetch_model
-from typing import List, Tuple, Optional
+from typing import List, Tuple, Optional, Callable
 from PIL import Image
 import torch
 
@@ -45,6 +45,7 @@ def __call__(
         ref_scale: float = 1.0,
         seed: int = 42,
         num_inference_steps: int = 20,
+        progress_callback: Optional[Callable] = None,
     ):
         assert image.size == mask.size
         self.pipe.redux.set_scale(ref_scale)
@@ -68,5 +69,6 @@ def __call__(
                 mask=mask,
                 scale=inpainting_scale,
             ),
+            progress_callback=progress_callback,
         )
         return result.resize((width, height))