From 926dcc63195b35ffef35d05037b39cbe0470dee1 Mon Sep 17 00:00:00 2001 From: DN6 Date: Fri, 13 Jun 2025 13:43:17 +0530 Subject: [PATCH] update to pad tokens --- src/diffusers/pipelines/chroma/pipeline_chroma.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py index f3074e0d09..a215d972e6 100644 --- a/src/diffusers/pipelines/chroma/pipeline_chroma.py +++ b/src/diffusers/pipelines/chroma/pipeline_chroma.py @@ -226,7 +226,14 @@ class ChromaPipeline( return_overflowing_tokens=False, return_tensors="pt", ) - text_input_ids = text_inputs.input_ids + self.tokenizer.pad_token_id + pad_token_id = self.tokenizer.pad_token_id + text_input_ids = torch.cat( + [ + text_inputs.input_ids, + torch.full((text_inputs.input_ids.size(0), 1), pad_token_id, dtype=text_inputs.input_ids.dtype), + ], + dim=1, + ) prompt_embeds = self.text_encoder( text_input_ids.to(device),