Update vae/config.json

vae/config.json

@@ -1 +1 @@
-{"hidden_size": 128, "in_channels": 3, "out_channels": 3, "z_channels": 4, "ch_mult": [1, 2, 4, 4], "num_res_block": 2, "attn_resolutions": [], "dropout": 0.0, "resolution": 64, "attn_type": "vanilla3D", "use_linear_attn": false, "embed_dim": 4, "time_compress": 2, "logvar_init": 0.0, "kl_weight": 1e-06, "pixelloss_weight": 1, "perceptual_weight": 1, "disc_loss": "hinge", "model": "CausalVAEModel"}
+{"_class_name": "CausalVAEModel", "hidden_size": 128, "in_channels": 3, "out_channels": 3, "z_channels": 4, "ch_mult": [1, 2, 4, 4], "num_res_block": 2, "attn_resolutions": [], "dropout": 0.0, "resolution": 64, "attn_type": "vanilla3D", "use_linear_attn": false, "embed_dim": 4, "time_compress": 2, "logvar_init": 0.0, "kl_weight": 1e-06, "pixelloss_weight": 1, "perceptual_weight": 1, "disc_loss": "hinge", "model": "CausalVAEModel"}