{ "_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 }