model_config.json

{"name": "VAEConfig", "input_dim": [32, 50257], "latent_dim": 256, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse"}