config.json

{
  "model_type": "autoencoder_kl",
  "sample_size": 256,
  "in_channels": 3,
  "out_channels": 3,
  "latent_channels": 8,
  "base_channels": 128,
  "channel_multipliers": [1, 2, 4, 4],
  "num_res_blocks": 3,
  "dropout": 0.0,
  "use_attention": true,
  "attention_resolutions": [32],
  "attention_heads": 4,
  "scaling_factor": 1.0,
  "recon_loss_type": "l1",
  "trained_on": "coco_captions",
  "resolution": 256
}