| { | |
| "SPADE_type": "spade", | |
| "_class_name": "UNetModel", | |
| "_diffusers_version": "0.16.1", | |
| "_name_or_path": "/data/leiqin/diffusion/SLDM/SLDM-VAE-15-ade20k/checkpoint-32800/unet_ema", | |
| "attention_resolutions": [ | |
| 2, | |
| 4, | |
| 8 | |
| ], | |
| "channel_mult": [ | |
| 1, | |
| 2, | |
| 3, | |
| 4 | |
| ], | |
| "conv_resample": true, | |
| "decay": 0.9999, | |
| "dims": 2, | |
| "dropout": 0, | |
| "image_size": [ | |
| 64, | |
| 64 | |
| ], | |
| "in_channels": 4, | |
| "inv_gamma": 1.0, | |
| "mask_emb": "resize", | |
| "min_decay": 0.0, | |
| "model_channels": 256, | |
| "num_classes": 151, | |
| "num_head_channels": -1, | |
| "num_heads": 8, | |
| "num_heads_upsample": -1, | |
| "num_res_blocks": 2, | |
| "optimization_step": 2850, | |
| "out_channels": 4, | |
| "power": 0.75, | |
| "resblock_updown": true, | |
| "update_after_step": 0, | |
| "use_checkpoint": true, | |
| "use_ema_warmup": true, | |
| "use_fp16": true, | |
| "use_new_attention_order": false, | |
| "use_scale_shift_norm": true | |
| } | |