| { | |
| "_class_name": "CausalVAEModel", | |
| "_diffusers_version": "0.27.2", | |
| "_name_or_path": "../results/pretrained_488_tail", | |
| "attn_resolutions": [], | |
| "decoder_attention": "AttnBlock3DFix", | |
| "decoder_conv_in": "CausalConv3d", | |
| "decoder_conv_out": "CausalConv3d", | |
| "decoder_mid_resnet": "ResnetBlock3D", | |
| "decoder_resnet_blocks": [ | |
| "ResnetBlock3D", | |
| "ResnetBlock3D", | |
| "ResnetBlock3D", | |
| "ResnetBlock3D" | |
| ], | |
| "decoder_spatial_upsample": [ | |
| "", | |
| "SpatialUpsample2x", | |
| "SpatialUpsample2x", | |
| "SpatialUpsample2x" | |
| ], | |
| "decoder_temporal_upsample": [ | |
| "", | |
| "", | |
| "TimeUpsampleRes2x", | |
| "TimeUpsampleRes2x" | |
| ], | |
| "double_z": true, | |
| "dropout": 0.0, | |
| "embed_dim": 4, | |
| "encoder_attention": "AttnBlock3DFix", | |
| "encoder_conv_in": "Conv2d", | |
| "encoder_conv_out": "CausalConv3d", | |
| "encoder_mid_resnet": "ResnetBlock3D", | |
| "encoder_resnet_blocks": [ | |
| "ResnetBlock2D", | |
| "ResnetBlock2D", | |
| "ResnetBlock3D", | |
| "ResnetBlock3D" | |
| ], | |
| "encoder_spatial_downsample": [ | |
| "Downsample", | |
| "Downsample", | |
| "Downsample", | |
| "" | |
| ], | |
| "encoder_temporal_downsample": [ | |
| "", | |
| "TimeDownsampleRes2x", | |
| "TimeDownsampleRes2x", | |
| "" | |
| ], | |
| "hidden_size": 128, | |
| "hidden_size_mult": [ | |
| 1, | |
| 2, | |
| 4, | |
| 4 | |
| ], | |
| "in_channels": 3, | |
| "loss_params": { | |
| "disc_start": 2001, | |
| "disc_weight": 0.5, | |
| "kl_weight": 1e-06, | |
| "logvar_init": 0.0 | |
| }, | |
| "loss_type": "opensora.models.ae.videobase.losses.LPIPSWithDiscriminator3D", | |
| "lr": 1e-05, | |
| "num_res_blocks": 2, | |
| "out_channels": 3, | |
| "q_conv": "CausalConv3d", | |
| "resolution": 256, | |
| "z_channels": 4 | |
| } | |