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

debug

Browse source
This commit is contained in:
shenchenhui 2024-03-28 22:51:22 +08:00
parent f1a6f7523e
commit 356ff604c0
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -47,9 +47,9 @@ class ResBlock(nn.Module):
self.norm2 = nn.GroupNorm(num_groups, self.filters, dtype=dtype, device=device)
self.conv2 = conv_fn(self.filters, self.filters, kernel_size=(3, 3, 3), bias=False)
if self.use_conv_shortcut:
self.conv3 = conv_fn(self.filters, self.filters, kernel_size=(3, 3, 3), bias=False)
self.conv3 = conv_fn(in_out_channels, self.filters, kernel_size=(3, 3, 3), bias=False)
else:
self.conv3 = conv_fn(self.filters, self.filters, kernel_size=(1, 1, 1), bias=False)
self.conv3 = conv_fn(in_out_channels, self.filters, kernel_size=(1, 1, 1), bias=False)
def forward(self, x):