File size: 797 Bytes
f8372bd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
{
"SPADE_type": "spade",
"_class_name": "UNetModel",
"_diffusers_version": "0.16.1",
"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": 32800,
"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
}
|