Created
January 25, 2023 17:09
-
-
Save mcmonkey4eva/af0b92249a00ee179fdeff15bc49b44d to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
{ | |
"__metadata__": { | |
"format": "pt" | |
}, | |
"alphas_cumprod": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
0, | |
4000 | |
] | |
}, | |
"alphas_cumprod_prev": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
4000, | |
8000 | |
] | |
}, | |
"betas": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
8000, | |
12000 | |
] | |
}, | |
"cond_stage_model.model.ln_final.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
12000, | |
16096 | |
] | |
}, | |
"cond_stage_model.model.ln_final.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
16096, | |
20192 | |
] | |
}, | |
"cond_stage_model.model.logit_scale": { | |
"dtype": "F32", | |
"shape": [], | |
"data_offsets": [ | |
20192, | |
20196 | |
] | |
}, | |
"cond_stage_model.model.positional_embedding": { | |
"dtype": "F32", | |
"shape": [ | |
77, | |
1024 | |
], | |
"data_offsets": [ | |
20196, | |
335588 | |
] | |
}, | |
"cond_stage_model.model.text_projection": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
335588, | |
4529892 | |
] | |
}, | |
"cond_stage_model.model.token_embedding.weight": { | |
"dtype": "F32", | |
"shape": [ | |
49408, | |
1024 | |
], | |
"data_offsets": [ | |
4529892, | |
206905060 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
206905060, | |
206917348 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
206917348, | |
219500260 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
219500260, | |
219504356 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
219504356, | |
223698660 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
223698660, | |
223702756 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
223702756, | |
223706852 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
223706852, | |
223710948 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
223710948, | |
223715044 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
223715044, | |
223731428 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
223731428, | |
240508644 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
240508644, | |
240512740 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.0.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
240512740, | |
257289956 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
257289956, | |
257302244 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
257302244, | |
269885156 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
269885156, | |
269889252 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
269889252, | |
274083556 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
274083556, | |
274087652 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
274087652, | |
274091748 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
274091748, | |
274095844 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
274095844, | |
274099940 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
274099940, | |
274116324 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
274116324, | |
290893540 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
290893540, | |
290897636 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.1.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
290897636, | |
307674852 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
307674852, | |
307687140 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
307687140, | |
320270052 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
320270052, | |
320274148 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
320274148, | |
324468452 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
324468452, | |
324472548 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
324472548, | |
324476644 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
324476644, | |
324480740 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
324480740, | |
324484836 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
324484836, | |
324501220 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
324501220, | |
341278436 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
341278436, | |
341282532 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.10.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
341282532, | |
358059748 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
358059748, | |
358072036 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
358072036, | |
370654948 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
370654948, | |
370659044 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
370659044, | |
374853348 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
374853348, | |
374857444 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
374857444, | |
374861540 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
374861540, | |
374865636 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
374865636, | |
374869732 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
374869732, | |
374886116 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
374886116, | |
391663332 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
391663332, | |
391667428 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.11.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
391667428, | |
408444644 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
408444644, | |
408456932 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
408456932, | |
421039844 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
421039844, | |
421043940 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
421043940, | |
425238244 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
425238244, | |
425242340 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
425242340, | |
425246436 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
425246436, | |
425250532 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
425250532, | |
425254628 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
425254628, | |
425271012 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
425271012, | |
442048228 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
442048228, | |
442052324 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.12.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
442052324, | |
458829540 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
458829540, | |
458841828 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
458841828, | |
471424740 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
471424740, | |
471428836 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
471428836, | |
475623140 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
475623140, | |
475627236 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
475627236, | |
475631332 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
475631332, | |
475635428 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
475635428, | |
475639524 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
475639524, | |
475655908 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
475655908, | |
492433124 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
492433124, | |
492437220 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.13.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
492437220, | |
509214436 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
509214436, | |
509226724 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
509226724, | |
521809636 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
521809636, | |
521813732 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
521813732, | |
526008036 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
526008036, | |
526012132 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
526012132, | |
526016228 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
526016228, | |
526020324 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
526020324, | |
526024420 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
526024420, | |
526040804 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
526040804, | |
542818020 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
542818020, | |
542822116 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.14.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
542822116, | |
559599332 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
559599332, | |
559611620 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
559611620, | |
572194532 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
572194532, | |
572198628 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
572198628, | |
576392932 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
576392932, | |
576397028 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
576397028, | |
576401124 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
576401124, | |
576405220 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
576405220, | |
576409316 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
576409316, | |
576425700 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
576425700, | |
593202916 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
593202916, | |
593207012 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.15.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
593207012, | |
609984228 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
609984228, | |
609996516 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
609996516, | |
622579428 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
622579428, | |
622583524 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
622583524, | |
626777828 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
626777828, | |
626781924 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
626781924, | |
626786020 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
626786020, | |
626790116 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
626790116, | |
626794212 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
626794212, | |
626810596 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
626810596, | |
643587812 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
643587812, | |
643591908 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.16.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
643591908, | |
660369124 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
660369124, | |
660381412 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
660381412, | |
672964324 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
672964324, | |
672968420 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
672968420, | |
677162724 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
677162724, | |
677166820 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
677166820, | |
677170916 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
677170916, | |
677175012 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
677175012, | |
677179108 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
677179108, | |
677195492 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
677195492, | |
693972708 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
693972708, | |
693976804 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.17.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
693976804, | |
710754020 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
710754020, | |
710766308 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
710766308, | |
723349220 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
723349220, | |
723353316 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
723353316, | |
727547620 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
727547620, | |
727551716 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
727551716, | |
727555812 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
727555812, | |
727559908 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
727559908, | |
727564004 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
727564004, | |
727580388 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
727580388, | |
744357604 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
744357604, | |
744361700 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.18.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
744361700, | |
761138916 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
761138916, | |
761151204 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
761151204, | |
773734116 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
773734116, | |
773738212 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
773738212, | |
777932516 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
777932516, | |
777936612 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
777936612, | |
777940708 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
777940708, | |
777944804 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
777944804, | |
777948900 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
777948900, | |
777965284 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
777965284, | |
794742500 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
794742500, | |
794746596 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.19.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
794746596, | |
811523812 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
811523812, | |
811536100 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
811536100, | |
824119012 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
824119012, | |
824123108 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
824123108, | |
828317412 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
828317412, | |
828321508 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
828321508, | |
828325604 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
828325604, | |
828329700 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
828329700, | |
828333796 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
828333796, | |
828350180 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
828350180, | |
845127396 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
845127396, | |
845131492 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.2.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
845131492, | |
861908708 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
861908708, | |
861920996 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
861920996, | |
874503908 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
874503908, | |
874508004 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
874508004, | |
878702308 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
878702308, | |
878706404 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
878706404, | |
878710500 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
878710500, | |
878714596 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
878714596, | |
878718692 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
878718692, | |
878735076 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
878735076, | |
895512292 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
895512292, | |
895516388 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.20.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
895516388, | |
912293604 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
912293604, | |
912305892 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
912305892, | |
924888804 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
924888804, | |
924892900 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
924892900, | |
929087204 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
929087204, | |
929091300 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
929091300, | |
929095396 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
929095396, | |
929099492 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
929099492, | |
929103588 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
929103588, | |
929119972 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
929119972, | |
945897188 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
945897188, | |
945901284 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.21.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
945901284, | |
962678500 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
962678500, | |
962690788 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
962690788, | |
975273700 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
975273700, | |
975277796 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
975277796, | |
979472100 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
979472100, | |
979476196 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
979476196, | |
979480292 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
979480292, | |
979484388 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
979484388, | |
979488484 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
979488484, | |
979504868 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
979504868, | |
996282084 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
996282084, | |
996286180 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.22.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
996286180, | |
1013063396 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1013063396, | |
1013075684 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1013075684, | |
1025658596 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1025658596, | |
1025662692 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1025662692, | |
1029856996 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1029856996, | |
1029861092 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1029861092, | |
1029865188 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1029865188, | |
1029869284 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1029869284, | |
1029873380 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1029873380, | |
1029889764 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1029889764, | |
1046666980 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1046666980, | |
1046671076 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.23.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1046671076, | |
1063448292 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1063448292, | |
1063460580 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1063460580, | |
1076043492 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1076043492, | |
1076047588 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1076047588, | |
1080241892 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1080241892, | |
1080245988 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1080245988, | |
1080250084 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1080250084, | |
1080254180 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1080254180, | |
1080258276 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1080258276, | |
1080274660 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1080274660, | |
1097051876 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1097051876, | |
1097055972 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.3.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1097055972, | |
1113833188 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1113833188, | |
1113845476 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1113845476, | |
1126428388 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1126428388, | |
1126432484 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1126432484, | |
1130626788 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1130626788, | |
1130630884 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1130630884, | |
1130634980 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1130634980, | |
1130639076 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1130639076, | |
1130643172 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1130643172, | |
1130659556 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1130659556, | |
1147436772 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1147436772, | |
1147440868 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.4.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1147440868, | |
1164218084 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1164218084, | |
1164230372 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1164230372, | |
1176813284 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1176813284, | |
1176817380 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1176817380, | |
1181011684 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1181011684, | |
1181015780 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1181015780, | |
1181019876 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1181019876, | |
1181023972 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1181023972, | |
1181028068 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1181028068, | |
1181044452 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1181044452, | |
1197821668 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1197821668, | |
1197825764 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.5.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1197825764, | |
1214602980 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1214602980, | |
1214615268 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1214615268, | |
1227198180 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1227198180, | |
1227202276 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1227202276, | |
1231396580 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1231396580, | |
1231400676 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1231400676, | |
1231404772 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1231404772, | |
1231408868 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1231408868, | |
1231412964 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1231412964, | |
1231429348 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1231429348, | |
1248206564 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1248206564, | |
1248210660 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.6.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1248210660, | |
1264987876 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1264987876, | |
1265000164 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1265000164, | |
1277583076 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1277583076, | |
1277587172 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1277587172, | |
1281781476 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1281781476, | |
1281785572 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1281785572, | |
1281789668 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1281789668, | |
1281793764 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1281793764, | |
1281797860 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1281797860, | |
1281814244 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1281814244, | |
1298591460 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1298591460, | |
1298595556 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.7.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1298595556, | |
1315372772 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1315372772, | |
1315385060 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1315385060, | |
1327967972 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1327967972, | |
1327972068 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1327972068, | |
1332166372 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1332166372, | |
1332170468 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1332170468, | |
1332174564 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1332174564, | |
1332178660 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1332178660, | |
1332182756 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1332182756, | |
1332199140 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1332199140, | |
1348976356 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1348976356, | |
1348980452 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.8.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1348980452, | |
1365757668 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.attn.in_proj_bias": { | |
"dtype": "F32", | |
"shape": [ | |
3072 | |
], | |
"data_offsets": [ | |
1365757668, | |
1365769956 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.attn.in_proj_weight": { | |
"dtype": "F32", | |
"shape": [ | |
3072, | |
1024 | |
], | |
"data_offsets": [ | |
1365769956, | |
1378352868 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.attn.out_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1378352868, | |
1378356964 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.attn.out_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
1024 | |
], | |
"data_offsets": [ | |
1378356964, | |
1382551268 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.ln_1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1382551268, | |
1382555364 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.ln_1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1382555364, | |
1382559460 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.ln_2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1382559460, | |
1382563556 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.ln_2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1382563556, | |
1382567652 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.mlp.c_fc.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4096 | |
], | |
"data_offsets": [ | |
1382567652, | |
1382584036 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.mlp.c_fc.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4096, | |
1024 | |
], | |
"data_offsets": [ | |
1382584036, | |
1399361252 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.mlp.c_proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1024 | |
], | |
"data_offsets": [ | |
1399361252, | |
1399365348 | |
] | |
}, | |
"cond_stage_model.model.transformer.resblocks.9.mlp.c_proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1024, | |
4096 | |
], | |
"data_offsets": [ | |
1399365348, | |
1416142564 | |
] | |
}, | |
"first_stage_model.decoder.conv_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1416142564, | |
1416144612 | |
] | |
}, | |
"first_stage_model.decoder.conv_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
4, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1416144612, | |
1416218340 | |
] | |
}, | |
"first_stage_model.decoder.conv_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
3 | |
], | |
"data_offsets": [ | |
1416218340, | |
1416218352 | |
] | |
}, | |
"first_stage_model.decoder.conv_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
3, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1416218352, | |
1416232176 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.k.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1416232176, | |
1416234224 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1416234224, | |
1417282800 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1417282800, | |
1417284848 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1417284848, | |
1417286896 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1417286896, | |
1417288944 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1417288944, | |
1418337520 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.q.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1418337520, | |
1418339568 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1418339568, | |
1419388144 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.v.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1419388144, | |
1419390192 | |
] | |
}, | |
"first_stage_model.decoder.mid.attn_1.v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1419390192, | |
1420438768 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1420438768, | |
1420440816 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1420440816, | |
1429878000 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1429878000, | |
1429880048 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1429880048, | |
1439317232 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1439317232, | |
1439319280 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1439319280, | |
1439321328 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1439321328, | |
1439323376 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1439323376, | |
1439325424 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1439325424, | |
1439327472 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1439327472, | |
1448764656 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1448764656, | |
1448766704 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1448766704, | |
1458203888 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1458203888, | |
1458205936 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1458205936, | |
1458207984 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1458207984, | |
1458210032 | |
] | |
}, | |
"first_stage_model.decoder.mid.block_2.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1458210032, | |
1458212080 | |
] | |
}, | |
"first_stage_model.decoder.norm_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1458212080, | |
1458212592 | |
] | |
}, | |
"first_stage_model.decoder.norm_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1458212592, | |
1458213104 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1458213104, | |
1458213616 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1458213616, | |
1459393264 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1459393264, | |
1459393776 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1459393776, | |
1459983600 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.nin_shortcut.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1459983600, | |
1459984112 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.nin_shortcut.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
256, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1459984112, | |
1460115184 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1460115184, | |
1460116208 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1460116208, | |
1460117232 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1460117232, | |
1460117744 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1460117744, | |
1460118256 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1460118256, | |
1460118768 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1460118768, | |
1460708592 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1460708592, | |
1460709104 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1460709104, | |
1461298928 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1461298928, | |
1461299440 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1461299440, | |
1461299952 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1461299952, | |
1461300464 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1461300464, | |
1461300976 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1461300976, | |
1461301488 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1461301488, | |
1461891312 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1461891312, | |
1461891824 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1461891824, | |
1462481648 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1462481648, | |
1462482160 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1462482160, | |
1462482672 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1462482672, | |
1462483184 | |
] | |
}, | |
"first_stage_model.decoder.up.0.block.2.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1462483184, | |
1462483696 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1462483696, | |
1462484720 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1462484720, | |
1467203312 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1467203312, | |
1467204336 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1467204336, | |
1469563632 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.nin_shortcut.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1469563632, | |
1469564656 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.nin_shortcut.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1469564656, | |
1470088944 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1470088944, | |
1470090992 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1470090992, | |
1470093040 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1470093040, | |
1470094064 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1470094064, | |
1470095088 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1470095088, | |
1470096112 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1470096112, | |
1472455408 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1472455408, | |
1472456432 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1472456432, | |
1474815728 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1474815728, | |
1474816752 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1474816752, | |
1474817776 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1474817776, | |
1474818800 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1474818800, | |
1474819824 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1474819824, | |
1474820848 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1474820848, | |
1477180144 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1477180144, | |
1477181168 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1477181168, | |
1479540464 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1479540464, | |
1479541488 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1479541488, | |
1479542512 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1479542512, | |
1479543536 | |
] | |
}, | |
"first_stage_model.decoder.up.1.block.2.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1479543536, | |
1479544560 | |
] | |
}, | |
"first_stage_model.decoder.up.1.upsample.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1479544560, | |
1479545584 | |
] | |
}, | |
"first_stage_model.decoder.up.1.upsample.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1479545584, | |
1481904880 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1481904880, | |
1481906928 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1481906928, | |
1491344112 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1491344112, | |
1491346160 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1491346160, | |
1500783344 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1500783344, | |
1500785392 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1500785392, | |
1500787440 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1500787440, | |
1500789488 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1500789488, | |
1500791536 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1500791536, | |
1500793584 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1500793584, | |
1510230768 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1510230768, | |
1510232816 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1510232816, | |
1519670000 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1519670000, | |
1519672048 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1519672048, | |
1519674096 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1519674096, | |
1519676144 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1519676144, | |
1519678192 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1519678192, | |
1519680240 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1519680240, | |
1529117424 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1529117424, | |
1529119472 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1529119472, | |
1538556656 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1538556656, | |
1538558704 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1538558704, | |
1538560752 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1538560752, | |
1538562800 | |
] | |
}, | |
"first_stage_model.decoder.up.2.block.2.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1538562800, | |
1538564848 | |
] | |
}, | |
"first_stage_model.decoder.up.2.upsample.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1538564848, | |
1538566896 | |
] | |
}, | |
"first_stage_model.decoder.up.2.upsample.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1538566896, | |
1548004080 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1548004080, | |
1548006128 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1548006128, | |
1557443312 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1557443312, | |
1557445360 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1557445360, | |
1566882544 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1566882544, | |
1566884592 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1566884592, | |
1566886640 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1566886640, | |
1566888688 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1566888688, | |
1566890736 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1566890736, | |
1566892784 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1566892784, | |
1576329968 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1576329968, | |
1576332016 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1576332016, | |
1585769200 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1585769200, | |
1585771248 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1585771248, | |
1585773296 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1585773296, | |
1585775344 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1585775344, | |
1585777392 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1585777392, | |
1585779440 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1585779440, | |
1595216624 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1595216624, | |
1595218672 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1595218672, | |
1604655856 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1604655856, | |
1604657904 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1604657904, | |
1604659952 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1604659952, | |
1604662000 | |
] | |
}, | |
"first_stage_model.decoder.up.3.block.2.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1604662000, | |
1604664048 | |
] | |
}, | |
"first_stage_model.decoder.up.3.upsample.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1604664048, | |
1604666096 | |
] | |
}, | |
"first_stage_model.decoder.up.3.upsample.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1604666096, | |
1614103280 | |
] | |
}, | |
"first_stage_model.encoder.conv_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1614103280, | |
1614103792 | |
] | |
}, | |
"first_stage_model.encoder.conv_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
3, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1614103792, | |
1614117616 | |
] | |
}, | |
"first_stage_model.encoder.conv_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
8 | |
], | |
"data_offsets": [ | |
1614117616, | |
1614117648 | |
] | |
}, | |
"first_stage_model.encoder.conv_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
8, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1614117648, | |
1614265104 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1614265104, | |
1614265616 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1614265616, | |
1614855440 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1614855440, | |
1614855952 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1614855952, | |
1615445776 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1615445776, | |
1615446288 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1615446288, | |
1615446800 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1615446800, | |
1615447312 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1615447312, | |
1615447824 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1615447824, | |
1615448336 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1615448336, | |
1616038160 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1616038160, | |
1616038672 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1616038672, | |
1616628496 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1616628496, | |
1616629008 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1616629008, | |
1616629520 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1616629520, | |
1616630032 | |
] | |
}, | |
"first_stage_model.encoder.down.0.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1616630032, | |
1616630544 | |
] | |
}, | |
"first_stage_model.encoder.down.0.downsample.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1616630544, | |
1616631056 | |
] | |
}, | |
"first_stage_model.encoder.down.0.downsample.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1616631056, | |
1617220880 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1617220880, | |
1617221904 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
128, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1617221904, | |
1618401552 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1618401552, | |
1618402576 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1618402576, | |
1620761872 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.nin_shortcut.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1620761872, | |
1620762896 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.nin_shortcut.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
128, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1620762896, | |
1620893968 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1620893968, | |
1620894480 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
128 | |
], | |
"data_offsets": [ | |
1620894480, | |
1620894992 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1620894992, | |
1620896016 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1620896016, | |
1620897040 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1620897040, | |
1620898064 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1620898064, | |
1623257360 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1623257360, | |
1623258384 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1623258384, | |
1625617680 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1625617680, | |
1625618704 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1625618704, | |
1625619728 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1625619728, | |
1625620752 | |
] | |
}, | |
"first_stage_model.encoder.down.1.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1625620752, | |
1625621776 | |
] | |
}, | |
"first_stage_model.encoder.down.1.downsample.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1625621776, | |
1625622800 | |
] | |
}, | |
"first_stage_model.encoder.down.1.downsample.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1625622800, | |
1627982096 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1627982096, | |
1627984144 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
256, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1627984144, | |
1632702736 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1632702736, | |
1632704784 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1632704784, | |
1642141968 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.nin_shortcut.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1642141968, | |
1642144016 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.nin_shortcut.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
256, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1642144016, | |
1642668304 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1642668304, | |
1642669328 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
256 | |
], | |
"data_offsets": [ | |
1642669328, | |
1642670352 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1642670352, | |
1642672400 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1642672400, | |
1642674448 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1642674448, | |
1642676496 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1642676496, | |
1652113680 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1652113680, | |
1652115728 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1652115728, | |
1661552912 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1661552912, | |
1661554960 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1661554960, | |
1661557008 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1661557008, | |
1661559056 | |
] | |
}, | |
"first_stage_model.encoder.down.2.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1661559056, | |
1661561104 | |
] | |
}, | |
"first_stage_model.encoder.down.2.downsample.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1661561104, | |
1661563152 | |
] | |
}, | |
"first_stage_model.encoder.down.2.downsample.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1661563152, | |
1671000336 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1671000336, | |
1671002384 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1671002384, | |
1680439568 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1680439568, | |
1680441616 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1680441616, | |
1689878800 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1689878800, | |
1689880848 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1689880848, | |
1689882896 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1689882896, | |
1689884944 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1689884944, | |
1689886992 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1689886992, | |
1689889040 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1689889040, | |
1699326224 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1699326224, | |
1699328272 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1699328272, | |
1708765456 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1708765456, | |
1708767504 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1708767504, | |
1708769552 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1708769552, | |
1708771600 | |
] | |
}, | |
"first_stage_model.encoder.down.3.block.1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1708771600, | |
1708773648 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.k.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1708773648, | |
1708775696 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1708775696, | |
1709824272 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1709824272, | |
1709826320 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1709826320, | |
1709828368 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1709828368, | |
1709830416 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1709830416, | |
1710878992 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.q.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1710878992, | |
1710881040 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1710881040, | |
1711929616 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.v.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1711929616, | |
1711931664 | |
] | |
}, | |
"first_stage_model.encoder.mid.attn_1.v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1711931664, | |
1712980240 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1712980240, | |
1712982288 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1712982288, | |
1722419472 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1722419472, | |
1722421520 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1722421520, | |
1731858704 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1731858704, | |
1731860752 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1731860752, | |
1731862800 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1731862800, | |
1731864848 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_1.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1731864848, | |
1731866896 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.conv1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1731866896, | |
1731868944 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.conv1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1731868944, | |
1741306128 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.conv2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1741306128, | |
1741308176 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.conv2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512, | |
512, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1741308176, | |
1750745360 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1750745360, | |
1750747408 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1750747408, | |
1750749456 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1750749456, | |
1750751504 | |
] | |
}, | |
"first_stage_model.encoder.mid.block_2.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1750751504, | |
1750753552 | |
] | |
}, | |
"first_stage_model.encoder.norm_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1750753552, | |
1750755600 | |
] | |
}, | |
"first_stage_model.encoder.norm_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
512 | |
], | |
"data_offsets": [ | |
1750755600, | |
1750757648 | |
] | |
}, | |
"first_stage_model.post_quant_conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4 | |
], | |
"data_offsets": [ | |
1750757648, | |
1750757664 | |
] | |
}, | |
"first_stage_model.post_quant_conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4, | |
4, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1750757664, | |
1750757728 | |
] | |
}, | |
"first_stage_model.quant_conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
8 | |
], | |
"data_offsets": [ | |
1750757728, | |
1750757760 | |
] | |
}, | |
"first_stage_model.quant_conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
8, | |
8, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
1750757760, | |
1750758016 | |
] | |
}, | |
"log_one_minus_alphas_cumprod": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
1750758016, | |
1750762016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.0.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1750762016, | |
1750763296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.0.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
4, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1750763296, | |
1750809376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1750809376, | |
1750810656 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
1750810656, | |
1752449056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1752449056, | |
1752450336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1752450336, | |
1752451616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1752451616, | |
1752452896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1752452896, | |
1756139296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1756139296, | |
1756140576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1756140576, | |
1756141856 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1756141856, | |
1756143136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1756143136, | |
1759829536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1759829536, | |
1759830816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1759830816, | |
1759832096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1759832096, | |
1759833376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1759833376, | |
1760242976 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1760242976, | |
1760244256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1760244256, | |
1760653856 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1760653856, | |
1761063456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1761063456, | |
1761064736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1761064736, | |
1761474336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1761474336, | |
1761883936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1761883936, | |
1762293536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
1762293536, | |
1763604256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1763604256, | |
1763605536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1763605536, | |
1764015136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
1764015136, | |
1764424736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
1764424736, | |
1765735456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
1765735456, | |
1765745696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560, | |
320 | |
], | |
"data_offsets": [ | |
1765745696, | |
1769022496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1769022496, | |
1769023776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
1769023776, | |
1770662176 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1770662176, | |
1770663456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1770663456, | |
1770664736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1770664736, | |
1770666016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1770666016, | |
1770667296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1770667296, | |
1770668576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
1770668576, | |
1770669856 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1770669856, | |
1770674976 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
1770674976, | |
1777228576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1777228576, | |
1777233696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1777233696, | |
1777238816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1777238816, | |
1777243936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1777243936, | |
1836226336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1836226336, | |
1836231456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1836231456, | |
1836236576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1836236576, | |
1836241696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.10.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1836241696, | |
1895224096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1895224096, | |
1895229216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
1895229216, | |
1901782816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1901782816, | |
1901787936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1901787936, | |
1901793056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1901793056, | |
1901798176 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1901798176, | |
1960780576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1960780576, | |
1960785696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1960785696, | |
1960790816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
1960790816, | |
1960795936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.11.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
1960795936, | |
2019778336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2019778336, | |
2019779616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
2019779616, | |
2021418016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2021418016, | |
2021419296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2021419296, | |
2021420576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2021420576, | |
2021421856 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2021421856, | |
2025108256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2025108256, | |
2025109536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2025109536, | |
2025110816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2025110816, | |
2025112096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2025112096, | |
2028798496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2028798496, | |
2028799776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2028799776, | |
2028801056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2028801056, | |
2028802336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2028802336, | |
2029211936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2029211936, | |
2029213216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2029213216, | |
2029622816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2029622816, | |
2030032416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2030032416, | |
2030033696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2030033696, | |
2030443296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2030443296, | |
2030852896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2030852896, | |
2031262496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
2031262496, | |
2032573216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2032573216, | |
2032574496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2032574496, | |
2032984096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
2032984096, | |
2033393696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
2033393696, | |
2034704416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
2034704416, | |
2034714656 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560, | |
320 | |
], | |
"data_offsets": [ | |
2034714656, | |
2037991456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2037991456, | |
2037992736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
2037992736, | |
2039631136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039631136, | |
2039632416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039632416, | |
2039633696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039633696, | |
2039634976 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039634976, | |
2039636256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039636256, | |
2039637536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039637536, | |
2039638816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.3.0.op.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2039638816, | |
2039640096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.3.0.op.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2039640096, | |
2043326496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2043326496, | |
2043329056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280 | |
], | |
"data_offsets": [ | |
2043329056, | |
2046605856 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2046605856, | |
2046607136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
2046607136, | |
2046608416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2046608416, | |
2046610976 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2046610976, | |
2053983776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2053983776, | |
2053986336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2053986336, | |
2053988896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2053988896, | |
2053991456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2053991456, | |
2068737056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2068737056, | |
2068739616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
320, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
2068739616, | |
2069558816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2069558816, | |
2069561376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2069561376, | |
2069563936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2069563936, | |
2069566496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2069566496, | |
2071204896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2071204896, | |
2071207456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2071207456, | |
2072845856 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2072845856, | |
2074484256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2074484256, | |
2074486816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2074486816, | |
2076125216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2076125216, | |
2077763616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2077763616, | |
2079402016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
2079402016, | |
2082023456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2082023456, | |
2082026016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2082026016, | |
2083664416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2083664416, | |
2085302816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
2085302816, | |
2087924256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
5120 | |
], | |
"data_offsets": [ | |
2087924256, | |
2087944736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
5120, | |
640 | |
], | |
"data_offsets": [ | |
2087944736, | |
2101051936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2101051936, | |
2101054496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
2560 | |
], | |
"data_offsets": [ | |
2101054496, | |
2107608096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107608096, | |
2107610656 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107610656, | |
2107613216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107613216, | |
2107615776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107615776, | |
2107618336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107618336, | |
2107620896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107620896, | |
2107623456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2107623456, | |
2107626016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280 | |
], | |
"data_offsets": [ | |
2107626016, | |
2110902816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2110902816, | |
2110905376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2110905376, | |
2110907936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2110907936, | |
2110910496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2110910496, | |
2125656096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2125656096, | |
2125658656 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2125658656, | |
2125661216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2125661216, | |
2125663776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2125663776, | |
2140409376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2140409376, | |
2140411936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2140411936, | |
2140414496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2140414496, | |
2140417056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2140417056, | |
2142055456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2142055456, | |
2142058016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2142058016, | |
2143696416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2143696416, | |
2145334816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2145334816, | |
2145337376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2145337376, | |
2146975776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2146975776, | |
2148614176 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2148614176, | |
2150252576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
2150252576, | |
2152874016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2152874016, | |
2152876576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2152876576, | |
2154514976 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
2154514976, | |
2156153376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
2156153376, | |
2158774816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
5120 | |
], | |
"data_offsets": [ | |
2158774816, | |
2158795296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
5120, | |
640 | |
], | |
"data_offsets": [ | |
2158795296, | |
2171902496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2171902496, | |
2171905056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
2560 | |
], | |
"data_offsets": [ | |
2171905056, | |
2178458656 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178458656, | |
2178461216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178461216, | |
2178463776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178463776, | |
2178466336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178466336, | |
2178468896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178468896, | |
2178471456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178471456, | |
2178474016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.6.0.op.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2178474016, | |
2178476576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.6.0.op.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2178476576, | |
2193222176 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2193222176, | |
2193227296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2193227296, | |
2199780896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2199780896, | |
2199783456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
2199783456, | |
2199786016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2199786016, | |
2199791136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2199791136, | |
2229282336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2229282336, | |
2229287456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2229287456, | |
2229292576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2229292576, | |
2229297696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2229297696, | |
2288280096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2288280096, | |
2288285216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
640, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
2288285216, | |
2291562016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2291562016, | |
2291567136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2291567136, | |
2291572256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2291572256, | |
2291577376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2291577376, | |
2298130976 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2298130976, | |
2298136096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2298136096, | |
2304689696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2304689696, | |
2311243296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2311243296, | |
2311248416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2311248416, | |
2317802016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2317802016, | |
2324355616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2324355616, | |
2330909216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
2330909216, | |
2336152096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2336152096, | |
2336157216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2336157216, | |
2342710816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2342710816, | |
2349264416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
2349264416, | |
2354507296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
10240 | |
], | |
"data_offsets": [ | |
2354507296, | |
2354548256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
10240, | |
1280 | |
], | |
"data_offsets": [ | |
2354548256, | |
2406977056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2406977056, | |
2406982176 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
5120 | |
], | |
"data_offsets": [ | |
2406982176, | |
2433196576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433196576, | |
2433201696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433201696, | |
2433206816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433206816, | |
2433211936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433211936, | |
2433217056 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433217056, | |
2433222176 | |
] | |
}, | |
"model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433222176, | |
2433227296 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2433227296, | |
2433232416 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2433232416, | |
2439786016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2439786016, | |
2439791136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2439791136, | |
2439796256 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2439796256, | |
2439801376 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2439801376, | |
2498783776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2498783776, | |
2498788896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2498788896, | |
2498794016 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2498794016, | |
2498799136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2498799136, | |
2557781536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2557781536, | |
2557786656 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2557786656, | |
2557791776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2557791776, | |
2557796896 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2557796896, | |
2564350496 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2564350496, | |
2564355616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2564355616, | |
2570909216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2570909216, | |
2577462816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2577462816, | |
2577467936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2577467936, | |
2584021536 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2584021536, | |
2590575136 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2590575136, | |
2597128736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
2597128736, | |
2602371616 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2602371616, | |
2602376736 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2602376736, | |
2608930336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2608930336, | |
2615483936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
2615483936, | |
2620726816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
10240 | |
], | |
"data_offsets": [ | |
2620726816, | |
2620767776 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
10240, | |
1280 | |
], | |
"data_offsets": [ | |
2620767776, | |
2673196576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2673196576, | |
2673201696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
5120 | |
], | |
"data_offsets": [ | |
2673201696, | |
2699416096 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699416096, | |
2699421216 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699421216, | |
2699426336 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699426336, | |
2699431456 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699431456, | |
2699436576 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699436576, | |
2699441696 | |
] | |
}, | |
"model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699441696, | |
2699446816 | |
] | |
}, | |
"model.diffusion_model.input_blocks.9.0.op.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2699446816, | |
2699451936 | |
] | |
}, | |
"model.diffusion_model.input_blocks.9.0.op.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2699451936, | |
2758434336 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2758434336, | |
2758439456 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2758439456, | |
2764993056 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2764993056, | |
2764998176 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2764998176, | |
2765003296 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2765003296, | |
2765008416 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2765008416, | |
2823990816 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2823990816, | |
2823995936 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2823995936, | |
2824001056 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2824001056, | |
2824006176 | |
] | |
}, | |
"model.diffusion_model.middle_block.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
2824006176, | |
2882988576 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2882988576, | |
2882993696 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2882993696, | |
2882998816 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2882998816, | |
2883003936 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2883003936, | |
2889557536 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2889557536, | |
2889562656 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2889562656, | |
2896116256 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2896116256, | |
2902669856 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2902669856, | |
2902674976 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2902674976, | |
2909228576 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2909228576, | |
2915782176 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2915782176, | |
2922335776 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
2922335776, | |
2927578656 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2927578656, | |
2927583776 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2927583776, | |
2934137376 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
2934137376, | |
2940690976 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
2940690976, | |
2945933856 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
10240 | |
], | |
"data_offsets": [ | |
2945933856, | |
2945974816 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
10240, | |
1280 | |
], | |
"data_offsets": [ | |
2945974816, | |
2998403616 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
2998403616, | |
2998408736 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
5120 | |
], | |
"data_offsets": [ | |
2998408736, | |
3024623136 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024623136, | |
3024628256 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024628256, | |
3024633376 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024633376, | |
3024638496 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024638496, | |
3024643616 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024643616, | |
3024648736 | |
] | |
}, | |
"model.diffusion_model.middle_block.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024648736, | |
3024653856 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3024653856, | |
3024658976 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
3024658976, | |
3031212576 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3031212576, | |
3031217696 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3031217696, | |
3031222816 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3031222816, | |
3031227936 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3031227936, | |
3090210336 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3090210336, | |
3090215456 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3090215456, | |
3090220576 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3090220576, | |
3090225696 | |
] | |
}, | |
"model.diffusion_model.middle_block.2.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3090225696, | |
3149208096 | |
] | |
}, | |
"model.diffusion_model.out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3149208096, | |
3149209376 | |
] | |
}, | |
"model.diffusion_model.out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3149209376, | |
3149210656 | |
] | |
}, | |
"model.diffusion_model.out.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
4 | |
], | |
"data_offsets": [ | |
3149210656, | |
3149210672 | |
] | |
}, | |
"model.diffusion_model.out.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
4, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3149210672, | |
3149256752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3149256752, | |
3149261872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
3149261872, | |
3155815472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3155815472, | |
3155825712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3155825712, | |
3155835952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3155835952, | |
3155841072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3155841072, | |
3273805872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3273805872, | |
3273810992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3273810992, | |
3273816112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3273816112, | |
3273821232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3273821232, | |
3332803632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3332803632, | |
3332808752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.0.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
3332808752, | |
3345915952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3345915952, | |
3345921072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
3345921072, | |
3352474672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3352474672, | |
3352484912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3352484912, | |
3352495152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3352495152, | |
3352500272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3352500272, | |
3470465072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3470465072, | |
3470470192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3470470192, | |
3470475312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3470475312, | |
3470480432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3470480432, | |
3529462832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3529462832, | |
3529467952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.1.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
3529467952, | |
3542575152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3542575152, | |
3542576432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
3542576432, | |
3544214832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
3544214832, | |
3544217392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
3544217392, | |
3544219952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3544219952, | |
3544221232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3544221232, | |
3551594032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3551594032, | |
3551595312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3551595312, | |
3551596592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3551596592, | |
3551597872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3551597872, | |
3555284272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3555284272, | |
3555285552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
640, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
3555285552, | |
3556104752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3556104752, | |
3556106032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3556106032, | |
3556107312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3556107312, | |
3556108592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3556108592, | |
3556518192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3556518192, | |
3556519472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3556519472, | |
3556929072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3556929072, | |
3557338672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3557338672, | |
3557339952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3557339952, | |
3557749552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3557749552, | |
3558159152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3558159152, | |
3558568752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
3558568752, | |
3559879472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3559879472, | |
3559880752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3559880752, | |
3560290352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3560290352, | |
3560699952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
3560699952, | |
3562010672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3562010672, | |
3562020912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560, | |
320 | |
], | |
"data_offsets": [ | |
3562020912, | |
3565297712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3565297712, | |
3565298992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
3565298992, | |
3566937392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566937392, | |
3566938672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566938672, | |
3566939952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566939952, | |
3566941232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566941232, | |
3566942512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566942512, | |
3566943792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566943792, | |
3566945072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3566945072, | |
3566946352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
3566946352, | |
3568584752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
3568584752, | |
3568587312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
3568587312, | |
3568589872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3568589872, | |
3568591152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3568591152, | |
3575963952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3575963952, | |
3575965232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3575965232, | |
3575966512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3575966512, | |
3575967792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3575967792, | |
3579654192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3579654192, | |
3579655472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
640, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
3579655472, | |
3580474672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3580474672, | |
3580475952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3580475952, | |
3580477232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3580477232, | |
3580478512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3580478512, | |
3580888112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3580888112, | |
3580889392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3580889392, | |
3581298992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3581298992, | |
3581708592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3581708592, | |
3581709872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3581709872, | |
3582119472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3582119472, | |
3582529072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3582529072, | |
3582938672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
3582938672, | |
3584249392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3584249392, | |
3584250672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3584250672, | |
3584660272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
3584660272, | |
3585069872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
3585069872, | |
3586380592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3586380592, | |
3586390832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560, | |
320 | |
], | |
"data_offsets": [ | |
3586390832, | |
3589667632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3589667632, | |
3589668912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
3589668912, | |
3591307312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3591307312, | |
3591308592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3591308592, | |
3591309872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3591309872, | |
3591311152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3591311152, | |
3591312432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3591312432, | |
3591313712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
3591313712, | |
3591314992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3591314992, | |
3591320112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
3591320112, | |
3597873712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3597873712, | |
3597883952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3597883952, | |
3597894192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3597894192, | |
3597899312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3597899312, | |
3715864112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3715864112, | |
3715869232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3715869232, | |
3715874352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3715874352, | |
3715879472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3715879472, | |
3774861872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3774861872, | |
3774866992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
3774866992, | |
3787974192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.1.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3787974192, | |
3787979312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.2.1.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3787979312, | |
3846961712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3846961712, | |
3846966832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
3846966832, | |
3853520432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3853520432, | |
3853530672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
3853530672, | |
3853540912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3853540912, | |
3853546032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3853546032, | |
3971510832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3971510832, | |
3971515952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3971515952, | |
3971521072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
3971521072, | |
3971526192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
3971526192, | |
4030508592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4030508592, | |
4030513712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
4030513712, | |
4043620912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4043620912, | |
4043626032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4043626032, | |
4043631152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4043631152, | |
4043636272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4043636272, | |
4050189872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4050189872, | |
4050194992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4050194992, | |
4056748592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4056748592, | |
4063302192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4063302192, | |
4063307312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4063307312, | |
4069860912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4069860912, | |
4076414512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4076414512, | |
4082968112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
4082968112, | |
4088210992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4088210992, | |
4088216112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4088216112, | |
4094769712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4094769712, | |
4101323312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
4101323312, | |
4106566192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
10240 | |
], | |
"data_offsets": [ | |
4106566192, | |
4106607152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
10240, | |
1280 | |
], | |
"data_offsets": [ | |
4106607152, | |
4159035952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4159035952, | |
4159041072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
5120 | |
], | |
"data_offsets": [ | |
4159041072, | |
4185255472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185255472, | |
4185260592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185260592, | |
4185265712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185265712, | |
4185270832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185270832, | |
4185275952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185275952, | |
4185281072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185281072, | |
4185286192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4185286192, | |
4185291312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4185291312, | |
4191844912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
4191844912, | |
4191855152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
4191855152, | |
4191865392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4191865392, | |
4191870512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4191870512, | |
4309835312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4309835312, | |
4309840432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4309840432, | |
4309845552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4309845552, | |
4309850672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4309850672, | |
4368833072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4368833072, | |
4368838192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
2560, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
4368838192, | |
4381945392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4381945392, | |
4381950512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4381950512, | |
4381955632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4381955632, | |
4381960752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4381960752, | |
4388514352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4388514352, | |
4388519472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4388519472, | |
4395073072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4395073072, | |
4401626672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4401626672, | |
4401631792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4401631792, | |
4408185392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4408185392, | |
4414738992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4414738992, | |
4421292592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
4421292592, | |
4426535472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4426535472, | |
4426540592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4426540592, | |
4433094192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4433094192, | |
4439647792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
4439647792, | |
4444890672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
10240 | |
], | |
"data_offsets": [ | |
4444890672, | |
4444931632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
10240, | |
1280 | |
], | |
"data_offsets": [ | |
4444931632, | |
4497360432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4497360432, | |
4497365552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
5120 | |
], | |
"data_offsets": [ | |
4497365552, | |
4523579952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523579952, | |
4523585072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523585072, | |
4523590192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523590192, | |
4523595312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523595312, | |
4523600432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523600432, | |
4523605552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523605552, | |
4523610672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4523610672, | |
4523615792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4523615792, | |
4530169392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1920 | |
], | |
"data_offsets": [ | |
4530169392, | |
4530177072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1920 | |
], | |
"data_offsets": [ | |
4530177072, | |
4530184752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4530184752, | |
4530189872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1920, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4530189872, | |
4618663472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4618663472, | |
4618668592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4618668592, | |
4618673712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4618673712, | |
4618678832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4618678832, | |
4677661232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4677661232, | |
4677666352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1920, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
4677666352, | |
4687496752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4687496752, | |
4687501872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4687501872, | |
4687506992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4687506992, | |
4687512112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4687512112, | |
4694065712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4694065712, | |
4694070832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4694070832, | |
4700624432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4700624432, | |
4707178032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4707178032, | |
4707183152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4707183152, | |
4713736752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4713736752, | |
4720290352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4720290352, | |
4726843952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
4726843952, | |
4732086832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4732086832, | |
4732091952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4732091952, | |
4738645552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
4738645552, | |
4745199152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1024 | |
], | |
"data_offsets": [ | |
4745199152, | |
4750442032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
10240 | |
], | |
"data_offsets": [ | |
4750442032, | |
4750482992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
10240, | |
1280 | |
], | |
"data_offsets": [ | |
4750482992, | |
4802911792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4802911792, | |
4802916912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
5120 | |
], | |
"data_offsets": [ | |
4802916912, | |
4829131312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829131312, | |
4829136432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829136432, | |
4829141552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829141552, | |
4829146672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829146672, | |
4829151792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829151792, | |
4829156912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829156912, | |
4829162032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.2.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4829162032, | |
4829167152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.5.2.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4829167152, | |
4888149552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4888149552, | |
4888152112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280 | |
], | |
"data_offsets": [ | |
4888152112, | |
4891428912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1920 | |
], | |
"data_offsets": [ | |
4891428912, | |
4891436592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1920 | |
], | |
"data_offsets": [ | |
4891436592, | |
4891444272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4891444272, | |
4891446832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1920, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4891446832, | |
4935683632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4935683632, | |
4935686192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4935686192, | |
4935688752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4935688752, | |
4935691312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4935691312, | |
4950436912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4950436912, | |
4950439472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1920, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
4950439472, | |
4955354672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4955354672, | |
4955357232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4955357232, | |
4955359792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4955359792, | |
4955362352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4955362352, | |
4957000752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4957000752, | |
4957003312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4957003312, | |
4958641712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4958641712, | |
4960280112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4960280112, | |
4960282672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4960282672, | |
4961921072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4961921072, | |
4963559472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4963559472, | |
4965197872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
4965197872, | |
4967819312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4967819312, | |
4967821872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4967821872, | |
4969460272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
4969460272, | |
4971098672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
4971098672, | |
4973720112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
5120 | |
], | |
"data_offsets": [ | |
4973720112, | |
4973740592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
5120, | |
640 | |
], | |
"data_offsets": [ | |
4973740592, | |
4986847792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4986847792, | |
4986850352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
2560 | |
], | |
"data_offsets": [ | |
4986850352, | |
4993403952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993403952, | |
4993406512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993406512, | |
4993409072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993409072, | |
4993411632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993411632, | |
4993414192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993414192, | |
4993416752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993416752, | |
4993419312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4993419312, | |
4993421872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280 | |
], | |
"data_offsets": [ | |
4993421872, | |
4996698672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4996698672, | |
4996703792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
4996703792, | |
4996708912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
4996708912, | |
4996711472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
4996711472, | |
5026202672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5026202672, | |
5026205232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5026205232, | |
5026207792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5026207792, | |
5026210352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
5026210352, | |
5040955952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5040955952, | |
5040958512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
5040958512, | |
5044235312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5044235312, | |
5044237872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5044237872, | |
5044240432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5044240432, | |
5044242992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5044242992, | |
5045881392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5045881392, | |
5045883952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5045883952, | |
5047522352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5047522352, | |
5049160752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5049160752, | |
5049163312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5049163312, | |
5050801712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5050801712, | |
5052440112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5052440112, | |
5054078512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
5054078512, | |
5056699952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5056699952, | |
5056702512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5056702512, | |
5058340912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5058340912, | |
5059979312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
5059979312, | |
5062600752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
5120 | |
], | |
"data_offsets": [ | |
5062600752, | |
5062621232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
5120, | |
640 | |
], | |
"data_offsets": [ | |
5062621232, | |
5075728432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5075728432, | |
5075730992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
2560 | |
], | |
"data_offsets": [ | |
5075730992, | |
5082284592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082284592, | |
5082287152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082287152, | |
5082289712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082289712, | |
5082292272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082292272, | |
5082294832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082294832, | |
5082297392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082297392, | |
5082299952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5082299952, | |
5082302512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1280 | |
], | |
"data_offsets": [ | |
5082302512, | |
5085579312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
960 | |
], | |
"data_offsets": [ | |
5085579312, | |
5085583152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
960 | |
], | |
"data_offsets": [ | |
5085583152, | |
5085586992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5085586992, | |
5085589552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
960, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
5085589552, | |
5107707952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5107707952, | |
5107710512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5107710512, | |
5107713072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5107713072, | |
5107715632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
5107715632, | |
5122461232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5122461232, | |
5122463792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
960, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
5122463792, | |
5124921392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5124921392, | |
5124923952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5124923952, | |
5124926512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5124926512, | |
5124929072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5124929072, | |
5126567472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5126567472, | |
5126570032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5126570032, | |
5128208432 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5128208432, | |
5129846832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5129846832, | |
5129849392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5129849392, | |
5131487792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5131487792, | |
5133126192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5133126192, | |
5134764592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
5134764592, | |
5137386032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5137386032, | |
5137388592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5137388592, | |
5139026992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640 | |
], | |
"data_offsets": [ | |
5139026992, | |
5140665392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
1024 | |
], | |
"data_offsets": [ | |
5140665392, | |
5143286832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
5120 | |
], | |
"data_offsets": [ | |
5143286832, | |
5143307312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
5120, | |
640 | |
], | |
"data_offsets": [ | |
5143307312, | |
5156414512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5156414512, | |
5156417072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
2560 | |
], | |
"data_offsets": [ | |
5156417072, | |
5162970672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162970672, | |
5162973232 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162973232, | |
5162975792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162975792, | |
5162978352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162978352, | |
5162980912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162980912, | |
5162983472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162983472, | |
5162986032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.2.conv.bias": { | |
"dtype": "F32", | |
"shape": [ | |
640 | |
], | |
"data_offsets": [ | |
5162986032, | |
5162988592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.8.2.conv.weight": { | |
"dtype": "F32", | |
"shape": [ | |
640, | |
640, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
5162988592, | |
5177734192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.emb_layers.1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5177734192, | |
5177735472 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.emb_layers.1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
5177735472, | |
5179373872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.in_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
960 | |
], | |
"data_offsets": [ | |
5179373872, | |
5179377712 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.in_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
960 | |
], | |
"data_offsets": [ | |
5179377712, | |
5179381552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.in_layers.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5179381552, | |
5179382832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.in_layers.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
960, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
5179382832, | |
5190442032 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.out_layers.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5190442032, | |
5190443312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.out_layers.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5190443312, | |
5190444592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.out_layers.3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5190444592, | |
5190445872 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.out_layers.3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320, | |
3, | |
3 | |
], | |
"data_offsets": [ | |
5190445872, | |
5194132272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.skip_connection.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5194132272, | |
5194133552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.0.skip_connection.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
960, | |
1, | |
1 | |
], | |
"data_offsets": [ | |
5194133552, | |
5195362352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.norm.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5195362352, | |
5195363632 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.norm.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5195363632, | |
5195364912 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.proj_in.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5195364912, | |
5195366192 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.proj_in.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5195366192, | |
5195775792 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.proj_out.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5195775792, | |
5195777072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.proj_out.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5195777072, | |
5196186672 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5196186672, | |
5196596272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5196596272, | |
5196597552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5196597552, | |
5197007152 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5197007152, | |
5197416752 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5197416752, | |
5197826352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_k.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
5197826352, | |
5199137072 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_out.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5199137072, | |
5199138352 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_out.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5199138352, | |
5199547952 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_q.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
320 | |
], | |
"data_offsets": [ | |
5199547952, | |
5199957552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_v.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1024 | |
], | |
"data_offsets": [ | |
5199957552, | |
5201268272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.0.proj.bias": { | |
"dtype": "F32", | |
"shape": [ | |
2560 | |
], | |
"data_offsets": [ | |
5201268272, | |
5201278512 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.0.proj.weight": { | |
"dtype": "F32", | |
"shape": [ | |
2560, | |
320 | |
], | |
"data_offsets": [ | |
5201278512, | |
5204555312 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5204555312, | |
5204556592 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320, | |
1280 | |
], | |
"data_offsets": [ | |
5204556592, | |
5206194992 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm1.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5206194992, | |
5206196272 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm1.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5206196272, | |
5206197552 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5206197552, | |
5206198832 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5206198832, | |
5206200112 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm3.bias": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5206200112, | |
5206201392 | |
] | |
}, | |
"model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm3.weight": { | |
"dtype": "F32", | |
"shape": [ | |
320 | |
], | |
"data_offsets": [ | |
5206201392, | |
5206202672 | |
] | |
}, | |
"model.diffusion_model.time_embed.0.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
5206202672, | |
5206207792 | |
] | |
}, | |
"model.diffusion_model.time_embed.0.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
320 | |
], | |
"data_offsets": [ | |
5206207792, | |
5207846192 | |
] | |
}, | |
"model.diffusion_model.time_embed.2.bias": { | |
"dtype": "F32", | |
"shape": [ | |
1280 | |
], | |
"data_offsets": [ | |
5207846192, | |
5207851312 | |
] | |
}, | |
"model.diffusion_model.time_embed.2.weight": { | |
"dtype": "F32", | |
"shape": [ | |
1280, | |
1280 | |
], | |
"data_offsets": [ | |
5207851312, | |
5214404912 | |
] | |
}, | |
"model_ema.decay": { | |
"dtype": "F32", | |
"shape": [], | |
"data_offsets": [ | |
5214404912, | |
5214404916 | |
] | |
}, | |
"model_ema.num_updates": { | |
"dtype": "I32", | |
"shape": [], | |
"data_offsets": [ | |
5214404916, | |
5214404920 | |
] | |
}, | |
"posterior_log_variance_clipped": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214404920, | |
5214408920 | |
] | |
}, | |
"posterior_mean_coef1": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214408920, | |
5214412920 | |
] | |
}, | |
"posterior_mean_coef2": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214412920, | |
5214416920 | |
] | |
}, | |
"posterior_variance": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214416920, | |
5214420920 | |
] | |
}, | |
"sqrt_alphas_cumprod": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214420920, | |
5214424920 | |
] | |
}, | |
"sqrt_one_minus_alphas_cumprod": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214424920, | |
5214428920 | |
] | |
}, | |
"sqrt_recip_alphas_cumprod": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214428920, | |
5214432920 | |
] | |
}, | |
"sqrt_recipm1_alphas_cumprod": { | |
"dtype": "F32", | |
"shape": [ | |
1000 | |
], | |
"data_offsets": [ | |
5214432920, | |
5214436920 | |
] | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment