AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

update

Browse Source
This commit is contained in:
Dhruv Nair
2025-06-17 05:19:30 +02:00
parent 172b2ef73b
commit 7cdd7d2df0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/transformers/transformer_chroma.py
View File

@@ -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: