Set attention_mask to dtype=torch.bool for ChromaInpaintPipeline.

hameerabbasi · hameerabbasi · commit 229629143356 · 2026-01-12T11:22:08.000+01:00
diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma_inpainting.py b/src/diffusers/pipelines/chroma/pipeline_chroma_inpainting.py
@@ -761,7 +761,7 @@ def _prepare_attention_mask(
 
         # Extend the prompt attention mask to account for image tokens in the final sequence
         attention_mask = torch.cat(
-            [attention_mask, torch.ones(batch_size, sequence_length, device=attention_mask.device)],
+            [attention_mask, torch.ones(batch_size, sequence_length, device=attention_mask.device, dtype=torch.bool)],
             dim=1,
         )
         attention_mask = attention_mask.to(dtype)

Original file line number	Diff line number	Diff line change
`@@ -761,7 +761,7 @@ def _prepare_attention_mask(`
`761`	`761`
`762`	`762`	`# Extend the prompt attention mask to account for image tokens in the final sequence`
`763`	`763`	`attention_mask = torch.cat(`
`764`		`- [attention_mask, torch.ones(batch_size, sequence_length, device=attention_mask.device)],`
	`764`	`+ [attention_mask, torch.ones(batch_size, sequence_length, device=attention_mask.device, dtype=torch.bool)],`
`765`	`765`	`dim=1,`
`766`	`766`	`)`
`767`	`767`	`attention_mask = attention_mask.to(dtype)`