UmerHA's picture
Upload 2 files
6927317
{
"_class_name": "ControlNetXSModel",
"_diffusers_version": "0.24.0.dev0",
"base_model_channel_sizes": {
"down": [
[
4,
320
],
[
320,
320
],
[
320,
320
],
[
320,
320
],
[
320,
640
],
[
640,
640
],
[
640,
640
],
[
640,
1280
],
[
1280,
1280
],
[
1280,
1280
],
[
1280,
1280
],
[
1280,
1280
]
],
"mid": [
[
1280,
1280
]
],
"up": [
[
2560,
1280
],
[
2560,
1280
],
[
2560,
1280
],
[
2560,
1280
],
[
2560,
1280
],
[
1920,
1280
],
[
1920,
640
],
[
1280,
640
],
[
960,
640
],
[
960,
320
],
[
640,
320
],
[
640,
320
]
]
},
"block_out_channels": [
4,
8,
16,
16
],
"conditioning_channels": 3,
"conditioning_embedding_out_channels": [
16,
32,
96,
256
],
"controlnet_conditioning_channel_order": "rgb",
"cross_attention_dim": 1024,
"down_block_types": [
"CrossAttnDownBlock2D",
"CrossAttnDownBlock2D",
"CrossAttnDownBlock2D",
"DownBlock2D"
],
"learn_embedding": true,
"norm_num_groups": 4,
"num_attention_heads": [
1,
1,
2,
2
],
"sample_size": 64,
"time_embedding_dim": 1280,
"time_embedding_input_dim": 320,
"time_embedding_mix": 1.0,
"transformer_layers_per_block": 1,
"up_block_types": [
"UpBlock2D",
"CrossAttnUpBlock2D",
"CrossAttnUpBlock2D",
"CrossAttnUpBlock2D"
],
"upcast_attention": false
}