model_config.json

{"name": "JNFConfig", "n_modalities": 2, "latent_dim": 96, "input_dims": null, "uses_likelihood_rescaling": false, "rescale_factors": null, "decoders_dist": {"image": "laplace", "text": "categorical"}, "decoder_dist_params": {}, "logits_to_std": "softplus", "custom_architectures": ["encoders", "decoders", "joint_encoder", "flows"], "warmup": 150, "two_steps_training": true, "alpha": 0.1, "beta": 1.0, "divide_by_prior": true}