weifeng-chen's picture
cyberpunk init
53576b9
raw
history blame
609 Bytes
{
"architectures": [
"HFUNetModel"
],
"attention_resolutions": [
16,
32,
64
],
"channel_mult": [
0.5,
1,
1,
2,
2,
4,
4
],
"dropout": 0.0,
"image_size": 512,
"in_channels": 3,
"model_channels": 256,
"num_classes": null,
"num_head_channels": 64,
"num_heads": 4,
"num_heads_upsample": -1,
"num_res_blocks": 2,
"out_channels": 6,
"resblock_updown": true,
"torch_dtype": "float32",
"transformers_version": "4.22.2",
"use_checkpoint": false,
"use_fp16": true,
"use_new_attention_order": false,
"use_scale_shift_norm": true
}