stricter ty

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-19 01:04:32 +02:00 · 2026-09-08 10:22:42 +02:00
parent 23b800a99b
commit 2ef1a0c87b
4 changed files with 6 additions and 4 deletions
@@ -964,7 +964,7 @@ class StableDiffusionXLSoftFillPipeline(
        aesthetic_score: float = 6.0,
        negative_aesthetic_score: float = 2.5,
        clip_skip: Optional[int] = None,
-        callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
+        callback_on_step_end: Optional[Callable[..., Dict]] = None, # invoked as (self, step_index, timestep, callback_kwargs) -> dict
        callback_on_step_end_tensor_inputs: List[str] = ["latents"],
        **kwargs,
    ):