config.json

{
  "num_components": 4,
  "resize_shape": [
    240,
    320
  ],
  "unet_cond_dim": 768,
  "unet_n_freeze_layers": 1,
  "unet_pretrained": true
}