From 1fbaaf3b647b971b0f9ed254d52668d351bd27cc Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Sat, 14 Feb 2026 20:14:54 +0100 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 43fe9f75307c..15ac665acd2b 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py @@ -17,7 +17,7 @@ import numpy as np import torch -from transformers import BertModel, BertTokenizer, CLIPImageProcessor, MT5Tokenizer, T5EncoderModel +from transformers import BertModel, BertTokenizer, CLIPImageProcessor, T5EncoderModel, T5Tokenizer from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput @@ -208,7 +208,7 @@ def __init__( feature_extractor: CLIPImageProcessor | None = None, requires_safety_checker: bool = True, text_encoder_2: T5EncoderModel | None = None, - tokenizer_2: MT5Tokenizer | None = None, + tokenizer_2: T5Tokenizer | None = None, pag_applied_layers: str | list[str] = "blocks.1", # "blocks.16.attn1", "blocks.16", "16", 16 ): super().__init__()