nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-20 17:35:58 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-03-28 23:24:37 +08:00
parent 1c3d6d4c4c
commit 907c71bcd6
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
opensora/models/vae/vae_3d.py
View file

@ -336,14 +336,14 @@ class Decoder(nn.Module):
if self.upsample == 'deconv':
assert self.custom_conv_padding is None, ('Custom padding not implemented for ConvTranspose')
x = self.conv_blocks[i-1](x)
elif self.upsample == 'nearest+conv':
if self.temporal_downsample[i - 1]:
x = self.upsampler_with_t(x)
elif self.upsample == 'nearest+conv':
if self.temporal_downsample[i - 1]:
x = self.upsampler_with_t(x)
else:
x = self.upsampler(x)
x = self.conv_blocks[i-1](x)
else:
x = self.upsampler(x)
x = self.conv_blocks[i-1](x)
else:
raise NotImplementedError(f'Unknown upsampler: {self.upsample}')
raise NotImplementedError(f'Unknown upsampler: {self.upsample}')
x = self.norm1(x)
x = self.activate(x)
x = self.conv2(x)