sdv2safetensors.json

{
  "__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
    ]
  }
}