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 17:15:04 +08:00
parent bdf8a1d144
commit d80ef70020
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -222,6 +222,7 @@ class Decoder(nn.Module):
self.channel_multipliers = channel_multipliers
self.temporal_downsample = temporal_downsample
self.num_groups = num_groups
self.dtype = dtype
if isinstance(self.temporal_downsample, int):
self.temporal_downsample = _get_selected_flags(