Skip to content

GitLab

Commit 516cb9e7 authored by patil-suraj's avatar patil-suraj
Browse files

fix Upsample

parent 60a98134
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/models/resnet.py
View file @ 516cb9e7
...@@ -88,6 +88,7 @@ class Upsample(nn.Module): ...@@ -88,6 +88,7 @@ class Upsample(nn.Module):
self.use_conv_transpose = use_conv_transpose self.use_conv_transpose = use_conv_transpose
self.name = name self.name = name
conv = None
if use_conv_transpose: if use_conv_transpose:
conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1) conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1)
elif use_conv: elif use_conv:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment