Skip to content

Commit 5b00a18

Browse files
authored
fix MT5Tokenizer (#13146)
up
1 parent 6141ae2 commit 5b00a18

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
import numpy as np
1919
import torch
20-
from transformers import BertModel, BertTokenizer, CLIPImageProcessor, MT5Tokenizer, T5EncoderModel
20+
from transformers import BertModel, BertTokenizer, CLIPImageProcessor, T5EncoderModel, T5Tokenizer
2121

2222
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
2323

@@ -208,7 +208,7 @@ def __init__(
208208
feature_extractor: CLIPImageProcessor | None = None,
209209
requires_safety_checker: bool = True,
210210
text_encoder_2: T5EncoderModel | None = None,
211-
tokenizer_2: MT5Tokenizer | None = None,
211+
tokenizer_2: T5Tokenizer | None = None,
212212
pag_applied_layers: str | list[str] = "blocks.1", # "blocks.16.attn1", "blocks.16", "16", 16
213213
):
214214
super().__init__()

0 commit comments

Comments
 (0)