Skip to content

Commit 646e95e

Browse files
authored
Update pipeline_lumina.py
1 parent 9747c0b commit 646e95e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/lumina/pipeline_lumina.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -939,8 +939,8 @@ def __init__(
939939
transformer: LuminaNextDiT2DModel,
940940
scheduler: FlowMatchEulerDiscreteScheduler,
941941
vae: AutoencoderKL,
942-
text_encoder: AutoModel,
943-
tokenizer: AutoTokenizer,
942+
text_encoder: GemmaPreTrainedModel,
943+
tokenizer: Union[GemmaTokenizer, GemmaTokenizerFast],
944944
):
945945
deprecation_message = "`LuminaText2ImgPipeline` has been renamed to `LuminaPipeline` and will be removed in a future version. Please use `LuminaPipeline` instead."
946946
deprecate("diffusers.pipelines.lumina.pipeline_lumina.LuminaText2ImgPipeline", "0.34", deprecation_message)

0 commit comments

Comments
 (0)