config.json

{
  "_model_class": "UNetModel",
  "channels": 3,
  "dim": 8,
  "dim_mults": [
    1,
    2
  ],
  "init_dim": null,
  "learned_variance": false,
  "out_dim": null,
  "resnet_block_groups": 2,
  "with_time_emb": true
}