refactor: removing redundant type conversion

akshay-babbar · akshay-babbar · commit 26c33efabb28 · 2025-09-25T16:54:31.000+05:30
diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py
@@ -233,7 +233,6 @@ def _get_t5_prompt_embeds(
         )
         text_input_ids = text_inputs.input_ids
         attention_mask = text_inputs.attention_mask.clone()
-        attention_mask = attention_mask.bool()
 
         # Chroma requires the attention mask to include one padding token
         seq_lengths = attention_mask.sum(dim=1)

Original file line number	Diff line number	Diff line change
`@@ -233,7 +233,6 @@ def _get_t5_prompt_embeds(`
`233`	`233`	`)`
`234`	`234`	`text_input_ids = text_inputs.input_ids`
`235`	`235`	`attention_mask = text_inputs.attention_mask.clone()`
`236`		`- attention_mask = attention_mask.bool()`
`237`	`236`
`238`	`237`	`# Chroma requires the attention mask to include one padding token`
`239`	`238`	`seq_lengths = attention_mask.sum(dim=1)`