diff --git a/src/diffusers/models/resnet.py b/src/diffusers/models/resnet.py index bad14f7e..ba0f9e81 100644 --- a/src/diffusers/models/resnet.py +++ b/src/diffusers/models/resnet.py @@ -86,7 +86,7 @@ class Upsample(nn.Module): self.use_conv = use_conv self.dims = dims self.use_conv_transpose = use_conv_transpose - name = self.name + self.name = self.name if use_conv_transpose: conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1)