Skip to content

Instantly share code, notes, and snippets.

@mcmonkey4eva
Created January 25, 2023 17:09
Show Gist options
  • Save mcmonkey4eva/af0b92249a00ee179fdeff15bc49b44d to your computer and use it in GitHub Desktop.
Save mcmonkey4eva/af0b92249a00ee179fdeff15bc49b44d to your computer and use it in GitHub Desktop.
{
"__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