From 7cdd7d2df00682eb89c32d7f27194e5cd37b9b8c Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Tue, 17 Jun 2025 05:19:30 +0200 Subject: [PATCH] update --- src/diffusers/models/transformers/transformer_chroma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/transformers/transformer_chroma.py b/src/diffusers/models/transformers/transformer_chroma.py index 45ae6a8781..d11f6c2a5e 100644 --- a/src/diffusers/models/transformers/transformer_chroma.py +++ b/src/diffusers/models/transformers/transformer_chroma.py @@ -671,7 +671,7 @@ class ChromaTransformer2DModel( ) if torch.is_grad_enabled() and self.gradient_checkpointing: encoder_hidden_states, hidden_states = self._gradient_checkpointing_func( - block, hidden_states, encoder_hidden_states, temb, image_rotary_emb, attention_mask=attention_mask + block, hidden_states, encoder_hidden_states, temb, image_rotary_emb, attention_mask ) else: