diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py
index e2081405c0..e376a402e5 100644
--- a/src/diffusers/pipelines/chroma/pipeline_chroma.py
+++ b/src/diffusers/pipelines/chroma/pipeline_chroma.py
@@ -220,14 +220,12 @@ class ChromaPipeline(
             padding="max_length",
             max_length=max_sequence_length,
             truncation=True,
-            return_length=True,
+            return_length=False,
             return_overflowing_tokens=False,
             return_tensors="pt",
         )
         text_input_ids = text_inputs.input_ids
         untruncated_ids = self.tokenizer(prompt, padding="longest", return_tensors="pt").input_ids
-
-        #text_inputs.attention_mask[:, : text_inputs.length + 1] = 1.0
         
         prompt_embeds = self.text_encoder(
             text_input_ids.to(device),