diff --git a/src/diffusers/models/resnet.py b/src/diffusers/models/resnet.py index 7a48d343a531..23af7d39fd9e 100644 --- a/src/diffusers/models/resnet.py +++ b/src/diffusers/models/resnet.py @@ -752,8 +752,8 @@ def forward( hidden_states = self.norm2(hidden_states) if temb is not None and self.time_embedding_norm == "scale_shift": - scale, shift = torch.chunk(temb, 2, dim=1) - hidden_states = hidden_states * (1 + scale) + shift + _scale, _shift = torch.chunk(temb, 2, dim=1) + hidden_states = hidden_states * (1 + _scale) + _shift hidden_states = self.nonlinearity(hidden_states)