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

save old config during trainin

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-19 15:16:14 +08:00
parent 78caeea999
commit d94abdb877
4 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
configs/vae_magvit_v2/inference/16x128x128.py
View file

@ -33,7 +33,8 @@ model = dict(
kl_embed_dim = 64,
activation_fn = 'swish',
separate_first_frame_encoding = False,
custom_conv_padding = None
custom_conv_padding = None,
encoder_double_z = False,
)

3
configs/vae_magvit_v2/inference/pipeline_16x128x128.py
View file

@ -42,7 +42,8 @@ model = dict(
activation_fn = 'swish',
separate_first_frame_encoding = False,
disable_space = True,
custom_conv_padding = None
custom_conv_padding = None,
encoder_double_z = False,
)

1
configs/vae_magvit_v2/train/16x128x128_trial.py
View file

@ -31,6 +31,7 @@ model = dict(
activation_fn = 'swish',
separate_first_frame_encoding = False,
disable_space = False,
encoder_double_z = False,
custom_conv_padding = None
)

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

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