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

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-18 13:29:59 +08:00
parent 8bf4a0fa77
commit f867ebc819
3 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
configs/vae_magvit_v2/train/16x128x128.py
View file

@ -30,7 +30,7 @@ model = dict(
kl_embed_dim = 64,
activation_fn = 'swish',
separate_first_frame_encoding = False,
disable_space = True,
disable_space = False,
custom_conv_padding = None
)

1
configs/vae_magvit_v2/train/pipeline_16x128x128.py
View file

@ -30,6 +30,7 @@ model = dict(
kl_embed_dim = 64,
activation_fn = 'swish',
separate_first_frame_encoding = False,
disable_space = True,
custom_conv_padding = None
)

2
opensora/models/vae/vae_3d_v2.py
View file

@ -617,7 +617,7 @@ class Decoder(nn.Module):
self.num_groups = num_groups
# self.upsample = upsample
self.s_stride = 1 if self.disable_spatial_upsample else 2 # spatial stride
self.s_stride = 1 if disable_spatial_upsample else 2 # spatial stride
self.custom_conv_padding = custom_conv_padding
# self.norm_type = self.config.vqvae.norm_type
# self.num_remat_block = self.config.vqvae.get('num_dec_remat_blocks', 0)