From 5b00a18374b834bfb32a45406f1a72a0a153fbd2 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Sat, 14 Feb 2026 09:40:07 -1000 Subject: [PATCH] fix MT5Tokenizer (#13146) 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 43fe9f7530..15ac665acd 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py @@ -17,7 +17,7 @@ from typing import Callable 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 @@ class HunyuanDiTPAGPipeline(DiffusionPipeline, PAGMixin): 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__()