From c13b2645892bd5b922335158294992fd5de00390 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 12 Jan 2026 13:43:11 +0530 Subject: [PATCH] up --- src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py index 5e4642c332..f63a8d64de 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py @@ -13,7 +13,7 @@ # limitations under the License. import inspect -from typing import Optional +from typing import Callable, Optional import numpy as np import torch @@ -600,7 +600,7 @@ class HunyuanDiTPAGPipeline(DiffusionPipeline, PAGMixin): negative_prompt_attention_mask_2: torch.Tensor | None = None, output_type: str = "pil", return_dict: bool = True, - callback_on_step_end: callable[[int, int, dict], None] + callback_on_step_end: Callable[[int, int, dict], None] | PipelineCallback | MultiPipelineCallbacks | None = None,