UmerHA commited on
Commit
813b9c2
·
1 Parent(s): 0c1b33a

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +44 -22
  2. diffusion_pytorch_model.safetensors +2 -2
config.json CHANGED
@@ -1,11 +1,6 @@
1
  {
2
  "_class_name": "ControlNetXSModel",
3
  "_diffusers_version": "0.24.0.dev0",
4
- "attention_head_dim": [
5
- 1,
6
- 1,
7
- 2
8
- ],
9
  "base_model_channel_sizes": {
10
  "down": [
11
  [
@@ -40,6 +35,18 @@
40
  640,
41
  1280
42
  ],
 
 
 
 
 
 
 
 
 
 
 
 
43
  [
44
  1280,
45
  1280
@@ -52,6 +59,18 @@
52
  ]
53
  ],
54
  "up": [
 
 
 
 
 
 
 
 
 
 
 
 
55
  [
56
  2560,
57
  1280
@@ -91,9 +110,10 @@
91
  ]
92
  },
93
  "block_out_channels": [
94
- 32,
95
- 64,
96
- 128
 
97
  ],
98
  "conditioning_channels": 3,
99
  "conditioning_embedding_out_channels": [
@@ -103,29 +123,31 @@
103
  256
104
  ],
105
  "controlnet_conditioning_channel_order": "rgb",
106
- "cross_attention_dim": 2048,
107
  "down_block_types": [
108
- "DownBlock2D",
109
  "CrossAttnDownBlock2D",
110
- "CrossAttnDownBlock2D"
 
 
111
  ],
112
  "learn_embedding": true,
113
- "norm_num_groups": 32,
114
- "num_attention_heads": null,
115
- "sample_size": 128,
116
- "time_embedding_dim": 1280,
117
- "time_embedding_input_dim": 320,
118
- "time_embedding_mix": 0.95,
119
- "transformer_layers_per_block": [
120
  1,
121
  2,
122
- 10
123
  ],
 
 
 
 
 
124
  "up_block_types": [
 
125
  "CrossAttnUpBlock2D",
126
  "CrossAttnUpBlock2D",
127
- "UpBlock2D"
128
  ],
129
- "upcast_attention": null,
130
- "use_linear_projection": true
131
  }
 
1
  {
2
  "_class_name": "ControlNetXSModel",
3
  "_diffusers_version": "0.24.0.dev0",
 
 
 
 
 
4
  "base_model_channel_sizes": {
5
  "down": [
6
  [
 
35
  640,
36
  1280
37
  ],
38
+ [
39
+ 1280,
40
+ 1280
41
+ ],
42
+ [
43
+ 1280,
44
+ 1280
45
+ ],
46
+ [
47
+ 1280,
48
+ 1280
49
+ ],
50
  [
51
  1280,
52
  1280
 
59
  ]
60
  ],
61
  "up": [
62
+ [
63
+ 2560,
64
+ 1280
65
+ ],
66
+ [
67
+ 2560,
68
+ 1280
69
+ ],
70
+ [
71
+ 2560,
72
+ 1280
73
+ ],
74
  [
75
  2560,
76
  1280
 
110
  ]
111
  },
112
  "block_out_channels": [
113
+ 4,
114
+ 8,
115
+ 16,
116
+ 16
117
  ],
118
  "conditioning_channels": 3,
119
  "conditioning_embedding_out_channels": [
 
123
  256
124
  ],
125
  "controlnet_conditioning_channel_order": "rgb",
126
+ "cross_attention_dim": 1024,
127
  "down_block_types": [
 
128
  "CrossAttnDownBlock2D",
129
+ "CrossAttnDownBlock2D",
130
+ "CrossAttnDownBlock2D",
131
+ "DownBlock2D"
132
  ],
133
  "learn_embedding": true,
134
+ "norm_num_groups": 4,
135
+ "num_attention_heads": [
136
+ 1,
 
 
 
 
137
  1,
138
  2,
139
+ 2
140
  ],
141
+ "sample_size": 64,
142
+ "time_embedding_dim": 1280,
143
+ "time_embedding_input_dim": 320,
144
+ "time_embedding_mix": 1.0,
145
+ "transformer_layers_per_block": 1,
146
  "up_block_types": [
147
+ "UpBlock2D",
148
  "CrossAttnUpBlock2D",
149
  "CrossAttnUpBlock2D",
150
+ "CrossAttnUpBlock2D"
151
  ],
152
+ "upcast_attention": false
 
153
  }
diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d4f07163d2be18ee9b08e219385805326b5f4a4d2b1d71ce785fd6047f275b4f
3
- size 169820736
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95090e89aae7a7a9ae3ebfd560f37a0dc08abe7244ea8ee1cb3c704992e55b22
3
+ size 56794304