modality_vae/config.json

{
  "model_type": "modality_completion_vae",
  "fusion_input_dims": 2816,
  "conv_dim": 256,
  "z_dim": 256,
  "n_codebook": 8192
}