fix Upsample

2022-07-01 12:58:50 +02:00 · 2022-07-01 12:58:50 +02:00 · 516cb9e7f8
parent 60a981343e
commit 516cb9e7f8
1 changed files with 1 additions and 0 deletions
--- a/src/diffusers/models/resnet.py
+++ b/src/diffusers/models/resnet.py
@ -88,6 +88,7 @@ class Upsample(nn.Module):
        self.use_conv_transpose = use_conv_transpose
        self.name = name

+        conv = None
        if use_conv_transpose:
            conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1)
        elif use_conv: