Update unet_2d_blocks.py

yao-matrix · web-flow · commit fd9fa9912172 · 2025-07-18T08:07:38.000-07:00
diff --git a/src/diffusers/models/unets/unet_2d_blocks.py b/src/diffusers/models/unets/unet_2d_blocks.py
@@ -2557,6 +2557,7 @@ def forward(
                     b1=self.b1,
                     b2=self.b2,
                 )
+
             hidden_states = torch.cat([hidden_states, res_hidden_states], dim=1)
 
             if torch.is_grad_enabled() and self.gradient_checkpointing:

Original file line number	Diff line number	Diff line change
`@@ -2557,6 +2557,7 @@ def forward(`
`2557`	`2557`	`b1=self.b1,`
`2558`	`2558`	`b2=self.b2,`
`2559`	`2559`	`)`
	`2560`	`+`
`2560`	`2561`	`hidden_states = torch.cat([hidden_states, res_hidden_states], dim=1)`
`2561`	`2562`
`2562`	`2563`	`if torch.is_grad_enabled() and self.gradient_checkpointing:`