Skip to content

Commit 7a805e4

Browse files
authored
Update pipeline_lumina2.py
1 parent 646e95e commit 7a805e4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/lumina2/pipeline_lumina2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -776,8 +776,8 @@ def __init__(
776776
transformer: Lumina2Transformer2DModel,
777777
scheduler: FlowMatchEulerDiscreteScheduler,
778778
vae: AutoencoderKL,
779-
text_encoder: AutoModel,
780-
tokenizer: AutoTokenizer,
779+
text_encoder: Gemma2PreTrainedModel,
780+
tokenizer: Union[GemmaTokenizer, GemmaTokenizerFast],
781781
):
782782
deprecation_message = "`Lumina2Text2ImgPipeline` has been renamed to `Lumina2Pipeline` and will be removed in a future version. Please use `Lumina2Pipeline` instead."
783783
deprecate("diffusers.pipelines.lumina2.pipeline_lumina2.Lumina2Text2ImgPipeline", "0.34", deprecation_message)

0 commit comments

Comments
 (0)