ApacheOne commited on
Commit
c1dabef
·
verified ·
1 Parent(s): ffdf7f6

Upload creaprompt_UNet_OrbitQuant_W4A4_projio_report.txt

Browse files
creaprompt_UNet_OrbitQuant_W4A4_projio_report.txt ADDED
@@ -0,0 +1,754 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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