fix: redefinition

huggingface · Nov 18, 2023 · bd81f75 · bd81f75
1 parent 4e54dfe
commit bd81f75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)