{
 "H": 480,
 "W": 640,
 "R": 30,
 "C": 40,
 "levels": 5,
 "channels": [
  32,
  64,
  128,
  256,
  256
 ],
 "enc_channels": 1024,
 "enc_out": 256,
 "depth_resolution": 256.0,
 "weights": {
  "input_proj.w": {
   "offset": 0,
   "count": 96,
   "shape": [
    3,
    32
   ],
   "dtype": "float16"
  },
  "input_proj.b": {
   "offset": 192,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "prompt_proj.w": {
   "offset": 256,
   "count": 64,
   "shape": [
    2,
    32
   ],
   "dtype": "float16"
  },
  "prompt_proj.b": {
   "offset": 384,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "out_proj.w": {
   "offset": 448,
   "count": 32,
   "shape": [
    32,
    1
   ],
   "dtype": "float16"
  },
  "out_proj.b": {
   "offset": 512,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float16"
  },
  "down0.n1w": {
   "offset": 514,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "down0.n1b": {
   "offset": 578,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "down0.c1w8": {
   "offset": 642,
   "count": 27648,
   "shape": [
    27,
    32,
    32
   ],
   "dtype": "int8"
  },
  "down0.c1ws": {
   "offset": 28290,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float32"
  },
  "down0.c1b": {
   "offset": 28418,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "down0.c1clip": {
   "offset": 28482,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down0.c2w8": {
   "offset": 28486,
   "count": 27648,
   "shape": [
    27,
    32,
    32
   ],
   "dtype": "int8"
  },
  "down0.c2ws": {
   "offset": 56134,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float32"
  },
  "down0.c2b": {
   "offset": 56262,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "down0.c2clip": {
   "offset": 56326,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "pool0.w8": {
   "offset": 56330,
   "count": 2048,
   "shape": [
    32,
    64
   ],
   "dtype": "int8"
  },
  "pool0.ws": {
   "offset": 58378,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float32"
  },
  "pool0.b": {
   "offset": 58634,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "pool0.clip": {
   "offset": 58762,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down1.n1w": {
   "offset": 58766,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "down1.n1b": {
   "offset": 58894,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "down1.c1w8": {
   "offset": 59022,
   "count": 110592,
   "shape": [
    27,
    64,
    64
   ],
   "dtype": "int8"
  },
  "down1.c1ws": {
   "offset": 169614,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float32"
  },
  "down1.c1b": {
   "offset": 169870,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "down1.c1clip": {
   "offset": 169998,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down1.c2w8": {
   "offset": 170002,
   "count": 110592,
   "shape": [
    27,
    64,
    64
   ],
   "dtype": "int8"
  },
  "down1.c2ws": {
   "offset": 280594,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float32"
  },
  "down1.c2b": {
   "offset": 280850,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "down1.c2clip": {
   "offset": 280978,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "pool1.w8": {
   "offset": 280982,
   "count": 8192,
   "shape": [
    64,
    128
   ],
   "dtype": "int8"
  },
  "pool1.ws": {
   "offset": 289174,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float32"
  },
  "pool1.b": {
   "offset": 289686,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "pool1.clip": {
   "offset": 289942,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down2.n1w": {
   "offset": 289946,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "down2.n1b": {
   "offset": 290202,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "down2.c1w8": {
   "offset": 290458,
   "count": 442368,
   "shape": [
    27,
    128,
    128
   ],
   "dtype": "int8"
  },
  "down2.c1ws": {
   "offset": 732826,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float32"
  },
  "down2.c1b": {
   "offset": 733338,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "down2.c1clip": {
   "offset": 733594,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down2.c2w8": {
   "offset": 733598,
   "count": 442368,
   "shape": [
    27,
    128,
    128
   ],
   "dtype": "int8"
  },
  "down2.c2ws": {
   "offset": 1175966,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float32"
  },
  "down2.c2b": {
   "offset": 1176478,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "down2.c2clip": {
   "offset": 1176734,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "pool2.w8": {
   "offset": 1176738,
   "count": 32768,
   "shape": [
    128,
    256
   ],
   "dtype": "int8"
  },
  "pool2.ws": {
   "offset": 1209506,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "pool2.b": {
   "offset": 1210530,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "pool2.clip": {
   "offset": 1211042,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down3.n1w": {
   "offset": 1211046,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down3.n1b": {
   "offset": 1211558,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down3.c1w8": {
   "offset": 1212070,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "down3.c1ws": {
   "offset": 2981542,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "down3.c1b": {
   "offset": 2982566,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down3.c1clip": {
   "offset": 2983078,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down3.c2w8": {
   "offset": 2983082,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "down3.c2ws": {
   "offset": 4752554,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "down3.c2b": {
   "offset": 4753578,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down3.c2clip": {
   "offset": 4754090,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "pool3.w8": {
   "offset": 4754094,
   "count": 65536,
   "shape": [
    256,
    256
   ],
   "dtype": "int8"
  },
  "pool3.ws": {
   "offset": 4819630,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "pool3.b": {
   "offset": 4820654,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "pool3.clip": {
   "offset": 4821166,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down4.n1w": {
   "offset": 4821170,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down4.n1b": {
   "offset": 4821682,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down4.c1w8": {
   "offset": 4822194,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "down4.c1ws": {
   "offset": 6591666,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "down4.c1b": {
   "offset": 6592690,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down4.c1clip": {
   "offset": 6593202,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "down4.c2w8": {
   "offset": 6593206,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "down4.c2ws": {
   "offset": 8362678,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "down4.c2b": {
   "offset": 8363702,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "down4.c2clip": {
   "offset": 8364214,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "encoder_fuse.w8": {
   "offset": 8364218,
   "count": 262144,
   "shape": [
    1024,
    256
   ],
   "dtype": "int8"
  },
  "encoder_fuse.ws": {
   "offset": 8626362,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "encoder_fuse.b": {
   "offset": 8627386,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "encoder_fuse.clip": {
   "offset": 8627898,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "fuse0.w8": {
   "offset": 8627902,
   "count": 131072,
   "shape": [
    512,
    256
   ],
   "dtype": "int8"
  },
  "fuse0.ws": {
   "offset": 8758974,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "fuse0.b": {
   "offset": 8759998,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "fuse0.clip": {
   "offset": 8760510,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "fuse2.w8": {
   "offset": 8760514,
   "count": 65536,
   "shape": [
    256,
    256
   ],
   "dtype": "int8"
  },
  "fuse2.ws": {
   "offset": 8826050,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "fuse2.b": {
   "offset": 8827074,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "fuse2.clip": {
   "offset": 8827586,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "bott.n1w": {
   "offset": 8827590,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "bott.n1b": {
   "offset": 8828102,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "bott.c1w8": {
   "offset": 8828614,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "bott.c1ws": {
   "offset": 10598086,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "bott.c1b": {
   "offset": 10599110,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "bott.c1clip": {
   "offset": 10599622,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "bott.c2w8": {
   "offset": 10599626,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "bott.c2ws": {
   "offset": 12369098,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "bott.c2b": {
   "offset": 12370122,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "bott.c2clip": {
   "offset": 12370634,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "up0.w8": {
   "offset": 12370638,
   "count": 65536,
   "shape": [
    256,
    256
   ],
   "dtype": "int8"
  },
  "up0.ws": {
   "offset": 12436174,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "up0.b": {
   "offset": 12437198,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "up0.clip": {
   "offset": 12437710,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec0.n1w": {
   "offset": 12437714,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "dec0.n1b": {
   "offset": 12438226,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "dec0.c1w8": {
   "offset": 12438738,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "dec0.c1ws": {
   "offset": 14208210,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "dec0.c1b": {
   "offset": 14209234,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "dec0.c1clip": {
   "offset": 14209746,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec0.c2w8": {
   "offset": 14209750,
   "count": 1769472,
   "shape": [
    27,
    256,
    256
   ],
   "dtype": "int8"
  },
  "dec0.c2ws": {
   "offset": 15979222,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float32"
  },
  "dec0.c2b": {
   "offset": 15980246,
   "count": 256,
   "shape": [
    256
   ],
   "dtype": "float16"
  },
  "dec0.c2clip": {
   "offset": 15980758,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "up1.w8": {
   "offset": 15980762,
   "count": 32768,
   "shape": [
    256,
    128
   ],
   "dtype": "int8"
  },
  "up1.ws": {
   "offset": 16013530,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float32"
  },
  "up1.b": {
   "offset": 16014042,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "up1.clip": {
   "offset": 16014298,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec1.n1w": {
   "offset": 16014302,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "dec1.n1b": {
   "offset": 16014558,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "dec1.c1w8": {
   "offset": 16014814,
   "count": 442368,
   "shape": [
    27,
    128,
    128
   ],
   "dtype": "int8"
  },
  "dec1.c1ws": {
   "offset": 16457182,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float32"
  },
  "dec1.c1b": {
   "offset": 16457694,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "dec1.c1clip": {
   "offset": 16457950,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec1.c2w8": {
   "offset": 16457954,
   "count": 442368,
   "shape": [
    27,
    128,
    128
   ],
   "dtype": "int8"
  },
  "dec1.c2ws": {
   "offset": 16900322,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float32"
  },
  "dec1.c2b": {
   "offset": 16900834,
   "count": 128,
   "shape": [
    128
   ],
   "dtype": "float16"
  },
  "dec1.c2clip": {
   "offset": 16901090,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "up2.w8": {
   "offset": 16901094,
   "count": 8192,
   "shape": [
    128,
    64
   ],
   "dtype": "int8"
  },
  "up2.ws": {
   "offset": 16909286,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float32"
  },
  "up2.b": {
   "offset": 16909542,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "up2.clip": {
   "offset": 16909670,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec2.n1w": {
   "offset": 16909674,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "dec2.n1b": {
   "offset": 16909802,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "dec2.c1w8": {
   "offset": 16909930,
   "count": 110592,
   "shape": [
    27,
    64,
    64
   ],
   "dtype": "int8"
  },
  "dec2.c1ws": {
   "offset": 17020522,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float32"
  },
  "dec2.c1b": {
   "offset": 17020778,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "dec2.c1clip": {
   "offset": 17020906,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec2.c2w8": {
   "offset": 17020910,
   "count": 110592,
   "shape": [
    27,
    64,
    64
   ],
   "dtype": "int8"
  },
  "dec2.c2ws": {
   "offset": 17131502,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float32"
  },
  "dec2.c2b": {
   "offset": 17131758,
   "count": 64,
   "shape": [
    64
   ],
   "dtype": "float16"
  },
  "dec2.c2clip": {
   "offset": 17131886,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "up3.w8": {
   "offset": 17131890,
   "count": 2048,
   "shape": [
    64,
    32
   ],
   "dtype": "int8"
  },
  "up3.ws": {
   "offset": 17133938,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float32"
  },
  "up3.b": {
   "offset": 17134066,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "up3.clip": {
   "offset": 17134130,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec3.n1w": {
   "offset": 17134134,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "dec3.n1b": {
   "offset": 17134198,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "dec3.c1w8": {
   "offset": 17134262,
   "count": 27648,
   "shape": [
    27,
    32,
    32
   ],
   "dtype": "int8"
  },
  "dec3.c1ws": {
   "offset": 17161910,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float32"
  },
  "dec3.c1b": {
   "offset": 17162038,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "dec3.c1clip": {
   "offset": 17162102,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "dec3.c2w8": {
   "offset": 17162106,
   "count": 27648,
   "shape": [
    27,
    32,
    32
   ],
   "dtype": "int8"
  },
  "dec3.c2ws": {
   "offset": 17189754,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float32"
  },
  "dec3.c2b": {
   "offset": 17189882,
   "count": 32,
   "shape": [
    32
   ],
   "dtype": "float16"
  },
  "dec3.c2clip": {
   "offset": 17189946,
   "count": 1,
   "shape": [
    1
   ],
   "dtype": "float32"
  },
  "uv_bias": {
   "offset": 17189950,
   "count": 307200,
   "shape": [
    256,
    30,
    40
   ],
   "dtype": "float32"
  }
 }
}