ApacheOne commited on
Commit
2aad4f5
·
verified ·
1 Parent(s): ff64c8c

Delete creaprompt_UNet_OrbitQuant_W4A4_projio_report.txt

Browse files
creaprompt_UNet_OrbitQuant_W4A4_projio_report.txt DELETED
@@ -1,754 +0,0 @@
1
- OrbitQuant SDXL UNet-only custom export W4A4
2
- ========================================================================================================================
3
- Source: /content/creaprompt_sdxl/creapromptLightning_creapromtHypersdxlV1.safetensors
4
- Output: /content/creaprompt_sdxl/orbitquant_unet/creaprompt_UNet_OrbitQuant_W4A4_projio.safetensors
5
- Output file size: 1,826,788,784 / 1.701 GiB
6
-
7
- Original UNet
8
- ------------------------------------------------------------------------------------------------------------------------
9
- UNet keys: 1,680
10
- UNet params: 2,567,463,684
11
- UNet FP16: 5,134,927,368 / 4.782 GiB
12
-
13
- Targeted OrbitQuant weights
14
- ------------------------------------------------------------------------------------------------------------------------
15
- Target keys: 722
16
- Target params: 2,207,744,000
17
- Target FP16: 4,415,488,000 / 4.112 GiB
18
- Target share: 85.989298% of UNet params
19
- Input dims: [640, 1280, 2048, 2560, 5120]
20
-
21
- Custom stored size estimate
22
- ------------------------------------------------------------------------------------------------------------------------
23
- Untouched kept UNet bytes: 719,439,368 / 686.111 MiB
24
- Packed quantized index bytes: 1,103,872,000 / 1.028 GiB
25
- Row norm bytes: 2,872,320 / 2.739 MiB
26
- Estimate without metadata: 1,826,183,688 / 1.701 GiB
27
- Actual safetensors file: 1,826,788,784 / 1.701 GiB
28
- Saved from original UNet FP16: 3,308,138,584 / 3.081 GiB
29
- UNet shrink vs FP16: 64.4243%
30
-
31
- Quantized keys
32
- ------------------------------------------------------------------------------------------------------------------------
33
- model.diffusion_model.input_blocks.4.1.proj_in.weight shape=[640, 640] input_dim=640 output_dim=640
34
- model.diffusion_model.input_blocks.4.1.proj_out.weight shape=[640, 640] input_dim=640 output_dim=640
35
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
36
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
37
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
38
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
39
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
40
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
41
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
42
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
43
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
44
- model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
45
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
46
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
47
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
48
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
49
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
50
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
51
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
52
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
53
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
54
- model.diffusion_model.input_blocks.4.1.transformer_blocks.1.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
55
- model.diffusion_model.input_blocks.5.1.proj_in.weight shape=[640, 640] input_dim=640 output_dim=640
56
- model.diffusion_model.input_blocks.5.1.proj_out.weight shape=[640, 640] input_dim=640 output_dim=640
57
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
58
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
59
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
60
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
61
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
62
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
63
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
64
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
65
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
66
- model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
67
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
68
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
69
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
70
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
71
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
72
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
73
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
74
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
75
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
76
- model.diffusion_model.input_blocks.5.1.transformer_blocks.1.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
77
- model.diffusion_model.input_blocks.7.1.proj_in.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
78
- model.diffusion_model.input_blocks.7.1.proj_out.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
79
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
80
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
81
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
82
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
83
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
84
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
85
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
86
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
87
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
88
- model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
89
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
90
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
91
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
92
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
93
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
94
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
95
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
96
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
97
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
98
- model.diffusion_model.input_blocks.7.1.transformer_blocks.1.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
99
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
100
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
101
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
102
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
103
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
104
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
105
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
106
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
107
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
108
- model.diffusion_model.input_blocks.7.1.transformer_blocks.2.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
109
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
110
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
111
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
112
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
113
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
114
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
115
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
116
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
117
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
118
- model.diffusion_model.input_blocks.7.1.transformer_blocks.3.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
119
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
120
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
121
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
122
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
123
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
124
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
125
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
126
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
127
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
128
- model.diffusion_model.input_blocks.7.1.transformer_blocks.4.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
129
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
130
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
131
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
132
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
133
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
134
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
135
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
136
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
137
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
138
- model.diffusion_model.input_blocks.7.1.transformer_blocks.5.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
139
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
140
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
141
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
142
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
143
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
144
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
145
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
146
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
147
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
148
- model.diffusion_model.input_blocks.7.1.transformer_blocks.6.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
149
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
150
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
151
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
152
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
153
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
154
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
155
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
156
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
157
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
158
- model.diffusion_model.input_blocks.7.1.transformer_blocks.7.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
159
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
160
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
161
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
162
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
163
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
164
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
165
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
166
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
167
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
168
- model.diffusion_model.input_blocks.7.1.transformer_blocks.8.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
169
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
170
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
171
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
172
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
173
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
174
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
175
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
176
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
177
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
178
- model.diffusion_model.input_blocks.7.1.transformer_blocks.9.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
179
- model.diffusion_model.input_blocks.8.1.proj_in.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
180
- model.diffusion_model.input_blocks.8.1.proj_out.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
181
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
182
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
183
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
184
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
185
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
186
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
187
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
188
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
189
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
190
- model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
191
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
192
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
193
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
194
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
195
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
196
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
197
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
198
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
199
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
200
- model.diffusion_model.input_blocks.8.1.transformer_blocks.1.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
201
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
202
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
203
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
204
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
205
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
206
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
207
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
208
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
209
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
210
- model.diffusion_model.input_blocks.8.1.transformer_blocks.2.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
211
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
212
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
213
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
214
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
215
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
216
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
217
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
218
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
219
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
220
- model.diffusion_model.input_blocks.8.1.transformer_blocks.3.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
221
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
222
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
223
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
224
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
225
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
226
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
227
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
228
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
229
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
230
- model.diffusion_model.input_blocks.8.1.transformer_blocks.4.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
231
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
232
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
233
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
234
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
235
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
236
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
237
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
238
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
239
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
240
- model.diffusion_model.input_blocks.8.1.transformer_blocks.5.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
241
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
242
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
243
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
244
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
245
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
246
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
247
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
248
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
249
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
250
- model.diffusion_model.input_blocks.8.1.transformer_blocks.6.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
251
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
252
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
253
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
254
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
255
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
256
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
257
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
258
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
259
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
260
- model.diffusion_model.input_blocks.8.1.transformer_blocks.7.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
261
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
262
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
263
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
264
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
265
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
266
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
267
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
268
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
269
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
270
- model.diffusion_model.input_blocks.8.1.transformer_blocks.8.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
271
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
272
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
273
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
274
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
275
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
276
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
277
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
278
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
279
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
280
- model.diffusion_model.input_blocks.8.1.transformer_blocks.9.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
281
- model.diffusion_model.middle_block.1.proj_in.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
282
- model.diffusion_model.middle_block.1.proj_out.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
283
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
284
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
285
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
286
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
287
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
288
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
289
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
290
- model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
291
- model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
292
- model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
293
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
294
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
295
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
296
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
297
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
298
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
299
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
300
- model.diffusion_model.middle_block.1.transformer_blocks.1.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
301
- model.diffusion_model.middle_block.1.transformer_blocks.1.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
302
- model.diffusion_model.middle_block.1.transformer_blocks.1.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
303
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
304
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
305
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
306
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
307
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
308
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
309
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
310
- model.diffusion_model.middle_block.1.transformer_blocks.2.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
311
- model.diffusion_model.middle_block.1.transformer_blocks.2.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
312
- model.diffusion_model.middle_block.1.transformer_blocks.2.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
313
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
314
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
315
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
316
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
317
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
318
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
319
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
320
- model.diffusion_model.middle_block.1.transformer_blocks.3.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
321
- model.diffusion_model.middle_block.1.transformer_blocks.3.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
322
- model.diffusion_model.middle_block.1.transformer_blocks.3.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
323
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
324
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
325
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
326
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
327
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
328
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
329
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
330
- model.diffusion_model.middle_block.1.transformer_blocks.4.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
331
- model.diffusion_model.middle_block.1.transformer_blocks.4.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
332
- model.diffusion_model.middle_block.1.transformer_blocks.4.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
333
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
334
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
335
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
336
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
337
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
338
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
339
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
340
- model.diffusion_model.middle_block.1.transformer_blocks.5.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
341
- model.diffusion_model.middle_block.1.transformer_blocks.5.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
342
- model.diffusion_model.middle_block.1.transformer_blocks.5.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
343
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
344
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
345
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
346
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
347
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
348
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
349
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
350
- model.diffusion_model.middle_block.1.transformer_blocks.6.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
351
- model.diffusion_model.middle_block.1.transformer_blocks.6.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
352
- model.diffusion_model.middle_block.1.transformer_blocks.6.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
353
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
354
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
355
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
356
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
357
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
358
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
359
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
360
- model.diffusion_model.middle_block.1.transformer_blocks.7.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
361
- model.diffusion_model.middle_block.1.transformer_blocks.7.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
362
- model.diffusion_model.middle_block.1.transformer_blocks.7.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
363
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
364
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
365
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
366
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
367
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
368
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
369
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
370
- model.diffusion_model.middle_block.1.transformer_blocks.8.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
371
- model.diffusion_model.middle_block.1.transformer_blocks.8.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
372
- model.diffusion_model.middle_block.1.transformer_blocks.8.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
373
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
374
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
375
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
376
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
377
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
378
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
379
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
380
- model.diffusion_model.middle_block.1.transformer_blocks.9.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
381
- model.diffusion_model.middle_block.1.transformer_blocks.9.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
382
- model.diffusion_model.middle_block.1.transformer_blocks.9.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
383
- model.diffusion_model.output_blocks.0.1.proj_in.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
384
- model.diffusion_model.output_blocks.0.1.proj_out.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
385
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
386
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
387
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
388
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
389
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
390
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
391
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
392
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
393
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
394
- model.diffusion_model.output_blocks.0.1.transformer_blocks.0.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
395
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
396
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
397
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
398
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
399
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
400
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
401
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
402
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
403
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
404
- model.diffusion_model.output_blocks.0.1.transformer_blocks.1.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
405
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
406
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
407
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
408
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
409
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
410
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
411
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
412
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
413
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
414
- model.diffusion_model.output_blocks.0.1.transformer_blocks.2.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
415
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
416
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
417
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
418
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
419
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
420
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
421
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
422
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
423
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
424
- model.diffusion_model.output_blocks.0.1.transformer_blocks.3.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
425
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
426
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
427
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
428
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
429
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
430
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
431
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
432
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
433
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
434
- model.diffusion_model.output_blocks.0.1.transformer_blocks.4.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
435
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
436
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
437
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
438
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
439
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
440
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
441
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
442
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
443
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
444
- model.diffusion_model.output_blocks.0.1.transformer_blocks.5.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
445
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
446
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
447
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
448
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
449
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
450
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
451
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
452
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
453
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
454
- model.diffusion_model.output_blocks.0.1.transformer_blocks.6.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
455
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
456
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
457
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
458
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
459
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
460
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
461
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
462
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
463
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
464
- model.diffusion_model.output_blocks.0.1.transformer_blocks.7.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
465
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
466
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
467
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
468
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
469
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
470
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
471
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
472
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
473
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
474
- model.diffusion_model.output_blocks.0.1.transformer_blocks.8.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
475
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
476
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
477
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
478
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
479
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
480
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
481
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
482
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
483
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
484
- model.diffusion_model.output_blocks.0.1.transformer_blocks.9.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
485
- model.diffusion_model.output_blocks.1.1.proj_in.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
486
- model.diffusion_model.output_blocks.1.1.proj_out.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
487
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
488
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
489
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
490
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
491
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
492
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
493
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
494
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
495
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
496
- model.diffusion_model.output_blocks.1.1.transformer_blocks.0.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
497
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
498
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
499
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
500
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
501
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
502
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
503
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
504
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
505
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
506
- model.diffusion_model.output_blocks.1.1.transformer_blocks.1.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
507
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
508
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
509
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
510
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
511
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
512
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
513
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
514
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
515
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
516
- model.diffusion_model.output_blocks.1.1.transformer_blocks.2.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
517
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
518
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
519
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
520
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
521
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
522
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
523
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
524
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
525
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
526
- model.diffusion_model.output_blocks.1.1.transformer_blocks.3.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
527
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
528
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
529
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
530
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
531
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
532
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
533
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
534
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
535
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
536
- model.diffusion_model.output_blocks.1.1.transformer_blocks.4.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
537
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
538
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
539
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
540
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
541
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
542
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
543
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
544
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
545
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
546
- model.diffusion_model.output_blocks.1.1.transformer_blocks.5.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
547
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
548
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
549
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
550
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
551
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
552
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
553
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
554
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
555
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
556
- model.diffusion_model.output_blocks.1.1.transformer_blocks.6.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
557
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
558
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
559
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
560
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
561
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
562
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
563
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
564
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
565
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
566
- model.diffusion_model.output_blocks.1.1.transformer_blocks.7.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
567
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
568
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
569
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
570
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
571
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
572
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
573
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
574
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
575
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
576
- model.diffusion_model.output_blocks.1.1.transformer_blocks.8.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
577
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
578
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
579
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
580
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
581
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
582
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
583
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
584
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
585
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
586
- model.diffusion_model.output_blocks.1.1.transformer_blocks.9.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
587
- model.diffusion_model.output_blocks.2.1.proj_in.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
588
- model.diffusion_model.output_blocks.2.1.proj_out.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
589
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
590
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
591
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
592
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
593
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
594
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
595
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
596
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
597
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
598
- model.diffusion_model.output_blocks.2.1.transformer_blocks.0.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
599
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
600
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
601
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
602
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
603
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
604
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
605
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
606
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
607
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
608
- model.diffusion_model.output_blocks.2.1.transformer_blocks.1.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
609
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
610
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
611
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
612
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
613
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
614
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
615
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
616
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
617
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
618
- model.diffusion_model.output_blocks.2.1.transformer_blocks.2.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
619
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
620
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
621
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
622
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
623
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
624
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
625
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
626
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
627
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
628
- model.diffusion_model.output_blocks.2.1.transformer_blocks.3.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
629
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
630
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
631
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
632
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
633
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
634
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
635
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
636
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
637
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
638
- model.diffusion_model.output_blocks.2.1.transformer_blocks.4.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
639
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
640
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
641
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
642
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
643
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
644
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
645
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
646
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
647
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
648
- model.diffusion_model.output_blocks.2.1.transformer_blocks.5.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
649
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
650
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
651
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
652
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
653
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
654
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
655
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
656
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
657
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
658
- model.diffusion_model.output_blocks.2.1.transformer_blocks.6.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
659
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
660
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
661
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
662
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
663
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
664
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
665
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
666
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
667
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
668
- model.diffusion_model.output_blocks.2.1.transformer_blocks.7.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
669
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
670
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
671
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
672
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
673
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
674
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
675
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
676
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
677
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
678
- model.diffusion_model.output_blocks.2.1.transformer_blocks.8.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
679
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_k.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
680
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
681
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
682
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_v.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
683
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn2.to_k.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
684
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn2.to_out.0.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
685
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn2.to_q.weight shape=[1280, 1280] input_dim=1280 output_dim=1280
686
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.attn2.to_v.weight shape=[1280, 2048] input_dim=2048 output_dim=1280
687
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.ff.net.0.proj.weight shape=[10240, 1280] input_dim=1280 output_dim=10240
688
- model.diffusion_model.output_blocks.2.1.transformer_blocks.9.ff.net.2.weight shape=[1280, 5120] input_dim=5120 output_dim=1280
689
- model.diffusion_model.output_blocks.3.1.proj_in.weight shape=[640, 640] input_dim=640 output_dim=640
690
- model.diffusion_model.output_blocks.3.1.proj_out.weight shape=[640, 640] input_dim=640 output_dim=640
691
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
692
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
693
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
694
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
695
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
696
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
697
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
698
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
699
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
700
- model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
701
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
702
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
703
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
704
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
705
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
706
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
707
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
708
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
709
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
710
- model.diffusion_model.output_blocks.3.1.transformer_blocks.1.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
711
- model.diffusion_model.output_blocks.4.1.proj_in.weight shape=[640, 640] input_dim=640 output_dim=640
712
- model.diffusion_model.output_blocks.4.1.proj_out.weight shape=[640, 640] input_dim=640 output_dim=640
713
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
714
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
715
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
716
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
717
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
718
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
719
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
720
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
721
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
722
- model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
723
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
724
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
725
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
726
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
727
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
728
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
729
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
730
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
731
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
732
- model.diffusion_model.output_blocks.4.1.transformer_blocks.1.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
733
- model.diffusion_model.output_blocks.5.1.proj_in.weight shape=[640, 640] input_dim=640 output_dim=640
734
- model.diffusion_model.output_blocks.5.1.proj_out.weight shape=[640, 640] input_dim=640 output_dim=640
735
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
736
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
737
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
738
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
739
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
740
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
741
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
742
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
743
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
744
- model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640
745
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_k.weight shape=[640, 640] input_dim=640 output_dim=640
746
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
747
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
748
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_v.weight shape=[640, 640] input_dim=640 output_dim=640
749
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn2.to_k.weight shape=[640, 2048] input_dim=2048 output_dim=640
750
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn2.to_out.0.weight shape=[640, 640] input_dim=640 output_dim=640
751
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn2.to_q.weight shape=[640, 640] input_dim=640 output_dim=640
752
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.attn2.to_v.weight shape=[640, 2048] input_dim=2048 output_dim=640
753
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.ff.net.0.proj.weight shape=[5120, 640] input_dim=640 output_dim=5120
754
- model.diffusion_model.output_blocks.5.1.transformer_blocks.1.ff.net.2.weight shape=[640, 2560] input_dim=2560 output_dim=640