From 19558cbb154c28a12ad6c35385fc241802a0dff2 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 12 Jan 2026 19:23:17 +0530 Subject: [PATCH] fix --- src/diffusers/pipelines/wan/pipeline_wan_animate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/wan/pipeline_wan_animate.py b/src/diffusers/pipelines/wan/pipeline_wan_animate.py index 9340609cb9..5806032c01 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_animate.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_animate.py @@ -785,7 +785,7 @@ class WanAnimatePipeline(DiffusionPipeline, WanLoraLoaderMixin): output_type: str | None = "np", return_dict: bool = True, attention_kwargs: dict[str, Any] | None = None, - callback_on_step_end: Callable[[int, int, None], PipelineCallback, MultiPipelineCallbacks] | None = None, + callback_on_step_end: Callable[[int, int, None], PipelineCallback | MultiPipelineCallbacks] | None = None, callback_on_step_end_tensor_inputs: list[str] = ["latents"], max_sequence_length: int = 512, ):