Skip to content

Commit 50d0a28

Browse files
committed
make fix-copies
1 parent 066465e commit 50d0a28

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

src/diffusers/models/unets/unet_2d_blocks.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -739,11 +739,7 @@ def forward(self, hidden_states: torch.Tensor, temb: Optional[torch.Tensor] = No
739739
if torch.is_grad_enabled() and self.gradient_checkpointing:
740740
if attn is not None:
741741
hidden_states = attn(hidden_states, temb=temb)
742-
hidden_states = self._gradient_checkpointing_func(
743-
resnet,
744-
hidden_states,
745-
temb,
746-
)
742+
hidden_states = self._gradient_checkpointing_func(resnet, hidden_states, temb)
747743
else:
748744
if attn is not None:
749745
hidden_states = attn(hidden_states, temb=temb)

0 commit comments

Comments
 (0)