diff --git a/configs/vae_magvit_v2/train/pipeline_16x128x128.py b/configs/vae_magvit_v2/train/pipeline_16x128x128.py index 8b2dd77..4b8bf35 100644 --- a/configs/vae_magvit_v2/train/pipeline_16x128x128.py +++ b/configs/vae_magvit_v2/train/pipeline_16x128x128.py @@ -37,7 +37,7 @@ model = dict( activation_fn = 'swish', separate_first_frame_encoding = False, disable_space = True, - ncoder_double_z = False, + encoder_double_z = False, custom_conv_padding = None )