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