{
    "metadata": {
        "unetParamSize": 1680,
        "vaeParamSize": 140,
        "clipParamSize": 196,
        "clip2ParamSize": 517
    },
    "records": [
        {
            "dataPath": "params_shard_0.bin",
            "format": "raw-shard",
            "nbytes": 28218880,
            "records": [
                {
                    "name": "unet_0",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 2560
                },
                {
                    "name": "unet_2",
                    "shape": [
                        320,
                        4,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 23040,
                    "byteOffset": 5120
                },
                {
                    "name": "unet_3",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 28160
                },
                {
                    "name": "unet_4",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 28800
                },
                {
                    "name": "unet_5",
                    "shape": [
                        4,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 23040,
                    "byteOffset": 29440
                },
                {
                    "name": "unet_6",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 52480
                },
                {
                    "name": "unet_7",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 1895680
                },
                {
                    "name": "unet_8",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 3738880
                },
                {
                    "name": "unet_9",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 5582080
                },
                {
                    "name": "unet_10",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 5582720
                },
                {
                    "name": "unet_11",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 5583360
                },
                {
                    "name": "unet_12",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 5584000
                },
                {
                    "name": "unet_13",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 5584640
                },
                {
                    "name": "unet_14",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 5585280
                },
                {
                    "name": "unet_15",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 7428480
                },
                {
                    "name": "unet_16",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 9271680
                },
                {
                    "name": "unet_17",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 9272320
                },
                {
                    "name": "unet_18",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 9272960
                },
                {
                    "name": "unet_19",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 9273600
                },
                {
                    "name": "unet_20",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 9274240
                },
                {
                    "name": "unet_21",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9274880
                },
                {
                    "name": "unet_22",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9276160
                },
                {
                    "name": "unet_23",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9277440
                },
                {
                    "name": "unet_24",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9278720
                },
                {
                    "name": "unet_25",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9280000
                },
                {
                    "name": "unet_26",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9281280
                },
                {
                    "name": "unet_27",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 9282560
                },
                {
                    "name": "unet_28",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9292800
                },
                {
                    "name": "unet_29",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9294080
                },
                {
                    "name": "unet_30",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9295360
                },
                {
                    "name": "unet_31",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9296640
                },
                {
                    "name": "unet_32",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9297920
                },
                {
                    "name": "unet_33",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9299200
                },
                {
                    "name": "unet_34",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9300480
                },
                {
                    "name": "unet_35",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9301760
                },
                {
                    "name": "unet_36",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9303040
                },
                {
                    "name": "unet_37",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 9304320
                },
                {
                    "name": "unet_38",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9314560
                },
                {
                    "name": "unet_39",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9315840
                },
                {
                    "name": "unet_40",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9317120
                },
                {
                    "name": "unet_41",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9318400
                },
                {
                    "name": "unet_42",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9319680
                },
                {
                    "name": "unet_43",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9320960
                },
                {
                    "name": "unet_44",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9322240
                },
                {
                    "name": "unet_45",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9323520
                },
                {
                    "name": "unet_46",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9324800
                },
                {
                    "name": "unet_47",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9326080
                },
                {
                    "name": "unet_48",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9327360
                },
                {
                    "name": "unet_49",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9328640
                },
                {
                    "name": "unet_50",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9329920
                },
                {
                    "name": "unet_51",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 9331200
                },
                {
                    "name": "unet_52",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9341440
                },
                {
                    "name": "unet_53",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9342720
                },
                {
                    "name": "unet_54",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9344000
                },
                {
                    "name": "unet_55",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9345280
                },
                {
                    "name": "unet_56",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9346560
                },
                {
                    "name": "unet_57",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9347840
                },
                {
                    "name": "unet_58",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9349120
                },
                {
                    "name": "unet_59",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9350400
                },
                {
                    "name": "unet_60",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9351680
                },
                {
                    "name": "unet_61",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 9352960
                },
                {
                    "name": "unet_62",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9363200
                },
                {
                    "name": "unet_63",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9364480
                },
                {
                    "name": "unet_64",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9365760
                },
                {
                    "name": "unet_65",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9367040
                },
                {
                    "name": "unet_66",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9368320
                },
                {
                    "name": "unet_67",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9369600
                },
                {
                    "name": "unet_68",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 9370880
                },
                {
                    "name": "unet_69",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 9372160
                },
                {
                    "name": "unet_70",
                    "shape": [
                        640,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3686400,
                    "byteOffset": 16744960
                },
                {
                    "name": "unet_71",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 20431360
                },
                {
                    "name": "unet_72",
                    "shape": [
                        640,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 409600,
                    "byteOffset": 27804160
                },
                {
                    "name": "unet_73",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 28213760
                },
                {
                    "name": "unet_74",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 28214400
                },
                {
                    "name": "unet_75",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 28215040
                },
                {
                    "name": "unet_76",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 28216320
                },
                {
                    "name": "unet_77",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 28217600
                }
            ]
        },
        {
            "dataPath": "params_shard_1.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_294",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_2.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_301",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_3.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_302",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_4.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_412",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_5.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_413",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_6.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_419",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_7.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_420",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_8.bin",
            "format": "raw-shard",
            "nbytes": 33552640,
            "records": [
                {
                    "name": "unet_78",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_79",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 7372800
                },
                {
                    "name": "unet_80",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14745600
                },
                {
                    "name": "unet_81",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14746880
                },
                {
                    "name": "unet_82",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14748160
                },
                {
                    "name": "unet_83",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14749440
                },
                {
                    "name": "unet_84",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14750720
                },
                {
                    "name": "unet_85",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14752000
                },
                {
                    "name": "unet_86",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14754560
                },
                {
                    "name": "unet_87",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14757120
                },
                {
                    "name": "unet_88",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14759680
                },
                {
                    "name": "unet_89",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14762240
                },
                {
                    "name": "unet_90",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14764800
                },
                {
                    "name": "unet_91",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 14767360
                },
                {
                    "name": "unet_92",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14787840
                },
                {
                    "name": "unet_93",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14790400
                },
                {
                    "name": "unet_94",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14792960
                },
                {
                    "name": "unet_95",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14795520
                },
                {
                    "name": "unet_96",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14798080
                },
                {
                    "name": "unet_97",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14800640
                },
                {
                    "name": "unet_98",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14803200
                },
                {
                    "name": "unet_99",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14805760
                },
                {
                    "name": "unet_100",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14808320
                },
                {
                    "name": "unet_101",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 14810880
                },
                {
                    "name": "unet_102",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14831360
                },
                {
                    "name": "unet_103",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14833920
                },
                {
                    "name": "unet_104",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14836480
                },
                {
                    "name": "unet_105",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14839040
                },
                {
                    "name": "unet_106",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14841600
                },
                {
                    "name": "unet_107",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14844160
                },
                {
                    "name": "unet_108",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14846720
                },
                {
                    "name": "unet_109",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14849280
                },
                {
                    "name": "unet_110",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14851840
                },
                {
                    "name": "unet_111",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 14854400
                },
                {
                    "name": "unet_112",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14874880
                },
                {
                    "name": "unet_113",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14877440
                },
                {
                    "name": "unet_114",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14880000
                },
                {
                    "name": "unet_115",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14882560
                },
                {
                    "name": "unet_116",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14885120
                },
                {
                    "name": "unet_117",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14887680
                },
                {
                    "name": "unet_118",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14890240
                },
                {
                    "name": "unet_119",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14892800
                },
                {
                    "name": "unet_120",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14895360
                },
                {
                    "name": "unet_121",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 14897920
                },
                {
                    "name": "unet_122",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14918400
                },
                {
                    "name": "unet_123",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14920960
                },
                {
                    "name": "unet_124",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14923520
                },
                {
                    "name": "unet_125",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14926080
                },
                {
                    "name": "unet_126",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14928640
                },
                {
                    "name": "unet_127",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14931200
                },
                {
                    "name": "unet_128",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14933760
                },
                {
                    "name": "unet_129",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14936320
                },
                {
                    "name": "unet_130",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14938880
                },
                {
                    "name": "unet_131",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 14941440
                },
                {
                    "name": "unet_132",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14961920
                },
                {
                    "name": "unet_133",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14964480
                },
                {
                    "name": "unet_134",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14967040
                },
                {
                    "name": "unet_135",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14969600
                },
                {
                    "name": "unet_136",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14972160
                },
                {
                    "name": "unet_137",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14974720
                },
                {
                    "name": "unet_138",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14977280
                },
                {
                    "name": "unet_139",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14979840
                },
                {
                    "name": "unet_140",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 14982400
                },
                {
                    "name": "unet_141",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 14984960
                },
                {
                    "name": "unet_142",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15005440
                },
                {
                    "name": "unet_143",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15008000
                },
                {
                    "name": "unet_144",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15010560
                },
                {
                    "name": "unet_145",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15013120
                },
                {
                    "name": "unet_146",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15015680
                },
                {
                    "name": "unet_147",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15018240
                },
                {
                    "name": "unet_148",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15020800
                },
                {
                    "name": "unet_149",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15023360
                },
                {
                    "name": "unet_150",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15025920
                },
                {
                    "name": "unet_151",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15028480
                },
                {
                    "name": "unet_152",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15048960
                },
                {
                    "name": "unet_153",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15051520
                },
                {
                    "name": "unet_154",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15054080
                },
                {
                    "name": "unet_155",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15056640
                },
                {
                    "name": "unet_156",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15059200
                },
                {
                    "name": "unet_157",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15061760
                },
                {
                    "name": "unet_158",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15064320
                },
                {
                    "name": "unet_159",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15066880
                },
                {
                    "name": "unet_160",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15069440
                },
                {
                    "name": "unet_161",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15072000
                },
                {
                    "name": "unet_162",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15092480
                },
                {
                    "name": "unet_163",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15095040
                },
                {
                    "name": "unet_164",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15097600
                },
                {
                    "name": "unet_165",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15100160
                },
                {
                    "name": "unet_166",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15102720
                },
                {
                    "name": "unet_167",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15105280
                },
                {
                    "name": "unet_168",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15107840
                },
                {
                    "name": "unet_169",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15110400
                },
                {
                    "name": "unet_170",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15112960
                },
                {
                    "name": "unet_171",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15115520
                },
                {
                    "name": "unet_172",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15136000
                },
                {
                    "name": "unet_173",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15138560
                },
                {
                    "name": "unet_174",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15141120
                },
                {
                    "name": "unet_175",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15143680
                },
                {
                    "name": "unet_176",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15146240
                },
                {
                    "name": "unet_177",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15148800
                },
                {
                    "name": "unet_178",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15151360
                },
                {
                    "name": "unet_179",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15153920
                },
                {
                    "name": "unet_180",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15156480
                },
                {
                    "name": "unet_181",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15159040
                },
                {
                    "name": "unet_182",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15179520
                },
                {
                    "name": "unet_183",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15182080
                },
                {
                    "name": "unet_184",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15184640
                },
                {
                    "name": "unet_185",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15187200
                },
                {
                    "name": "unet_186",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15189760
                },
                {
                    "name": "unet_187",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15192320
                },
                {
                    "name": "unet_188",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15194880
                },
                {
                    "name": "unet_189",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15197440
                },
                {
                    "name": "unet_190",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15200000
                },
                {
                    "name": "unet_191",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15202560
                },
                {
                    "name": "unet_192",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15205120
                },
                {
                    "name": "unet_193",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15207680
                },
                {
                    "name": "unet_194",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15210240
                },
                {
                    "name": "unet_195",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15212800
                },
                {
                    "name": "unet_196",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15233280
                },
                {
                    "name": "unet_197",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15235840
                },
                {
                    "name": "unet_198",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15238400
                },
                {
                    "name": "unet_199",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15240960
                },
                {
                    "name": "unet_200",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15243520
                },
                {
                    "name": "unet_201",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15246080
                },
                {
                    "name": "unet_202",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15248640
                },
                {
                    "name": "unet_203",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15251200
                },
                {
                    "name": "unet_204",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15253760
                },
                {
                    "name": "unet_205",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15256320
                },
                {
                    "name": "unet_206",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15276800
                },
                {
                    "name": "unet_207",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15279360
                },
                {
                    "name": "unet_208",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15281920
                },
                {
                    "name": "unet_209",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15284480
                },
                {
                    "name": "unet_210",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15287040
                },
                {
                    "name": "unet_211",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15289600
                },
                {
                    "name": "unet_212",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15292160
                },
                {
                    "name": "unet_213",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15294720
                },
                {
                    "name": "unet_214",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15297280
                },
                {
                    "name": "unet_215",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15299840
                },
                {
                    "name": "unet_216",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15320320
                },
                {
                    "name": "unet_217",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15322880
                },
                {
                    "name": "unet_218",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15325440
                },
                {
                    "name": "unet_219",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15328000
                },
                {
                    "name": "unet_220",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15330560
                },
                {
                    "name": "unet_221",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15333120
                },
                {
                    "name": "unet_222",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15335680
                },
                {
                    "name": "unet_223",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15338240
                },
                {
                    "name": "unet_224",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15340800
                },
                {
                    "name": "unet_225",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15343360
                },
                {
                    "name": "unet_226",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15363840
                },
                {
                    "name": "unet_227",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15366400
                },
                {
                    "name": "unet_228",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15368960
                },
                {
                    "name": "unet_229",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15371520
                },
                {
                    "name": "unet_230",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15374080
                },
                {
                    "name": "unet_231",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15376640
                },
                {
                    "name": "unet_232",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15379200
                },
                {
                    "name": "unet_233",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15381760
                },
                {
                    "name": "unet_234",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15384320
                },
                {
                    "name": "unet_235",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15386880
                },
                {
                    "name": "unet_236",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15407360
                },
                {
                    "name": "unet_237",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15409920
                },
                {
                    "name": "unet_238",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15412480
                },
                {
                    "name": "unet_239",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15415040
                },
                {
                    "name": "unet_240",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15417600
                },
                {
                    "name": "unet_241",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15420160
                },
                {
                    "name": "unet_242",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15422720
                },
                {
                    "name": "unet_243",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15425280
                },
                {
                    "name": "unet_244",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15427840
                },
                {
                    "name": "unet_245",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15430400
                },
                {
                    "name": "unet_246",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15450880
                },
                {
                    "name": "unet_247",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15453440
                },
                {
                    "name": "unet_248",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15456000
                },
                {
                    "name": "unet_249",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15458560
                },
                {
                    "name": "unet_250",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15461120
                },
                {
                    "name": "unet_251",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15463680
                },
                {
                    "name": "unet_252",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15466240
                },
                {
                    "name": "unet_253",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15468800
                },
                {
                    "name": "unet_254",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15471360
                },
                {
                    "name": "unet_255",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15473920
                },
                {
                    "name": "unet_256",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15494400
                },
                {
                    "name": "unet_257",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15496960
                },
                {
                    "name": "unet_258",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15499520
                },
                {
                    "name": "unet_259",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15502080
                },
                {
                    "name": "unet_260",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15504640
                },
                {
                    "name": "unet_261",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15507200
                },
                {
                    "name": "unet_262",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15509760
                },
                {
                    "name": "unet_263",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15512320
                },
                {
                    "name": "unet_264",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15514880
                },
                {
                    "name": "unet_265",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15517440
                },
                {
                    "name": "unet_266",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15537920
                },
                {
                    "name": "unet_267",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15540480
                },
                {
                    "name": "unet_268",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15543040
                },
                {
                    "name": "unet_269",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15545600
                },
                {
                    "name": "unet_270",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15548160
                },
                {
                    "name": "unet_271",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15550720
                },
                {
                    "name": "unet_272",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15553280
                },
                {
                    "name": "unet_273",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15555840
                },
                {
                    "name": "unet_274",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15558400
                },
                {
                    "name": "unet_275",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15560960
                },
                {
                    "name": "unet_276",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15581440
                },
                {
                    "name": "unet_277",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15584000
                },
                {
                    "name": "unet_278",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15586560
                },
                {
                    "name": "unet_279",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15589120
                },
                {
                    "name": "unet_280",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15591680
                },
                {
                    "name": "unet_281",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15594240
                },
                {
                    "name": "unet_282",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15596800
                },
                {
                    "name": "unet_283",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15599360
                },
                {
                    "name": "unet_284",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15601920
                },
                {
                    "name": "unet_285",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 15604480
                },
                {
                    "name": "unet_286",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15624960
                },
                {
                    "name": "unet_287",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15627520
                },
                {
                    "name": "unet_288",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15630080
                },
                {
                    "name": "unet_289",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15632640
                },
                {
                    "name": "unet_290",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15635200
                },
                {
                    "name": "unet_291",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15637760
                },
                {
                    "name": "unet_292",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 15640320
                },
                {
                    "name": "unet_293",
                    "shape": [
                        1280,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 14745600,
                    "byteOffset": 15642880
                },
                {
                    "name": "unet_295",
                    "shape": [
                        1280,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 30388480
                },
                {
                    "name": "unet_296",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 32026880
                },
                {
                    "name": "unet_297",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 32028160
                },
                {
                    "name": "unet_298",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32029440
                },
                {
                    "name": "unet_299",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32032000
                },
                {
                    "name": "unet_300",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32034560
                },
                {
                    "name": "unet_303",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32037120
                },
                {
                    "name": "unet_304",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32039680
                },
                {
                    "name": "unet_305",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32042240
                },
                {
                    "name": "unet_306",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32044800
                },
                {
                    "name": "unet_307",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32047360
                },
                {
                    "name": "unet_308",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32049920
                },
                {
                    "name": "unet_309",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32052480
                },
                {
                    "name": "unet_310",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32055040
                },
                {
                    "name": "unet_311",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32057600
                },
                {
                    "name": "unet_312",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32060160
                },
                {
                    "name": "unet_313",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32062720
                },
                {
                    "name": "unet_314",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32065280
                },
                {
                    "name": "unet_315",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32085760
                },
                {
                    "name": "unet_316",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32088320
                },
                {
                    "name": "unet_317",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32090880
                },
                {
                    "name": "unet_318",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32093440
                },
                {
                    "name": "unet_319",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32096000
                },
                {
                    "name": "unet_320",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32098560
                },
                {
                    "name": "unet_321",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32101120
                },
                {
                    "name": "unet_322",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32103680
                },
                {
                    "name": "unet_323",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32106240
                },
                {
                    "name": "unet_324",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32108800
                },
                {
                    "name": "unet_325",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32129280
                },
                {
                    "name": "unet_326",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32131840
                },
                {
                    "name": "unet_327",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32134400
                },
                {
                    "name": "unet_328",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32136960
                },
                {
                    "name": "unet_329",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32139520
                },
                {
                    "name": "unet_330",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32142080
                },
                {
                    "name": "unet_331",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32144640
                },
                {
                    "name": "unet_332",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32147200
                },
                {
                    "name": "unet_333",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32149760
                },
                {
                    "name": "unet_334",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32152320
                },
                {
                    "name": "unet_335",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32172800
                },
                {
                    "name": "unet_336",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32175360
                },
                {
                    "name": "unet_337",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32177920
                },
                {
                    "name": "unet_338",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32180480
                },
                {
                    "name": "unet_339",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32183040
                },
                {
                    "name": "unet_340",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32185600
                },
                {
                    "name": "unet_341",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32188160
                },
                {
                    "name": "unet_342",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32190720
                },
                {
                    "name": "unet_343",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32193280
                },
                {
                    "name": "unet_344",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32195840
                },
                {
                    "name": "unet_345",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32216320
                },
                {
                    "name": "unet_346",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32218880
                },
                {
                    "name": "unet_347",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32221440
                },
                {
                    "name": "unet_348",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32224000
                },
                {
                    "name": "unet_349",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32226560
                },
                {
                    "name": "unet_350",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32229120
                },
                {
                    "name": "unet_351",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32231680
                },
                {
                    "name": "unet_352",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32234240
                },
                {
                    "name": "unet_353",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32236800
                },
                {
                    "name": "unet_354",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32239360
                },
                {
                    "name": "unet_355",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32259840
                },
                {
                    "name": "unet_356",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32262400
                },
                {
                    "name": "unet_357",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32264960
                },
                {
                    "name": "unet_358",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32267520
                },
                {
                    "name": "unet_359",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32270080
                },
                {
                    "name": "unet_360",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32272640
                },
                {
                    "name": "unet_361",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32275200
                },
                {
                    "name": "unet_362",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32277760
                },
                {
                    "name": "unet_363",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32280320
                },
                {
                    "name": "unet_364",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32282880
                },
                {
                    "name": "unet_365",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32303360
                },
                {
                    "name": "unet_366",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32305920
                },
                {
                    "name": "unet_367",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32308480
                },
                {
                    "name": "unet_368",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32311040
                },
                {
                    "name": "unet_369",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32313600
                },
                {
                    "name": "unet_370",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32316160
                },
                {
                    "name": "unet_371",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32318720
                },
                {
                    "name": "unet_372",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32321280
                },
                {
                    "name": "unet_373",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32323840
                },
                {
                    "name": "unet_374",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32326400
                },
                {
                    "name": "unet_375",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32346880
                },
                {
                    "name": "unet_376",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32349440
                },
                {
                    "name": "unet_377",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32352000
                },
                {
                    "name": "unet_378",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32354560
                },
                {
                    "name": "unet_379",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32357120
                },
                {
                    "name": "unet_380",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32359680
                },
                {
                    "name": "unet_381",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32362240
                },
                {
                    "name": "unet_382",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32364800
                },
                {
                    "name": "unet_383",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32367360
                },
                {
                    "name": "unet_384",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32369920
                },
                {
                    "name": "unet_385",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32390400
                },
                {
                    "name": "unet_386",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32392960
                },
                {
                    "name": "unet_387",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32395520
                },
                {
                    "name": "unet_388",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32398080
                },
                {
                    "name": "unet_389",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32400640
                },
                {
                    "name": "unet_390",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32403200
                },
                {
                    "name": "unet_391",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32405760
                },
                {
                    "name": "unet_392",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32408320
                },
                {
                    "name": "unet_393",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32410880
                },
                {
                    "name": "unet_394",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32413440
                },
                {
                    "name": "unet_395",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32433920
                },
                {
                    "name": "unet_396",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32436480
                },
                {
                    "name": "unet_397",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32439040
                },
                {
                    "name": "unet_398",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32441600
                },
                {
                    "name": "unet_399",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32444160
                },
                {
                    "name": "unet_400",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32446720
                },
                {
                    "name": "unet_401",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32449280
                },
                {
                    "name": "unet_402",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32451840
                },
                {
                    "name": "unet_403",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32454400
                },
                {
                    "name": "unet_404",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32456960
                },
                {
                    "name": "unet_405",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32477440
                },
                {
                    "name": "unet_406",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32480000
                },
                {
                    "name": "unet_407",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32482560
                },
                {
                    "name": "unet_408",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32485120
                },
                {
                    "name": "unet_409",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32487680
                },
                {
                    "name": "unet_410",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32490240
                },
                {
                    "name": "unet_411",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32492800
                },
                {
                    "name": "unet_414",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32495360
                },
                {
                    "name": "unet_415",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32497920
                },
                {
                    "name": "unet_416",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32500480
                },
                {
                    "name": "unet_417",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32503040
                },
                {
                    "name": "unet_418",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32505600
                },
                {
                    "name": "unet_421",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32508160
                },
                {
                    "name": "unet_422",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32510720
                },
                {
                    "name": "unet_423",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32513280
                },
                {
                    "name": "unet_424",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32515840
                },
                {
                    "name": "unet_425",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32518400
                },
                {
                    "name": "unet_426",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32520960
                },
                {
                    "name": "unet_427",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32523520
                },
                {
                    "name": "unet_428",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32526080
                },
                {
                    "name": "unet_429",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32528640
                },
                {
                    "name": "unet_430",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32531200
                },
                {
                    "name": "unet_431",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32533760
                },
                {
                    "name": "unet_432",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32536320
                },
                {
                    "name": "unet_433",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32538880
                },
                {
                    "name": "unet_434",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32541440
                },
                {
                    "name": "unet_435",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32561920
                },
                {
                    "name": "unet_436",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32564480
                },
                {
                    "name": "unet_437",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32567040
                },
                {
                    "name": "unet_438",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32569600
                },
                {
                    "name": "unet_439",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32572160
                },
                {
                    "name": "unet_440",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32574720
                },
                {
                    "name": "unet_441",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32577280
                },
                {
                    "name": "unet_442",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32579840
                },
                {
                    "name": "unet_443",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32582400
                },
                {
                    "name": "unet_444",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32584960
                },
                {
                    "name": "unet_445",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32605440
                },
                {
                    "name": "unet_446",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32608000
                },
                {
                    "name": "unet_447",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32610560
                },
                {
                    "name": "unet_448",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32613120
                },
                {
                    "name": "unet_449",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32615680
                },
                {
                    "name": "unet_450",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32618240
                },
                {
                    "name": "unet_451",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32620800
                },
                {
                    "name": "unet_452",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32623360
                },
                {
                    "name": "unet_453",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32625920
                },
                {
                    "name": "unet_454",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32628480
                },
                {
                    "name": "unet_455",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32648960
                },
                {
                    "name": "unet_456",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32651520
                },
                {
                    "name": "unet_457",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32654080
                },
                {
                    "name": "unet_458",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32656640
                },
                {
                    "name": "unet_459",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32659200
                },
                {
                    "name": "unet_460",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32661760
                },
                {
                    "name": "unet_461",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32664320
                },
                {
                    "name": "unet_462",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32666880
                },
                {
                    "name": "unet_463",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32669440
                },
                {
                    "name": "unet_464",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32672000
                },
                {
                    "name": "unet_465",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32692480
                },
                {
                    "name": "unet_466",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32695040
                },
                {
                    "name": "unet_467",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32697600
                },
                {
                    "name": "unet_468",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32700160
                },
                {
                    "name": "unet_469",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32702720
                },
                {
                    "name": "unet_470",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32705280
                },
                {
                    "name": "unet_471",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32707840
                },
                {
                    "name": "unet_472",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32710400
                },
                {
                    "name": "unet_473",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32712960
                },
                {
                    "name": "unet_474",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32715520
                },
                {
                    "name": "unet_475",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32736000
                },
                {
                    "name": "unet_476",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32738560
                },
                {
                    "name": "unet_477",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32741120
                },
                {
                    "name": "unet_478",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32743680
                },
                {
                    "name": "unet_479",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32746240
                },
                {
                    "name": "unet_480",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32748800
                },
                {
                    "name": "unet_481",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32751360
                },
                {
                    "name": "unet_482",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32753920
                },
                {
                    "name": "unet_483",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32756480
                },
                {
                    "name": "unet_484",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32759040
                },
                {
                    "name": "unet_485",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32779520
                },
                {
                    "name": "unet_486",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32782080
                },
                {
                    "name": "unet_487",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32784640
                },
                {
                    "name": "unet_488",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32787200
                },
                {
                    "name": "unet_489",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32789760
                },
                {
                    "name": "unet_490",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32792320
                },
                {
                    "name": "unet_491",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32794880
                },
                {
                    "name": "unet_492",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32797440
                },
                {
                    "name": "unet_493",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32800000
                },
                {
                    "name": "unet_494",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32802560
                },
                {
                    "name": "unet_495",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32823040
                },
                {
                    "name": "unet_496",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32825600
                },
                {
                    "name": "unet_497",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32828160
                },
                {
                    "name": "unet_498",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32830720
                },
                {
                    "name": "unet_499",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32833280
                },
                {
                    "name": "unet_500",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32835840
                },
                {
                    "name": "unet_501",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32838400
                },
                {
                    "name": "unet_502",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32840960
                },
                {
                    "name": "unet_503",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32843520
                },
                {
                    "name": "unet_504",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32846080
                },
                {
                    "name": "unet_505",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32866560
                },
                {
                    "name": "unet_506",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32869120
                },
                {
                    "name": "unet_507",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32871680
                },
                {
                    "name": "unet_508",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32874240
                },
                {
                    "name": "unet_509",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32876800
                },
                {
                    "name": "unet_510",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32879360
                },
                {
                    "name": "unet_511",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32881920
                },
                {
                    "name": "unet_512",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32884480
                },
                {
                    "name": "unet_513",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32887040
                },
                {
                    "name": "unet_514",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32889600
                },
                {
                    "name": "unet_515",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32910080
                },
                {
                    "name": "unet_516",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32912640
                },
                {
                    "name": "unet_517",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32915200
                },
                {
                    "name": "unet_518",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32917760
                },
                {
                    "name": "unet_519",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32920320
                },
                {
                    "name": "unet_520",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32922880
                },
                {
                    "name": "unet_521",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32925440
                },
                {
                    "name": "unet_522",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32928000
                },
                {
                    "name": "unet_523",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32930560
                },
                {
                    "name": "unet_524",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32933120
                },
                {
                    "name": "unet_525",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32953600
                },
                {
                    "name": "unet_526",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32956160
                },
                {
                    "name": "unet_527",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32958720
                },
                {
                    "name": "unet_528",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32961280
                },
                {
                    "name": "unet_529",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32963840
                },
                {
                    "name": "unet_530",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32966400
                },
                {
                    "name": "unet_531",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32968960
                },
                {
                    "name": "unet_532",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32971520
                },
                {
                    "name": "unet_533",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32974080
                },
                {
                    "name": "unet_534",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32976640
                },
                {
                    "name": "unet_535",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32979200
                },
                {
                    "name": "unet_536",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32981760
                },
                {
                    "name": "unet_537",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 32984320
                },
                {
                    "name": "unet_538",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 32986880
                },
                {
                    "name": "unet_539",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33007360
                },
                {
                    "name": "unet_540",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33009920
                },
                {
                    "name": "unet_541",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33012480
                },
                {
                    "name": "unet_542",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33015040
                },
                {
                    "name": "unet_543",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33017600
                },
                {
                    "name": "unet_544",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33020160
                },
                {
                    "name": "unet_545",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33022720
                },
                {
                    "name": "unet_546",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33025280
                },
                {
                    "name": "unet_547",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33027840
                },
                {
                    "name": "unet_548",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33030400
                },
                {
                    "name": "unet_549",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33050880
                },
                {
                    "name": "unet_550",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33053440
                },
                {
                    "name": "unet_551",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33056000
                },
                {
                    "name": "unet_552",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33058560
                },
                {
                    "name": "unet_553",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33061120
                },
                {
                    "name": "unet_554",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33063680
                },
                {
                    "name": "unet_555",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33066240
                },
                {
                    "name": "unet_556",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33068800
                },
                {
                    "name": "unet_557",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33071360
                },
                {
                    "name": "unet_558",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33073920
                },
                {
                    "name": "unet_559",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33094400
                },
                {
                    "name": "unet_560",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33096960
                },
                {
                    "name": "unet_561",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33099520
                },
                {
                    "name": "unet_562",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33102080
                },
                {
                    "name": "unet_563",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33104640
                },
                {
                    "name": "unet_564",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33107200
                },
                {
                    "name": "unet_565",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33109760
                },
                {
                    "name": "unet_566",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33112320
                },
                {
                    "name": "unet_567",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33114880
                },
                {
                    "name": "unet_568",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33117440
                },
                {
                    "name": "unet_569",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33137920
                },
                {
                    "name": "unet_570",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33140480
                },
                {
                    "name": "unet_571",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33143040
                },
                {
                    "name": "unet_572",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33145600
                },
                {
                    "name": "unet_573",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33148160
                },
                {
                    "name": "unet_574",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33150720
                },
                {
                    "name": "unet_575",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33153280
                },
                {
                    "name": "unet_576",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33155840
                },
                {
                    "name": "unet_577",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33158400
                },
                {
                    "name": "unet_578",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33160960
                },
                {
                    "name": "unet_579",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33181440
                },
                {
                    "name": "unet_580",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33184000
                },
                {
                    "name": "unet_581",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33186560
                },
                {
                    "name": "unet_582",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33189120
                },
                {
                    "name": "unet_583",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33191680
                },
                {
                    "name": "unet_584",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33194240
                },
                {
                    "name": "unet_585",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33196800
                },
                {
                    "name": "unet_586",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33199360
                },
                {
                    "name": "unet_587",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33201920
                },
                {
                    "name": "unet_588",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33204480
                },
                {
                    "name": "unet_589",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33224960
                },
                {
                    "name": "unet_590",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33227520
                },
                {
                    "name": "unet_591",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33230080
                },
                {
                    "name": "unet_592",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33232640
                },
                {
                    "name": "unet_593",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33235200
                },
                {
                    "name": "unet_594",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33237760
                },
                {
                    "name": "unet_595",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33240320
                },
                {
                    "name": "unet_596",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33242880
                },
                {
                    "name": "unet_597",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33245440
                },
                {
                    "name": "unet_598",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33248000
                },
                {
                    "name": "unet_599",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33268480
                },
                {
                    "name": "unet_600",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33271040
                },
                {
                    "name": "unet_601",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33273600
                },
                {
                    "name": "unet_602",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33276160
                },
                {
                    "name": "unet_603",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33278720
                },
                {
                    "name": "unet_604",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33281280
                },
                {
                    "name": "unet_605",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33283840
                },
                {
                    "name": "unet_606",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33286400
                },
                {
                    "name": "unet_607",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33288960
                },
                {
                    "name": "unet_608",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33291520
                },
                {
                    "name": "unet_609",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33312000
                },
                {
                    "name": "unet_610",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33314560
                },
                {
                    "name": "unet_611",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33317120
                },
                {
                    "name": "unet_612",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33319680
                },
                {
                    "name": "unet_613",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33322240
                },
                {
                    "name": "unet_614",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33324800
                },
                {
                    "name": "unet_615",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33327360
                },
                {
                    "name": "unet_616",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33329920
                },
                {
                    "name": "unet_617",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33332480
                },
                {
                    "name": "unet_618",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33335040
                },
                {
                    "name": "unet_619",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33355520
                },
                {
                    "name": "unet_620",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33358080
                },
                {
                    "name": "unet_621",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33360640
                },
                {
                    "name": "unet_622",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33363200
                },
                {
                    "name": "unet_623",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33365760
                },
                {
                    "name": "unet_624",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33368320
                },
                {
                    "name": "unet_625",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33370880
                },
                {
                    "name": "unet_626",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33373440
                },
                {
                    "name": "unet_627",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33376000
                },
                {
                    "name": "unet_628",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33378560
                },
                {
                    "name": "unet_629",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33399040
                },
                {
                    "name": "unet_630",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33401600
                },
                {
                    "name": "unet_631",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33404160
                },
                {
                    "name": "unet_632",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33406720
                },
                {
                    "name": "unet_633",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33409280
                },
                {
                    "name": "unet_634",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33411840
                },
                {
                    "name": "unet_635",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33414400
                },
                {
                    "name": "unet_636",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33416960
                },
                {
                    "name": "unet_637",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33419520
                },
                {
                    "name": "unet_638",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33422080
                },
                {
                    "name": "unet_639",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33424640
                },
                {
                    "name": "unet_640",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33427200
                },
                {
                    "name": "unet_641",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33429760
                },
                {
                    "name": "unet_642",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33432320
                },
                {
                    "name": "unet_643",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33452800
                },
                {
                    "name": "unet_644",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33455360
                },
                {
                    "name": "unet_645",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33457920
                },
                {
                    "name": "unet_646",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33460480
                },
                {
                    "name": "unet_647",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33463040
                },
                {
                    "name": "unet_648",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33465600
                },
                {
                    "name": "unet_649",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33468160
                },
                {
                    "name": "unet_650",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33470720
                },
                {
                    "name": "unet_651",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33473280
                },
                {
                    "name": "unet_652",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33475840
                },
                {
                    "name": "unet_653",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33496320
                },
                {
                    "name": "unet_654",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33498880
                },
                {
                    "name": "unet_655",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33501440
                },
                {
                    "name": "unet_656",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33504000
                },
                {
                    "name": "unet_657",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33506560
                },
                {
                    "name": "unet_658",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33509120
                },
                {
                    "name": "unet_659",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33511680
                },
                {
                    "name": "unet_660",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33514240
                },
                {
                    "name": "unet_661",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33516800
                },
                {
                    "name": "unet_662",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 33519360
                },
                {
                    "name": "unet_663",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33539840
                },
                {
                    "name": "unet_664",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33542400
                },
                {
                    "name": "unet_665",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33544960
                },
                {
                    "name": "unet_666",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33547520
                },
                {
                    "name": "unet_667",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33550080
                }
            ]
        },
        {
            "dataPath": "params_shard_9.bin",
            "format": "raw-shard",
            "nbytes": 58982400,
            "records": [
                {
                    "name": "unet_740",
                    "shape": [
                        1280,
                        2560,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 58982400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_10.bin",
            "format": "raw-shard",
            "nbytes": 29800960,
            "records": [
                {
                    "name": "unet_668",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 0
                },
                {
                    "name": "unet_669",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 2560
                },
                {
                    "name": "unet_670",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5120
                },
                {
                    "name": "unet_671",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 7680
                },
                {
                    "name": "unet_672",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 10240
                },
                {
                    "name": "unet_673",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30720
                },
                {
                    "name": "unet_674",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33280
                },
                {
                    "name": "unet_675",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 35840
                },
                {
                    "name": "unet_676",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 38400
                },
                {
                    "name": "unet_677",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 40960
                },
                {
                    "name": "unet_678",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 43520
                },
                {
                    "name": "unet_679",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 46080
                },
                {
                    "name": "unet_680",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 48640
                },
                {
                    "name": "unet_681",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 51200
                },
                {
                    "name": "unet_682",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 53760
                },
                {
                    "name": "unet_683",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 74240
                },
                {
                    "name": "unet_684",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 76800
                },
                {
                    "name": "unet_685",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 79360
                },
                {
                    "name": "unet_686",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 81920
                },
                {
                    "name": "unet_687",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 84480
                },
                {
                    "name": "unet_688",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 87040
                },
                {
                    "name": "unet_689",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 89600
                },
                {
                    "name": "unet_690",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 92160
                },
                {
                    "name": "unet_691",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 94720
                },
                {
                    "name": "unet_692",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 97280
                },
                {
                    "name": "unet_693",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 117760
                },
                {
                    "name": "unet_694",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 120320
                },
                {
                    "name": "unet_695",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 122880
                },
                {
                    "name": "unet_696",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 125440
                },
                {
                    "name": "unet_697",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 128000
                },
                {
                    "name": "unet_698",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 130560
                },
                {
                    "name": "unet_699",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 133120
                },
                {
                    "name": "unet_700",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 135680
                },
                {
                    "name": "unet_701",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 138240
                },
                {
                    "name": "unet_702",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 140800
                },
                {
                    "name": "unet_703",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 161280
                },
                {
                    "name": "unet_704",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 163840
                },
                {
                    "name": "unet_705",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 166400
                },
                {
                    "name": "unet_706",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 168960
                },
                {
                    "name": "unet_707",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 171520
                },
                {
                    "name": "unet_708",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 174080
                },
                {
                    "name": "unet_709",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 176640
                },
                {
                    "name": "unet_710",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 179200
                },
                {
                    "name": "unet_711",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 181760
                },
                {
                    "name": "unet_712",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 184320
                },
                {
                    "name": "unet_713",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 204800
                },
                {
                    "name": "unet_714",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 207360
                },
                {
                    "name": "unet_715",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 209920
                },
                {
                    "name": "unet_716",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 212480
                },
                {
                    "name": "unet_717",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 215040
                },
                {
                    "name": "unet_718",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 217600
                },
                {
                    "name": "unet_719",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 220160
                },
                {
                    "name": "unet_720",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 222720
                },
                {
                    "name": "unet_721",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 225280
                },
                {
                    "name": "unet_722",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 227840
                },
                {
                    "name": "unet_723",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 248320
                },
                {
                    "name": "unet_724",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 250880
                },
                {
                    "name": "unet_725",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 253440
                },
                {
                    "name": "unet_726",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 256000
                },
                {
                    "name": "unet_727",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 258560
                },
                {
                    "name": "unet_728",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 261120
                },
                {
                    "name": "unet_729",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 263680
                },
                {
                    "name": "unet_730",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 266240
                },
                {
                    "name": "unet_731",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 268800
                },
                {
                    "name": "unet_732",
                    "shape": [
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 20480,
                    "byteOffset": 271360
                },
                {
                    "name": "unet_733",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 291840
                },
                {
                    "name": "unet_734",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 294400
                },
                {
                    "name": "unet_735",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 296960
                },
                {
                    "name": "unet_736",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 299520
                },
                {
                    "name": "unet_737",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 302080
                },
                {
                    "name": "unet_738",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 304640
                },
                {
                    "name": "unet_739",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 307200
                },
                {
                    "name": "unet_741",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 309760
                }
            ]
        },
        {
            "dataPath": "params_shard_11.bin",
            "format": "raw-shard",
            "nbytes": 58982400,
            "records": [
                {
                    "name": "unet_748",
                    "shape": [
                        1280,
                        2560,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 58982400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_12.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_749",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_13.bin",
            "format": "raw-shard",
            "nbytes": 44236800,
            "records": [
                {
                    "name": "unet_756",
                    "shape": [
                        1280,
                        1920,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 44236800,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_14.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_757",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_15.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_764",
                    "shape": [
                        1280,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 29491200,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_16.bin",
            "format": "raw-shard",
            "nbytes": 22118400,
            "records": [
                {
                    "name": "unet_837",
                    "shape": [
                        640,
                        1920,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 22118400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_17.bin",
            "format": "raw-shard",
            "nbytes": 28061440,
            "records": [
                {
                    "name": "unet_742",
                    "shape": [
                        1280,
                        2560,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 0
                },
                {
                    "name": "unet_743",
                    "shape": [
                        2560
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5120,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_744",
                    "shape": [
                        2560
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5120,
                    "byteOffset": 6558720
                },
                {
                    "name": "unet_745",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 6563840
                },
                {
                    "name": "unet_746",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 6566400
                },
                {
                    "name": "unet_747",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 6568960
                },
                {
                    "name": "unet_750",
                    "shape": [
                        1280,
                        2560,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 6571520
                },
                {
                    "name": "unet_751",
                    "shape": [
                        2560
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5120,
                    "byteOffset": 13125120
                },
                {
                    "name": "unet_752",
                    "shape": [
                        2560
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5120,
                    "byteOffset": 13130240
                },
                {
                    "name": "unet_753",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 13135360
                },
                {
                    "name": "unet_754",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 13137920
                },
                {
                    "name": "unet_755",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 13140480
                },
                {
                    "name": "unet_758",
                    "shape": [
                        1280,
                        1920,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4915200,
                    "byteOffset": 13143040
                },
                {
                    "name": "unet_759",
                    "shape": [
                        1920
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3840,
                    "byteOffset": 18058240
                },
                {
                    "name": "unet_760",
                    "shape": [
                        1920
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3840,
                    "byteOffset": 18062080
                },
                {
                    "name": "unet_761",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 18065920
                },
                {
                    "name": "unet_762",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 18068480
                },
                {
                    "name": "unet_763",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 18071040
                },
                {
                    "name": "unet_765",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18073600
                },
                {
                    "name": "unet_766",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18074880
                },
                {
                    "name": "unet_767",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18076160
                },
                {
                    "name": "unet_768",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18077440
                },
                {
                    "name": "unet_769",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18078720
                },
                {
                    "name": "unet_770",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18080000
                },
                {
                    "name": "unet_771",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 18081280
                },
                {
                    "name": "unet_772",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18091520
                },
                {
                    "name": "unet_773",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18092800
                },
                {
                    "name": "unet_774",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18094080
                },
                {
                    "name": "unet_775",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18095360
                },
                {
                    "name": "unet_776",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18096640
                },
                {
                    "name": "unet_777",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18097920
                },
                {
                    "name": "unet_778",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18099200
                },
                {
                    "name": "unet_779",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18100480
                },
                {
                    "name": "unet_780",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18101760
                },
                {
                    "name": "unet_781",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 18103040
                },
                {
                    "name": "unet_782",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18113280
                },
                {
                    "name": "unet_783",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18114560
                },
                {
                    "name": "unet_784",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18115840
                },
                {
                    "name": "unet_785",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18117120
                },
                {
                    "name": "unet_786",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18118400
                },
                {
                    "name": "unet_787",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18119680
                },
                {
                    "name": "unet_788",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18120960
                },
                {
                    "name": "unet_789",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18122240
                },
                {
                    "name": "unet_790",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18123520
                },
                {
                    "name": "unet_791",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18124800
                },
                {
                    "name": "unet_792",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18126080
                },
                {
                    "name": "unet_793",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18127360
                },
                {
                    "name": "unet_794",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18128640
                },
                {
                    "name": "unet_795",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 18129920
                },
                {
                    "name": "unet_796",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18140160
                },
                {
                    "name": "unet_797",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18141440
                },
                {
                    "name": "unet_798",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18142720
                },
                {
                    "name": "unet_799",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18144000
                },
                {
                    "name": "unet_800",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18145280
                },
                {
                    "name": "unet_801",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18146560
                },
                {
                    "name": "unet_802",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18147840
                },
                {
                    "name": "unet_803",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18149120
                },
                {
                    "name": "unet_804",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18150400
                },
                {
                    "name": "unet_805",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 18151680
                },
                {
                    "name": "unet_806",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18161920
                },
                {
                    "name": "unet_807",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18163200
                },
                {
                    "name": "unet_808",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18164480
                },
                {
                    "name": "unet_809",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18165760
                },
                {
                    "name": "unet_810",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18167040
                },
                {
                    "name": "unet_811",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18168320
                },
                {
                    "name": "unet_812",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18169600
                },
                {
                    "name": "unet_813",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18170880
                },
                {
                    "name": "unet_814",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18172160
                },
                {
                    "name": "unet_815",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18173440
                },
                {
                    "name": "unet_816",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18174720
                },
                {
                    "name": "unet_817",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18176000
                },
                {
                    "name": "unet_818",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18177280
                },
                {
                    "name": "unet_819",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 18178560
                },
                {
                    "name": "unet_820",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18188800
                },
                {
                    "name": "unet_821",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18190080
                },
                {
                    "name": "unet_822",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18191360
                },
                {
                    "name": "unet_823",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18192640
                },
                {
                    "name": "unet_824",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18193920
                },
                {
                    "name": "unet_825",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18195200
                },
                {
                    "name": "unet_826",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18196480
                },
                {
                    "name": "unet_827",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18197760
                },
                {
                    "name": "unet_828",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18199040
                },
                {
                    "name": "unet_829",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 18200320
                },
                {
                    "name": "unet_830",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18210560
                },
                {
                    "name": "unet_831",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18211840
                },
                {
                    "name": "unet_832",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18213120
                },
                {
                    "name": "unet_833",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18214400
                },
                {
                    "name": "unet_834",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18215680
                },
                {
                    "name": "unet_835",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18216960
                },
                {
                    "name": "unet_836",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 18218240
                },
                {
                    "name": "unet_838",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 18219520
                },
                {
                    "name": "unet_839",
                    "shape": [
                        640,
                        1920,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2457600,
                    "byteOffset": 25592320
                },
                {
                    "name": "unet_840",
                    "shape": [
                        1920
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3840,
                    "byteOffset": 28049920
                },
                {
                    "name": "unet_841",
                    "shape": [
                        1920
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3840,
                    "byteOffset": 28053760
                },
                {
                    "name": "unet_842",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 28057600
                },
                {
                    "name": "unet_843",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 28058880
                },
                {
                    "name": "unet_844",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 28060160
                }
            ]
        },
        {
            "dataPath": "params_shard_18.bin",
            "format": "raw-shard",
            "nbytes": 23765760,
            "records": [
                {
                    "name": "unet_845",
                    "shape": [
                        640,
                        1280,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 14745600,
                    "byteOffset": 0
                },
                {
                    "name": "unet_846",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 14745600
                },
                {
                    "name": "unet_847",
                    "shape": [
                        640,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 22118400
                },
                {
                    "name": "unet_848",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 23756800
                },
                {
                    "name": "unet_849",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 23759360
                },
                {
                    "name": "unet_850",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 23761920
                },
                {
                    "name": "unet_851",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 23763200
                },
                {
                    "name": "unet_852",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 23764480
                }
            ]
        },
        {
            "dataPath": "params_shard_19.bin",
            "format": "raw-shard",
            "nbytes": 32570880,
            "records": [
                {
                    "name": "unet_853",
                    "shape": [
                        640,
                        960,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 11059200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_854",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 11059200
                },
                {
                    "name": "unet_855",
                    "shape": [
                        640,
                        960,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1228800,
                    "byteOffset": 18432000
                },
                {
                    "name": "unet_856",
                    "shape": [
                        960
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1920,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_857",
                    "shape": [
                        960
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1920,
                    "byteOffset": 19662720
                },
                {
                    "name": "unet_858",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 19664640
                },
                {
                    "name": "unet_859",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 19665920
                },
                {
                    "name": "unet_860",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 19667200
                },
                {
                    "name": "unet_861",
                    "shape": [
                        640,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7372800,
                    "byteOffset": 19668480
                },
                {
                    "name": "unet_862",
                    "shape": [
                        320,
                        960,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5529600,
                    "byteOffset": 27041280
                }
            ]
        },
        {
            "dataPath": "params_shard_20.bin",
            "format": "raw-shard",
            "nbytes": 33036160,
            "records": [
                {
                    "name": "unet_863",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_864",
                    "shape": [
                        320,
                        960,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 614400,
                    "byteOffset": 1843200
                },
                {
                    "name": "unet_865",
                    "shape": [
                        960
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1920,
                    "byteOffset": 2457600
                },
                {
                    "name": "unet_866",
                    "shape": [
                        960
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1920,
                    "byteOffset": 2459520
                },
                {
                    "name": "unet_867",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 2461440
                },
                {
                    "name": "unet_868",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 2462080
                },
                {
                    "name": "unet_869",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 2462720
                },
                {
                    "name": "unet_870",
                    "shape": [
                        320,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3686400,
                    "byteOffset": 2463360
                },
                {
                    "name": "unet_871",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 6149760
                },
                {
                    "name": "unet_872",
                    "shape": [
                        320,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 409600,
                    "byteOffset": 7992960
                },
                {
                    "name": "unet_873",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 8402560
                },
                {
                    "name": "unet_874",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 8403840
                },
                {
                    "name": "unet_875",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 8405120
                },
                {
                    "name": "unet_876",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 8405760
                },
                {
                    "name": "unet_877",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 8406400
                },
                {
                    "name": "unet_878",
                    "shape": [
                        320,
                        640,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3686400,
                    "byteOffset": 8407040
                },
                {
                    "name": "unet_879",
                    "shape": [
                        320,
                        320,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1843200,
                    "byteOffset": 12093440
                },
                {
                    "name": "unet_880",
                    "shape": [
                        320,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 409600,
                    "byteOffset": 13936640
                },
                {
                    "name": "unet_881",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14346240
                },
                {
                    "name": "unet_882",
                    "shape": [
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 14347520
                },
                {
                    "name": "unet_883",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 14348800
                },
                {
                    "name": "unet_884",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 14349440
                },
                {
                    "name": "unet_885",
                    "shape": [
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 14350080
                },
                {
                    "name": "unet_886",
                    "shape": [
                        320,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 14350720
                },
                {
                    "name": "unet_887",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 15169920
                },
                {
                    "name": "unet_888",
                    "shape": [
                        2816,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 7208960,
                    "byteOffset": 18446720
                },
                {
                    "name": "unet_889",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 25655680
                },
                {
                    "name": "unet_890",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 28932480
                },
                {
                    "name": "unet_891",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 28933120
                },
                {
                    "name": "unet_892",
                    "shape": [
                        1280,
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 28933760
                },
                {
                    "name": "unet_893",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 29752960
                },
                {
                    "name": "unet_894",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 29753600
                },
                {
                    "name": "unet_895",
                    "shape": [
                        1280,
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 29754240
                },
                {
                    "name": "unet_896",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 30573440
                },
                {
                    "name": "unet_897",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 30574080
                },
                {
                    "name": "unet_898",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 30574720
                },
                {
                    "name": "unet_899",
                    "shape": [
                        1280,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 30576000
                },
                {
                    "name": "unet_900",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 32214400
                },
                {
                    "name": "unet_901",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 32215680
                },
                {
                    "name": "unet_902",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 32216960
                }
            ]
        },
        {
            "dataPath": "params_shard_21.bin",
            "format": "raw-shard",
            "nbytes": 30146560,
            "records": [
                {
                    "name": "unet_903",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_904",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 819200
                },
                {
                    "name": "unet_905",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 1638400
                },
                {
                    "name": "unet_906",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 2457600
                },
                {
                    "name": "unet_907",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_908",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 4096000
                },
                {
                    "name": "unet_909",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 6717440
                },
                {
                    "name": "unet_910",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 9338880
                },
                {
                    "name": "unet_911",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 10158080
                },
                {
                    "name": "unet_912",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16711680
                },
                {
                    "name": "unet_913",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 19988480
                },
                {
                    "name": "unet_914",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 20807680
                },
                {
                    "name": "unet_915",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 21626880
                },
                {
                    "name": "unet_916",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 22446080
                },
                {
                    "name": "unet_917",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 23265280
                },
                {
                    "name": "unet_918",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 24084480
                },
                {
                    "name": "unet_919",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 26705920
                },
                {
                    "name": "unet_920",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 29327360
                }
            ]
        },
        {
            "dataPath": "params_shard_22.bin",
            "format": "raw-shard",
            "nbytes": 33098240,
            "records": [
                {
                    "name": "unet_921",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 0
                },
                {
                    "name": "unet_922",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_923",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_924",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 10649600
                },
                {
                    "name": "unet_925",
                    "shape": [
                        1280,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 10650880
                },
                {
                    "name": "unet_926",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 12289280
                },
                {
                    "name": "unet_927",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 12290560
                },
                {
                    "name": "unet_928",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 13109760
                },
                {
                    "name": "unet_929",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 13928960
                },
                {
                    "name": "unet_930",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 14748160
                },
                {
                    "name": "unet_931",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 15567360
                },
                {
                    "name": "unet_932",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 16386560
                },
                {
                    "name": "unet_933",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 17205760
                },
                {
                    "name": "unet_934",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 19827200
                },
                {
                    "name": "unet_935",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 22448640
                },
                {
                    "name": "unet_936",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 23267840
                },
                {
                    "name": "unet_937",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29821440
                }
            ]
        },
        {
            "dataPath": "params_shard_23.bin",
            "format": "raw-shard",
            "nbytes": 30647040,
            "records": [
                {
                    "name": "unet_938",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_939",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 819200
                },
                {
                    "name": "unet_940",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 1638400
                },
                {
                    "name": "unet_941",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 2457600
                },
                {
                    "name": "unet_942",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_943",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 4096000
                },
                {
                    "name": "unet_944",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 6717440
                },
                {
                    "name": "unet_945",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 9338880
                },
                {
                    "name": "unet_946",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 10158080
                },
                {
                    "name": "unet_947",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16711680
                },
                {
                    "name": "unet_948",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 19988480
                },
                {
                    "name": "unet_949",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 20807680
                },
                {
                    "name": "unet_950",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 20808960
                },
                {
                    "name": "unet_951",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20811520
                },
                {
                    "name": "unet_952",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 24088320
                },
                {
                    "name": "unet_953",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 24090880
                },
                {
                    "name": "unet_954",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 24093440
                },
                {
                    "name": "unet_955",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 27370240
                }
            ]
        },
        {
            "dataPath": "params_shard_24.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_963",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_25.bin",
            "format": "raw-shard",
            "nbytes": 26869760,
            "records": [
                {
                    "name": "unet_956",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_957",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_958",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_959",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_960",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_961",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 18350080
                },
                {
                    "name": "unet_962",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 23592960
                }
            ]
        },
        {
            "dataPath": "params_shard_26.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_964",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_965",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_966",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_967",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_968",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_969",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_27.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_973",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_28.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_970",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_971",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_972",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_974",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_975",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_976",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_29.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_983",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_30.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_977",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_978",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_979",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_980",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_981",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_982",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_31.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_984",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_985",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_986",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_987",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_988",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_989",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_32.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_993",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_33.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_990",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_991",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_992",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_994",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_995",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_996",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_34.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1003",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_35.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_997",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_998",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_999",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1000",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1001",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1002",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_36.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1004",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1005",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1006",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1007",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1008",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1009",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_37.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1013",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_38.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1010",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1011",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1012",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1014",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1015",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1016",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_39.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1023",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_40.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1017",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1018",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1019",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1020",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1021",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1022",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_41.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1024",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1025",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1026",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1027",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1028",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1029",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_42.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1033",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_43.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1030",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1031",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1032",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1034",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1035",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1036",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_44.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1043",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_45.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1037",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1038",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1039",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1040",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1041",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1042",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_46.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1044",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1045",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1046",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1047",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1048",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1049",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_47.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1053",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_48.bin",
            "format": "raw-shard",
            "nbytes": 33428480,
            "records": [
                {
                    "name": "unet_1050",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1051",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1052",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1054",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1055",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1056",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30146560
                },
                {
                    "name": "unet_1057",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30149120
                },
                {
                    "name": "unet_1058",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33425920
                }
            ]
        },
        {
            "dataPath": "params_shard_49.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1068",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_50.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1059",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1060",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1061",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1062",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1063",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1064",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1065",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1066",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 24903680
                },
                {
                    "name": "unet_1067",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_51.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1069",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1070",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1071",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1072",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1073",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1074",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_52.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1078",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_53.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1075",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1076",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1077",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1079",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1080",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1081",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_54.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1088",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_55.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1082",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1083",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1084",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1085",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1086",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1087",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_56.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1089",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1090",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1091",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1092",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1093",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1094",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_57.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1098",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_58.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1095",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1096",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1097",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1099",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1100",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1101",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_59.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1108",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_60.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1102",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1103",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1104",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1105",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1106",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1107",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_61.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1109",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1110",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1111",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1112",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1113",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1114",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_62.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1118",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_63.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1115",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1116",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1117",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1119",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1120",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1121",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_64.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1128",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_65.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1122",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1123",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1124",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1125",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1126",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1127",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_66.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1129",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1130",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1131",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1132",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1133",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1134",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_67.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1138",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_68.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1135",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1136",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1137",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1139",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1140",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1141",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_69.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1148",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_70.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1142",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1143",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1144",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1145",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1146",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1147",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_71.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1149",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1150",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1151",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1152",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1153",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1154",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_72.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1158",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_73.bin",
            "format": "raw-shard",
            "nbytes": 33428480,
            "records": [
                {
                    "name": "unet_1155",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1156",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1157",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1159",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1160",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1161",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30146560
                },
                {
                    "name": "unet_1162",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30149120
                },
                {
                    "name": "unet_1163",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33425920
                }
            ]
        },
        {
            "dataPath": "params_shard_74.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1173",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_75.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1164",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1165",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1166",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1167",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1168",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1169",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1170",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1171",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 24903680
                },
                {
                    "name": "unet_1172",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_76.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1174",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1175",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1176",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1177",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1178",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1179",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_77.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1183",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_78.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1180",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1181",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1182",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1184",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1185",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1186",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_79.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1193",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_80.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1187",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1188",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1189",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1190",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1191",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1192",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_81.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1194",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1195",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1196",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1197",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1198",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1199",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_82.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1203",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_83.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1200",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1201",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1202",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1204",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1205",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1206",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_84.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1213",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_85.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1207",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1208",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1209",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1210",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1211",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1212",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_86.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1214",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1215",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1216",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1217",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1218",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1219",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_87.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1223",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_88.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1220",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1221",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1222",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1224",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1225",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1226",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_89.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1233",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_90.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1227",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1228",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1229",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1230",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1231",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1232",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_91.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1234",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1235",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1236",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1237",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1238",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1239",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_92.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1243",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_93.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1240",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1241",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1242",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1244",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1245",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1246",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_94.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1253",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_95.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1247",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1248",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1249",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1250",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1251",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1252",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_96.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1254",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1255",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1256",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1257",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1258",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1259",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_97.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1263",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_98.bin",
            "format": "raw-shard",
            "nbytes": 33431040,
            "records": [
                {
                    "name": "unet_1260",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1261",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1262",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1264",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1265",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1266",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30146560
                },
                {
                    "name": "unet_1267",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30149120
                },
                {
                    "name": "unet_1268",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33425920
                },
                {
                    "name": "unet_1269",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33428480
                }
            ]
        },
        {
            "dataPath": "params_shard_99.bin",
            "format": "raw-shard",
            "nbytes": 33428480,
            "records": [
                {
                    "name": "unet_1270",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1271",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1272",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 3279360
                },
                {
                    "name": "unet_1273",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3281920
                },
                {
                    "name": "unet_1274",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6558720
                },
                {
                    "name": "unet_1275",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9835520
                },
                {
                    "name": "unet_1276",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13112320
                },
                {
                    "name": "unet_1277",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16389120
                },
                {
                    "name": "unet_1278",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19665920
                },
                {
                    "name": "unet_1279",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 22942720
                },
                {
                    "name": "unet_1280",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 28185600
                }
            ]
        },
        {
            "dataPath": "params_shard_100.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1281",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1282",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 3276800
                }
            ]
        },
        {
            "dataPath": "params_shard_101.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1283",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1284",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1285",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1286",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1287",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1288",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_102.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1292",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_103.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1289",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1290",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1291",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1293",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1294",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1295",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_104.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1302",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_105.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1296",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1297",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1298",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1299",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1300",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1301",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_106.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1303",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1304",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1305",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1306",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1307",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1308",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_107.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1312",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_108.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1309",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1310",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1311",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1313",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1314",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1315",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_109.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1322",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_110.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1316",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1317",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1318",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1319",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1320",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1321",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_111.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1323",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1324",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1325",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1326",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1327",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1328",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_112.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1332",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_113.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1329",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1330",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1331",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1333",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1334",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1335",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_114.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1342",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_115.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1336",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1337",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1338",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1339",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1340",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1341",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_116.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1343",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1344",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1345",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1346",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1347",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1348",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_117.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1352",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_118.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1349",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1350",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1351",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1353",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1354",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1355",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_119.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1362",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_120.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1356",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1357",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1358",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1359",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1360",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1361",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_121.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1363",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1364",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1365",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1366",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1367",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1368",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_122.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1372",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_123.bin",
            "format": "raw-shard",
            "nbytes": 33431040,
            "records": [
                {
                    "name": "unet_1369",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1370",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1371",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1373",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1374",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1375",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30146560
                },
                {
                    "name": "unet_1376",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30149120
                },
                {
                    "name": "unet_1377",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33425920
                },
                {
                    "name": "unet_1378",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33428480
                }
            ]
        },
        {
            "dataPath": "params_shard_124.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1388",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_125.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1379",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1380",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1381",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1382",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1383",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1384",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1385",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1386",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 24903680
                },
                {
                    "name": "unet_1387",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_126.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1389",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1390",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1391",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1392",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1393",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1394",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_127.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1398",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_128.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1395",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1396",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1397",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1399",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1400",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1401",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_129.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1408",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_130.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1402",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1403",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1404",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1405",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1406",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1407",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_131.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1409",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1410",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1411",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1412",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1413",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1414",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_132.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1418",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_133.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1415",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1416",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1417",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1419",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1420",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1421",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_134.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1428",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_135.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1422",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1423",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1424",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1425",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1426",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1427",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_136.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1429",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1430",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1431",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1432",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1433",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1434",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_137.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1438",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_138.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1435",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1436",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1437",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1439",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1440",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1441",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_139.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1448",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_140.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1442",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1443",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1444",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1445",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1446",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1447",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_141.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1449",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1450",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1451",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1452",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1453",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1454",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_142.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1458",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_143.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1455",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1456",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1457",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1459",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1460",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1461",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_144.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1468",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_145.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1462",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1463",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1464",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1465",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1466",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1467",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_146.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1469",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1470",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1471",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1472",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1473",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1474",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_147.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1478",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_148.bin",
            "format": "raw-shard",
            "nbytes": 33431040,
            "records": [
                {
                    "name": "unet_1475",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1476",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1477",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1479",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1480",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1481",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30146560
                },
                {
                    "name": "unet_1482",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30149120
                },
                {
                    "name": "unet_1483",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33425920
                },
                {
                    "name": "unet_1484",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 33428480
                }
            ]
        },
        {
            "dataPath": "params_shard_149.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1494",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_150.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1485",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1486",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1487",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1488",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1489",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1490",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1491",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1492",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 24903680
                },
                {
                    "name": "unet_1493",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_151.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1495",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1496",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1497",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1498",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1499",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1500",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_152.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1504",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_153.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1501",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1502",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1503",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1505",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1506",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1507",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_154.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1514",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_155.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1508",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1509",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1510",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1511",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1512",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1513",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_156.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1515",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1516",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1517",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1518",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1519",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1520",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_157.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1524",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_158.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1521",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1522",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1523",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1525",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1526",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1527",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_159.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1534",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_160.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1528",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1529",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1530",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1531",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1532",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1533",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_161.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1535",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1536",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1537",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1538",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1539",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1540",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_162.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1544",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_163.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1541",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1542",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1543",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1545",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1546",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1547",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_164.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1554",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_165.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1548",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1549",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1550",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1551",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1552",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1553",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_166.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1555",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1556",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1557",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1558",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1559",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1560",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_167.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1564",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_168.bin",
            "format": "raw-shard",
            "nbytes": 33423360,
            "records": [
                {
                    "name": "unet_1561",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1562",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1563",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1565",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1566",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1567",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 30146560
                }
            ]
        },
        {
            "dataPath": "params_shard_169.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1574",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_170.bin",
            "format": "raw-shard",
            "nbytes": 23592960,
            "records": [
                {
                    "name": "unet_1568",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1569",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1570",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1571",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1572",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 15073280
                },
                {
                    "name": "unet_1573",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 20316160
                }
            ]
        },
        {
            "dataPath": "params_shard_171.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "unet_1575",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1576",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1577",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "unet_1578",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "unet_1579",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                },
                {
                    "name": "unet_1580",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        },
        {
            "dataPath": "params_shard_172.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "unet_1584",
                    "shape": [
                        1280,
                        10240
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 26214400,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_173.bin",
            "format": "raw-shard",
            "nbytes": 33429760,
            "records": [
                {
                    "name": "unet_1581",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1582",
                    "shape": [
                        2048,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 5242880,
                    "byteOffset": 5242880
                },
                {
                    "name": "unet_1583",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 10485760
                },
                {
                    "name": "unet_1585",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13762560
                },
                {
                    "name": "unet_1586",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26869760
                },
                {
                    "name": "unet_1587",
                    "shape": [
                        1,
                        1280,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 30146560
                },
                {
                    "name": "unet_1588",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 30149120
                },
                {
                    "name": "unet_1589",
                    "shape": [
                        1280,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 30150400
                },
                {
                    "name": "unet_1590",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 31788800
                },
                {
                    "name": "unet_1591",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 31790080
                },
                {
                    "name": "unet_1592",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 31791360
                },
                {
                    "name": "unet_1593",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 32610560
                }
            ]
        },
        {
            "dataPath": "params_shard_174.bin",
            "format": "raw-shard",
            "nbytes": 29327360,
            "records": [
                {
                    "name": "unet_1594",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1595",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 819200
                },
                {
                    "name": "unet_1596",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 1638400
                },
                {
                    "name": "unet_1597",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 2457600
                },
                {
                    "name": "unet_1598",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1599",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 5898240
                },
                {
                    "name": "unet_1600",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 8519680
                },
                {
                    "name": "unet_1601",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 9338880
                },
                {
                    "name": "unet_1602",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 15892480
                },
                {
                    "name": "unet_1603",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 19169280
                },
                {
                    "name": "unet_1604",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 19988480
                },
                {
                    "name": "unet_1605",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 20807680
                },
                {
                    "name": "unet_1606",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 21626880
                },
                {
                    "name": "unet_1607",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 22446080
                },
                {
                    "name": "unet_1608",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 23265280
                },
                {
                    "name": "unet_1609",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 25886720
                },
                {
                    "name": "unet_1610",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 28508160
                }
            ]
        },
        {
            "dataPath": "params_shard_175.bin",
            "format": "raw-shard",
            "nbytes": 33099520,
            "records": [
                {
                    "name": "unet_1611",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1612",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1613",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1614",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 10649600
                },
                {
                    "name": "unet_1615",
                    "shape": [
                        1280,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 10650880
                },
                {
                    "name": "unet_1616",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 12289280
                },
                {
                    "name": "unet_1617",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 12290560
                },
                {
                    "name": "unet_1618",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 12291840
                },
                {
                    "name": "unet_1619",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 13111040
                },
                {
                    "name": "unet_1620",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 13930240
                },
                {
                    "name": "unet_1621",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 14749440
                },
                {
                    "name": "unet_1622",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 15568640
                },
                {
                    "name": "unet_1623",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 16387840
                },
                {
                    "name": "unet_1624",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 17207040
                },
                {
                    "name": "unet_1625",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 19828480
                },
                {
                    "name": "unet_1626",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 22449920
                },
                {
                    "name": "unet_1627",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 23269120
                },
                {
                    "name": "unet_1628",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29822720
                }
            ]
        },
        {
            "dataPath": "params_shard_176.bin",
            "format": "raw-shard",
            "nbytes": 33427200,
            "records": [
                {
                    "name": "unet_1629",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1630",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 819200
                },
                {
                    "name": "unet_1631",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 1638400
                },
                {
                    "name": "unet_1632",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 2457600
                },
                {
                    "name": "unet_1633",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 3276800
                },
                {
                    "name": "unet_1634",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 4096000
                },
                {
                    "name": "unet_1635",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 6717440
                },
                {
                    "name": "unet_1636",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 9338880
                },
                {
                    "name": "unet_1637",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 10158080
                },
                {
                    "name": "unet_1638",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16711680
                },
                {
                    "name": "unet_1639",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 19988480
                },
                {
                    "name": "unet_1640",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 20807680
                },
                {
                    "name": "unet_1641",
                    "shape": [
                        1280,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1638400,
                    "byteOffset": 20808960
                },
                {
                    "name": "unet_1642",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 22447360
                },
                {
                    "name": "unet_1643",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 22448640
                },
                {
                    "name": "unet_1644",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 22449920
                },
                {
                    "name": "unet_1645",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 23269120
                },
                {
                    "name": "unet_1646",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 24088320
                },
                {
                    "name": "unet_1647",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 24907520
                },
                {
                    "name": "unet_1648",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 25726720
                },
                {
                    "name": "unet_1649",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 26545920
                },
                {
                    "name": "unet_1650",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 27365120
                },
                {
                    "name": "unet_1651",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 29986560
                },
                {
                    "name": "unet_1652",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 32608000
                }
            ]
        },
        {
            "dataPath": "params_shard_177.bin",
            "format": "raw-shard",
            "nbytes": 33153160,
            "records": [
                {
                    "name": "unet_1653",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 0
                },
                {
                    "name": "unet_1654",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "unet_1655",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 9830400
                },
                {
                    "name": "unet_1656",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 10649600
                },
                {
                    "name": "unet_1657",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 11468800
                },
                {
                    "name": "unet_1658",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 12288000
                },
                {
                    "name": "unet_1659",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 13107200
                },
                {
                    "name": "unet_1660",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 13926400
                },
                {
                    "name": "unet_1661",
                    "shape": [
                        2048,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2621440,
                    "byteOffset": 16547840
                },
                {
                    "name": "unet_1662",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 19169280
                },
                {
                    "name": "unet_1663",
                    "shape": [
                        640,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6553600,
                    "byteOffset": 19988480
                },
                {
                    "name": "unet_1664",
                    "shape": [
                        2560,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26542080
                },
                {
                    "name": "unet_1665",
                    "shape": [
                        640,
                        640
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 29818880
                },
                {
                    "name": "unet_1666",
                    "shape": [
                        1,
                        640,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1280,
                    "byteOffset": 30638080
                },
                {
                    "name": "unet_1667",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 30639360
                },
                {
                    "name": "unet_1668",
                    "shape": [
                        1280,
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 30640000
                },
                {
                    "name": "unet_1669",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 31459200
                },
                {
                    "name": "unet_1670",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 31459840
                },
                {
                    "name": "unet_1671",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 31460480
                },
                {
                    "name": "unet_1672",
                    "shape": [
                        1280,
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 31461120
                },
                {
                    "name": "unet_1673",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 32280320
                },
                {
                    "name": "unet_1674",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 32280960
                },
                {
                    "name": "unet_1675",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 32281600
                },
                {
                    "name": "unet_1676",
                    "shape": [
                        1280,
                        320
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 819200,
                    "byteOffset": 32282240
                },
                {
                    "name": "unet_1677",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 33101440
                },
                {
                    "name": "unet_1678",
                    "shape": [
                        1,
                        320,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 640,
                    "byteOffset": 33102080
                },
                {
                    "name": "unet_1679",
                    "shape": [
                        1,
                        4,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 8,
                    "byteOffset": 33102720
                },
                {
                    "name": "vae_0",
                    "shape": [
                        512,
                        4,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 36864,
                    "byteOffset": 33102728
                },
                {
                    "name": "vae_1",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 33139592
                },
                {
                    "name": "vae_2",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 33139848
                },
                {
                    "name": "vae_3",
                    "shape": [
                        3,
                        128,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6912,
                    "byteOffset": 33140104
                },
                {
                    "name": "vae_4",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 33147016
                },
                {
                    "name": "vae_5",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 33148040
                },
                {
                    "name": "vae_6",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 33149064
                },
                {
                    "name": "vae_7",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 33150088
                },
                {
                    "name": "vae_8",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 33151112
                },
                {
                    "name": "vae_9",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 33152136
                }
            ]
        },
        {
            "dataPath": "params_shard_178.bin",
            "format": "raw-shard",
            "nbytes": 33042432,
            "records": [
                {
                    "name": "vae_10",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "vae_11",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4718592
                },
                {
                    "name": "vae_12",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 9437184
                },
                {
                    "name": "vae_13",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 9438208
                },
                {
                    "name": "vae_14",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 9439232
                },
                {
                    "name": "vae_15",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 9440256
                },
                {
                    "name": "vae_16",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 9441280
                },
                {
                    "name": "vae_17",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 14159872
                },
                {
                    "name": "vae_18",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 18878464
                },
                {
                    "name": "vae_19",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 18879488
                },
                {
                    "name": "vae_20",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 18880512
                },
                {
                    "name": "vae_21",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 18881536
                },
                {
                    "name": "vae_22",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 18882560
                },
                {
                    "name": "vae_23",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 23601152
                },
                {
                    "name": "vae_24",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28319744
                },
                {
                    "name": "vae_25",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28320768
                },
                {
                    "name": "vae_26",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28321792
                },
                {
                    "name": "vae_27",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28322816
                },
                {
                    "name": "vae_28",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 28323840
                }
            ]
        },
        {
            "dataPath": "params_shard_179.bin",
            "format": "raw-shard",
            "nbytes": 33042432,
            "records": [
                {
                    "name": "vae_29",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "vae_30",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4718592
                },
                {
                    "name": "vae_31",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4719616
                },
                {
                    "name": "vae_32",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4720640
                },
                {
                    "name": "vae_33",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4721664
                },
                {
                    "name": "vae_34",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4722688
                },
                {
                    "name": "vae_35",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 9441280
                },
                {
                    "name": "vae_36",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14159872
                },
                {
                    "name": "vae_37",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14160896
                },
                {
                    "name": "vae_38",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14161920
                },
                {
                    "name": "vae_39",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14162944
                },
                {
                    "name": "vae_40",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 14163968
                },
                {
                    "name": "vae_41",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 18882560
                },
                {
                    "name": "vae_42",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 23601152
                },
                {
                    "name": "vae_43",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28319744
                },
                {
                    "name": "vae_44",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28320768
                },
                {
                    "name": "vae_45",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28321792
                },
                {
                    "name": "vae_46",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 28322816
                },
                {
                    "name": "vae_47",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 28323840
                }
            ]
        },
        {
            "dataPath": "params_shard_180.bin",
            "format": "raw-shard",
            "nbytes": 75890688,
            "records": [
                {
                    "name": "clip_0",
                    "shape": [
                        49408,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 75890688,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_181.bin",
            "format": "raw-shard",
            "nbytes": 33206062,
            "records": [
                {
                    "name": "vae_48",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "vae_49",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4718592
                },
                {
                    "name": "vae_50",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4719616
                },
                {
                    "name": "vae_51",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4720640
                },
                {
                    "name": "vae_52",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 4721664
                },
                {
                    "name": "vae_53",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4722688
                },
                {
                    "name": "vae_54",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 9441280
                },
                {
                    "name": "vae_55",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14159872
                },
                {
                    "name": "vae_56",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14160896
                },
                {
                    "name": "vae_57",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14161920
                },
                {
                    "name": "vae_58",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 14162944
                },
                {
                    "name": "vae_59",
                    "shape": [
                        512,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 14163968
                },
                {
                    "name": "vae_60",
                    "shape": [
                        256,
                        512,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2359296,
                    "byteOffset": 18882560
                },
                {
                    "name": "vae_61",
                    "shape": [
                        256,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 21241856
                },
                {
                    "name": "vae_62",
                    "shape": [
                        256,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 262144,
                    "byteOffset": 22421504
                },
                {
                    "name": "vae_63",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 22683648
                },
                {
                    "name": "vae_64",
                    "shape": [
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 22684672
                },
                {
                    "name": "vae_65",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 22685696
                },
                {
                    "name": "vae_66",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 22686208
                },
                {
                    "name": "vae_67",
                    "shape": [
                        256,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 22686720
                },
                {
                    "name": "vae_68",
                    "shape": [
                        256,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 23866368
                },
                {
                    "name": "vae_69",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 25046016
                },
                {
                    "name": "vae_70",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 25046528
                },
                {
                    "name": "vae_71",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 25047040
                },
                {
                    "name": "vae_72",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 25047552
                },
                {
                    "name": "vae_73",
                    "shape": [
                        256,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 25048064
                },
                {
                    "name": "vae_74",
                    "shape": [
                        256,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 26227712
                },
                {
                    "name": "vae_75",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 27407360
                },
                {
                    "name": "vae_76",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 27407872
                },
                {
                    "name": "vae_77",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 27408384
                },
                {
                    "name": "vae_78",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 27408896
                },
                {
                    "name": "vae_79",
                    "shape": [
                        256,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 27409408
                },
                {
                    "name": "vae_80",
                    "shape": [
                        128,
                        256,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 589824,
                    "byteOffset": 28589056
                },
                {
                    "name": "vae_81",
                    "shape": [
                        128,
                        128,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 294912,
                    "byteOffset": 29178880
                },
                {
                    "name": "vae_82",
                    "shape": [
                        128,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 65536,
                    "byteOffset": 29473792
                },
                {
                    "name": "vae_83",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 29539328
                },
                {
                    "name": "vae_84",
                    "shape": [
                        256
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 29539840
                },
                {
                    "name": "vae_85",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 29540352
                },
                {
                    "name": "vae_86",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 29540608
                },
                {
                    "name": "vae_87",
                    "shape": [
                        128,
                        128,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 294912,
                    "byteOffset": 29540864
                },
                {
                    "name": "vae_88",
                    "shape": [
                        128,
                        128,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 294912,
                    "byteOffset": 29835776
                },
                {
                    "name": "vae_89",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30130688
                },
                {
                    "name": "vae_90",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30130944
                },
                {
                    "name": "vae_91",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30131200
                },
                {
                    "name": "vae_92",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30131456
                },
                {
                    "name": "vae_93",
                    "shape": [
                        128,
                        128,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 294912,
                    "byteOffset": 30131712
                },
                {
                    "name": "vae_94",
                    "shape": [
                        128,
                        128,
                        3,
                        3
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 294912,
                    "byteOffset": 30426624
                },
                {
                    "name": "vae_95",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30721536
                },
                {
                    "name": "vae_96",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30721792
                },
                {
                    "name": "vae_97",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30722048
                },
                {
                    "name": "vae_98",
                    "shape": [
                        128
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 30722304
                },
                {
                    "name": "vae_99",
                    "shape": [
                        4,
                        4,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 32,
                    "byteOffset": 30722560
                },
                {
                    "name": "vae_100",
                    "shape": [
                        1,
                        4,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 8,
                    "byteOffset": 30722592
                },
                {
                    "name": "vae_101",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 30722600
                },
                {
                    "name": "vae_102",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 30723624
                },
                {
                    "name": "vae_103",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 30724648
                },
                {
                    "name": "vae_104",
                    "shape": [
                        512,
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 524288,
                    "byteOffset": 30725672
                },
                {
                    "name": "vae_105",
                    "shape": [
                        512,
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 524288,
                    "byteOffset": 31249960
                },
                {
                    "name": "vae_106",
                    "shape": [
                        512,
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 524288,
                    "byteOffset": 31774248
                },
                {
                    "name": "vae_107",
                    "shape": [
                        512,
                        512
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 524288,
                    "byteOffset": 32298536
                },
                {
                    "name": "vae_108",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32822824
                },
                {
                    "name": "vae_109",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32823848
                },
                {
                    "name": "vae_110",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32824872
                },
                {
                    "name": "vae_111",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32825896
                },
                {
                    "name": "vae_112",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32826920
                },
                {
                    "name": "vae_113",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32827944
                },
                {
                    "name": "vae_114",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32828968
                },
                {
                    "name": "vae_115",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32829992
                },
                {
                    "name": "vae_116",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32831016
                },
                {
                    "name": "vae_117",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32832040
                },
                {
                    "name": "vae_118",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32833064
                },
                {
                    "name": "vae_119",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32834088
                },
                {
                    "name": "vae_120",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32835112
                },
                {
                    "name": "vae_121",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32836136
                },
                {
                    "name": "vae_122",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32837160
                },
                {
                    "name": "vae_123",
                    "shape": [
                        1,
                        512,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1024,
                    "byteOffset": 32838184
                },
                {
                    "name": "vae_124",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32839208
                },
                {
                    "name": "vae_125",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32839720
                },
                {
                    "name": "vae_126",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32840232
                },
                {
                    "name": "vae_127",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32840744
                },
                {
                    "name": "vae_128",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32841256
                },
                {
                    "name": "vae_129",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32841768
                },
                {
                    "name": "vae_130",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32842280
                },
                {
                    "name": "vae_131",
                    "shape": [
                        1,
                        256,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 512,
                    "byteOffset": 32842792
                },
                {
                    "name": "vae_132",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32843304
                },
                {
                    "name": "vae_133",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32843560
                },
                {
                    "name": "vae_134",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32843816
                },
                {
                    "name": "vae_135",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32844072
                },
                {
                    "name": "vae_136",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32844328
                },
                {
                    "name": "vae_137",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32844584
                },
                {
                    "name": "vae_138",
                    "shape": [
                        1,
                        128,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 256,
                    "byteOffset": 32844840
                },
                {
                    "name": "vae_139",
                    "shape": [
                        1,
                        3,
                        1,
                        1
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6,
                    "byteOffset": 32845096
                },
                {
                    "name": "clip_1",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32845102
                },
                {
                    "name": "clip_2",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32846638
                },
                {
                    "name": "clip_3",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32848174
                },
                {
                    "name": "clip_4",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32849710
                },
                {
                    "name": "clip_5",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32851246
                },
                {
                    "name": "clip_6",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32857390
                },
                {
                    "name": "clip_7",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32858926
                },
                {
                    "name": "clip_8",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32860462
                },
                {
                    "name": "clip_9",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32861998
                },
                {
                    "name": "clip_10",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32863534
                },
                {
                    "name": "clip_11",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32865070
                },
                {
                    "name": "clip_12",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32866606
                },
                {
                    "name": "clip_13",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32868142
                },
                {
                    "name": "clip_14",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32869678
                },
                {
                    "name": "clip_15",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32871214
                },
                {
                    "name": "clip_16",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32877358
                },
                {
                    "name": "clip_17",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32878894
                },
                {
                    "name": "clip_18",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32880430
                },
                {
                    "name": "clip_19",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32881966
                },
                {
                    "name": "clip_20",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32883502
                },
                {
                    "name": "clip_21",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32885038
                },
                {
                    "name": "clip_22",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32886574
                },
                {
                    "name": "clip_23",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32888110
                },
                {
                    "name": "clip_24",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32889646
                },
                {
                    "name": "clip_25",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32891182
                },
                {
                    "name": "clip_26",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32897326
                },
                {
                    "name": "clip_27",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32898862
                },
                {
                    "name": "clip_28",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32900398
                },
                {
                    "name": "clip_29",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32901934
                },
                {
                    "name": "clip_30",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32903470
                },
                {
                    "name": "clip_31",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32905006
                },
                {
                    "name": "clip_32",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32906542
                },
                {
                    "name": "clip_33",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32908078
                },
                {
                    "name": "clip_34",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32909614
                },
                {
                    "name": "clip_35",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32911150
                },
                {
                    "name": "clip_36",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32917294
                },
                {
                    "name": "clip_37",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32918830
                },
                {
                    "name": "clip_38",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32920366
                },
                {
                    "name": "clip_39",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32921902
                },
                {
                    "name": "clip_40",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32923438
                },
                {
                    "name": "clip_41",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32924974
                },
                {
                    "name": "clip_42",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32926510
                },
                {
                    "name": "clip_43",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32928046
                },
                {
                    "name": "clip_44",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32929582
                },
                {
                    "name": "clip_45",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32931118
                },
                {
                    "name": "clip_46",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32937262
                },
                {
                    "name": "clip_47",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32938798
                },
                {
                    "name": "clip_48",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32940334
                },
                {
                    "name": "clip_49",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32941870
                },
                {
                    "name": "clip_50",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32943406
                },
                {
                    "name": "clip_51",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32944942
                },
                {
                    "name": "clip_52",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32946478
                },
                {
                    "name": "clip_53",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32948014
                },
                {
                    "name": "clip_54",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32949550
                },
                {
                    "name": "clip_55",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32951086
                },
                {
                    "name": "clip_56",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32957230
                },
                {
                    "name": "clip_57",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32958766
                },
                {
                    "name": "clip_58",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32960302
                },
                {
                    "name": "clip_59",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32961838
                },
                {
                    "name": "clip_60",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32963374
                },
                {
                    "name": "clip_61",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32964910
                },
                {
                    "name": "clip_62",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32966446
                },
                {
                    "name": "clip_63",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32967982
                },
                {
                    "name": "clip_64",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32969518
                },
                {
                    "name": "clip_65",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32971054
                },
                {
                    "name": "clip_66",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32977198
                },
                {
                    "name": "clip_67",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32978734
                },
                {
                    "name": "clip_68",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32980270
                },
                {
                    "name": "clip_69",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32981806
                },
                {
                    "name": "clip_70",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32983342
                },
                {
                    "name": "clip_71",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32984878
                },
                {
                    "name": "clip_72",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32986414
                },
                {
                    "name": "clip_73",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32987950
                },
                {
                    "name": "clip_74",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32989486
                },
                {
                    "name": "clip_75",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 32991022
                },
                {
                    "name": "clip_76",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32997166
                },
                {
                    "name": "clip_77",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 32998702
                },
                {
                    "name": "clip_78",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33000238
                },
                {
                    "name": "clip_79",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33001774
                },
                {
                    "name": "clip_80",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33003310
                },
                {
                    "name": "clip_81",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33004846
                },
                {
                    "name": "clip_82",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33006382
                },
                {
                    "name": "clip_83",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33007918
                },
                {
                    "name": "clip_84",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33009454
                },
                {
                    "name": "clip_85",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 33010990
                },
                {
                    "name": "clip_86",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33017134
                },
                {
                    "name": "clip_87",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33018670
                },
                {
                    "name": "clip_88",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33020206
                },
                {
                    "name": "clip_89",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33021742
                },
                {
                    "name": "clip_90",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33023278
                },
                {
                    "name": "clip_91",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33024814
                },
                {
                    "name": "clip_92",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33026350
                },
                {
                    "name": "clip_93",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33027886
                },
                {
                    "name": "clip_94",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33029422
                },
                {
                    "name": "clip_95",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 33030958
                },
                {
                    "name": "clip_96",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33037102
                },
                {
                    "name": "clip_97",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33038638
                },
                {
                    "name": "clip_98",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33040174
                },
                {
                    "name": "clip_99",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33041710
                },
                {
                    "name": "clip_100",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33043246
                },
                {
                    "name": "clip_101",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33044782
                },
                {
                    "name": "clip_102",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33046318
                },
                {
                    "name": "clip_103",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33047854
                },
                {
                    "name": "clip_104",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33049390
                },
                {
                    "name": "clip_105",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 33050926
                },
                {
                    "name": "clip_106",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33057070
                },
                {
                    "name": "clip_107",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33058606
                },
                {
                    "name": "clip_108",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33060142
                },
                {
                    "name": "clip_109",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33061678
                },
                {
                    "name": "clip_110",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33063214
                },
                {
                    "name": "clip_111",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33064750
                },
                {
                    "name": "clip_112",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33066286
                },
                {
                    "name": "clip_113",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33067822
                },
                {
                    "name": "clip_114",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33069358
                },
                {
                    "name": "clip_115",
                    "shape": [
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 6144,
                    "byteOffset": 33070894
                },
                {
                    "name": "clip_116",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33077038
                },
                {
                    "name": "clip_117",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33078574
                },
                {
                    "name": "clip_118",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33080110
                },
                {
                    "name": "clip_119",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33081646
                },
                {
                    "name": "clip_120",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33083182
                },
                {
                    "name": "clip_121",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33084718
                },
                {
                    "name": "clip_122",
                    "shape": [
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1536,
                    "byteOffset": 33086254
                },
                {
                    "name": "clip_123",
                    "shape": [
                        77,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 118272,
                    "byteOffset": 33087790
                }
            ]
        },
        {
            "dataPath": "params_shard_182.bin",
            "format": "raw-shard",
            "nbytes": 33030144,
            "records": [
                {
                    "name": "clip_124",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 0
                },
                {
                    "name": "clip_125",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 1179648
                },
                {
                    "name": "clip_126",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 2359296
                },
                {
                    "name": "clip_127",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 3538944
                },
                {
                    "name": "clip_128",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4718592
                },
                {
                    "name": "clip_129",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 9437184
                },
                {
                    "name": "clip_130",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 14155776
                },
                {
                    "name": "clip_131",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 15335424
                },
                {
                    "name": "clip_132",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 16515072
                },
                {
                    "name": "clip_133",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 17694720
                },
                {
                    "name": "clip_134",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 18874368
                },
                {
                    "name": "clip_135",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 23592960
                },
                {
                    "name": "clip_136",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 28311552
                },
                {
                    "name": "clip_137",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 29491200
                },
                {
                    "name": "clip_138",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 30670848
                },
                {
                    "name": "clip_139",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 31850496
                }
            ]
        },
        {
            "dataPath": "params_shard_183.bin",
            "format": "raw-shard",
            "nbytes": 33030144,
            "records": [
                {
                    "name": "clip_140",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "clip_141",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4718592
                },
                {
                    "name": "clip_142",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 9437184
                },
                {
                    "name": "clip_143",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 10616832
                },
                {
                    "name": "clip_144",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 11796480
                },
                {
                    "name": "clip_145",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 12976128
                },
                {
                    "name": "clip_146",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 14155776
                },
                {
                    "name": "clip_147",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 18874368
                },
                {
                    "name": "clip_148",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 23592960
                },
                {
                    "name": "clip_149",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 24772608
                },
                {
                    "name": "clip_150",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 25952256
                },
                {
                    "name": "clip_151",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 27131904
                },
                {
                    "name": "clip_152",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 28311552
                }
            ]
        },
        {
            "dataPath": "params_shard_184.bin",
            "format": "raw-shard",
            "nbytes": 33030144,
            "records": [
                {
                    "name": "clip_153",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "clip_154",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 4718592
                },
                {
                    "name": "clip_155",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 5898240
                },
                {
                    "name": "clip_156",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 7077888
                },
                {
                    "name": "clip_157",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 8257536
                },
                {
                    "name": "clip_158",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 9437184
                },
                {
                    "name": "clip_159",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 14155776
                },
                {
                    "name": "clip_160",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 18874368
                },
                {
                    "name": "clip_161",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 20054016
                },
                {
                    "name": "clip_162",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 21233664
                },
                {
                    "name": "clip_163",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 22413312
                },
                {
                    "name": "clip_164",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 23592960
                },
                {
                    "name": "clip_165",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 28311552
                }
            ]
        },
        {
            "dataPath": "params_shard_185.bin",
            "format": "raw-shard",
            "nbytes": 33030144,
            "records": [
                {
                    "name": "clip_166",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 0
                },
                {
                    "name": "clip_167",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 1179648
                },
                {
                    "name": "clip_168",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 2359296
                },
                {
                    "name": "clip_169",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 3538944
                },
                {
                    "name": "clip_170",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4718592
                },
                {
                    "name": "clip_171",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 9437184
                },
                {
                    "name": "clip_172",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 14155776
                },
                {
                    "name": "clip_173",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 15335424
                },
                {
                    "name": "clip_174",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 16515072
                },
                {
                    "name": "clip_175",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 17694720
                },
                {
                    "name": "clip_176",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 18874368
                },
                {
                    "name": "clip_177",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 23592960
                },
                {
                    "name": "clip_178",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 28311552
                },
                {
                    "name": "clip_179",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 29491200
                },
                {
                    "name": "clip_180",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 30670848
                },
                {
                    "name": "clip_181",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 31850496
                }
            ]
        },
        {
            "dataPath": "params_shard_186.bin",
            "format": "raw-shard",
            "nbytes": 33030144,
            "records": [
                {
                    "name": "clip_182",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "clip_183",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 4718592
                },
                {
                    "name": "clip_184",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 9437184
                },
                {
                    "name": "clip_185",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 10616832
                },
                {
                    "name": "clip_186",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 11796480
                },
                {
                    "name": "clip_187",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 12976128
                },
                {
                    "name": "clip_188",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 14155776
                },
                {
                    "name": "clip_189",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 18874368
                },
                {
                    "name": "clip_190",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 23592960
                },
                {
                    "name": "clip_191",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 24772608
                },
                {
                    "name": "clip_192",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 25952256
                },
                {
                    "name": "clip_193",
                    "shape": [
                        768,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 1179648,
                    "byteOffset": 27131904
                },
                {
                    "name": "clip_194",
                    "shape": [
                        768,
                        3072
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 28311552
                }
            ]
        },
        {
            "dataPath": "params_shard_187.bin",
            "format": "raw-shard",
            "nbytes": 126484480,
            "records": [
                {
                    "name": "clip2_0",
                    "shape": [
                        49408,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 126484480,
                    "byteOffset": 0
                }
            ]
        },
        {
            "dataPath": "params_shard_188.bin",
            "format": "raw-shard",
            "nbytes": 32200192,
            "records": [
                {
                    "name": "clip_195",
                    "shape": [
                        3072,
                        768
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 4718592,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_1",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4718592
                },
                {
                    "name": "clip2_2",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4721152
                },
                {
                    "name": "clip2_3",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4723712
                },
                {
                    "name": "clip2_4",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4726272
                },
                {
                    "name": "clip2_5",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4728832
                },
                {
                    "name": "clip2_6",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4739072
                },
                {
                    "name": "clip2_7",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4741632
                },
                {
                    "name": "clip2_8",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4744192
                },
                {
                    "name": "clip2_9",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4746752
                },
                {
                    "name": "clip2_10",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4749312
                },
                {
                    "name": "clip2_11",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4751872
                },
                {
                    "name": "clip2_12",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4754432
                },
                {
                    "name": "clip2_13",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4756992
                },
                {
                    "name": "clip2_14",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4759552
                },
                {
                    "name": "clip2_15",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4762112
                },
                {
                    "name": "clip2_16",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4772352
                },
                {
                    "name": "clip2_17",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4774912
                },
                {
                    "name": "clip2_18",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4777472
                },
                {
                    "name": "clip2_19",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4780032
                },
                {
                    "name": "clip2_20",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4782592
                },
                {
                    "name": "clip2_21",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4785152
                },
                {
                    "name": "clip2_22",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4787712
                },
                {
                    "name": "clip2_23",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4790272
                },
                {
                    "name": "clip2_24",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4792832
                },
                {
                    "name": "clip2_25",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4795392
                },
                {
                    "name": "clip2_26",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4805632
                },
                {
                    "name": "clip2_27",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4808192
                },
                {
                    "name": "clip2_28",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4810752
                },
                {
                    "name": "clip2_29",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4813312
                },
                {
                    "name": "clip2_30",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4815872
                },
                {
                    "name": "clip2_31",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4818432
                },
                {
                    "name": "clip2_32",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4820992
                },
                {
                    "name": "clip2_33",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4823552
                },
                {
                    "name": "clip2_34",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4826112
                },
                {
                    "name": "clip2_35",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4828672
                },
                {
                    "name": "clip2_36",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4838912
                },
                {
                    "name": "clip2_37",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4841472
                },
                {
                    "name": "clip2_38",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4844032
                },
                {
                    "name": "clip2_39",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4846592
                },
                {
                    "name": "clip2_40",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4849152
                },
                {
                    "name": "clip2_41",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4851712
                },
                {
                    "name": "clip2_42",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4854272
                },
                {
                    "name": "clip2_43",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4856832
                },
                {
                    "name": "clip2_44",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4859392
                },
                {
                    "name": "clip2_45",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4861952
                },
                {
                    "name": "clip2_46",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4872192
                },
                {
                    "name": "clip2_47",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4874752
                },
                {
                    "name": "clip2_48",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4877312
                },
                {
                    "name": "clip2_49",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4879872
                },
                {
                    "name": "clip2_50",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4882432
                },
                {
                    "name": "clip2_51",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4884992
                },
                {
                    "name": "clip2_52",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4887552
                },
                {
                    "name": "clip2_53",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4890112
                },
                {
                    "name": "clip2_54",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4892672
                },
                {
                    "name": "clip2_55",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4895232
                },
                {
                    "name": "clip2_56",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4905472
                },
                {
                    "name": "clip2_57",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4908032
                },
                {
                    "name": "clip2_58",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4910592
                },
                {
                    "name": "clip2_59",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4913152
                },
                {
                    "name": "clip2_60",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4915712
                },
                {
                    "name": "clip2_61",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4918272
                },
                {
                    "name": "clip2_62",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4920832
                },
                {
                    "name": "clip2_63",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4923392
                },
                {
                    "name": "clip2_64",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4925952
                },
                {
                    "name": "clip2_65",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4928512
                },
                {
                    "name": "clip2_66",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4938752
                },
                {
                    "name": "clip2_67",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4941312
                },
                {
                    "name": "clip2_68",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4943872
                },
                {
                    "name": "clip2_69",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4946432
                },
                {
                    "name": "clip2_70",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4948992
                },
                {
                    "name": "clip2_71",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4951552
                },
                {
                    "name": "clip2_72",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4954112
                },
                {
                    "name": "clip2_73",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4956672
                },
                {
                    "name": "clip2_74",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4959232
                },
                {
                    "name": "clip2_75",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4961792
                },
                {
                    "name": "clip2_76",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4972032
                },
                {
                    "name": "clip2_77",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4974592
                },
                {
                    "name": "clip2_78",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4977152
                },
                {
                    "name": "clip2_79",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4979712
                },
                {
                    "name": "clip2_80",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4982272
                },
                {
                    "name": "clip2_81",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4984832
                },
                {
                    "name": "clip2_82",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4987392
                },
                {
                    "name": "clip2_83",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4989952
                },
                {
                    "name": "clip2_84",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 4992512
                },
                {
                    "name": "clip2_85",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 4995072
                },
                {
                    "name": "clip2_86",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5005312
                },
                {
                    "name": "clip2_87",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5007872
                },
                {
                    "name": "clip2_88",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5010432
                },
                {
                    "name": "clip2_89",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5012992
                },
                {
                    "name": "clip2_90",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5015552
                },
                {
                    "name": "clip2_91",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5018112
                },
                {
                    "name": "clip2_92",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5020672
                },
                {
                    "name": "clip2_93",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5023232
                },
                {
                    "name": "clip2_94",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5025792
                },
                {
                    "name": "clip2_95",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5028352
                },
                {
                    "name": "clip2_96",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5038592
                },
                {
                    "name": "clip2_97",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5041152
                },
                {
                    "name": "clip2_98",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5043712
                },
                {
                    "name": "clip2_99",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5046272
                },
                {
                    "name": "clip2_100",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5048832
                },
                {
                    "name": "clip2_101",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5051392
                },
                {
                    "name": "clip2_102",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5053952
                },
                {
                    "name": "clip2_103",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5056512
                },
                {
                    "name": "clip2_104",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5059072
                },
                {
                    "name": "clip2_105",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5061632
                },
                {
                    "name": "clip2_106",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5071872
                },
                {
                    "name": "clip2_107",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5074432
                },
                {
                    "name": "clip2_108",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5076992
                },
                {
                    "name": "clip2_109",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5079552
                },
                {
                    "name": "clip2_110",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5082112
                },
                {
                    "name": "clip2_111",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5084672
                },
                {
                    "name": "clip2_112",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5087232
                },
                {
                    "name": "clip2_113",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5089792
                },
                {
                    "name": "clip2_114",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5092352
                },
                {
                    "name": "clip2_115",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5094912
                },
                {
                    "name": "clip2_116",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5105152
                },
                {
                    "name": "clip2_117",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5107712
                },
                {
                    "name": "clip2_118",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5110272
                },
                {
                    "name": "clip2_119",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5112832
                },
                {
                    "name": "clip2_120",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5115392
                },
                {
                    "name": "clip2_121",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5117952
                },
                {
                    "name": "clip2_122",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5120512
                },
                {
                    "name": "clip2_123",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5123072
                },
                {
                    "name": "clip2_124",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5125632
                },
                {
                    "name": "clip2_125",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5128192
                },
                {
                    "name": "clip2_126",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5138432
                },
                {
                    "name": "clip2_127",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5140992
                },
                {
                    "name": "clip2_128",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5143552
                },
                {
                    "name": "clip2_129",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5146112
                },
                {
                    "name": "clip2_130",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5148672
                },
                {
                    "name": "clip2_131",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5151232
                },
                {
                    "name": "clip2_132",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5153792
                },
                {
                    "name": "clip2_133",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5156352
                },
                {
                    "name": "clip2_134",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5158912
                },
                {
                    "name": "clip2_135",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5161472
                },
                {
                    "name": "clip2_136",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5171712
                },
                {
                    "name": "clip2_137",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5174272
                },
                {
                    "name": "clip2_138",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5176832
                },
                {
                    "name": "clip2_139",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5179392
                },
                {
                    "name": "clip2_140",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5181952
                },
                {
                    "name": "clip2_141",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5184512
                },
                {
                    "name": "clip2_142",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5187072
                },
                {
                    "name": "clip2_143",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5189632
                },
                {
                    "name": "clip2_144",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5192192
                },
                {
                    "name": "clip2_145",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5194752
                },
                {
                    "name": "clip2_146",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5204992
                },
                {
                    "name": "clip2_147",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5207552
                },
                {
                    "name": "clip2_148",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5210112
                },
                {
                    "name": "clip2_149",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5212672
                },
                {
                    "name": "clip2_150",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5215232
                },
                {
                    "name": "clip2_151",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5217792
                },
                {
                    "name": "clip2_152",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5220352
                },
                {
                    "name": "clip2_153",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5222912
                },
                {
                    "name": "clip2_154",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5225472
                },
                {
                    "name": "clip2_155",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5228032
                },
                {
                    "name": "clip2_156",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5238272
                },
                {
                    "name": "clip2_157",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5240832
                },
                {
                    "name": "clip2_158",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5243392
                },
                {
                    "name": "clip2_159",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5245952
                },
                {
                    "name": "clip2_160",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5248512
                },
                {
                    "name": "clip2_161",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5251072
                },
                {
                    "name": "clip2_162",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5253632
                },
                {
                    "name": "clip2_163",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5256192
                },
                {
                    "name": "clip2_164",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5258752
                },
                {
                    "name": "clip2_165",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5261312
                },
                {
                    "name": "clip2_166",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5271552
                },
                {
                    "name": "clip2_167",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5274112
                },
                {
                    "name": "clip2_168",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5276672
                },
                {
                    "name": "clip2_169",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5279232
                },
                {
                    "name": "clip2_170",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5281792
                },
                {
                    "name": "clip2_171",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5284352
                },
                {
                    "name": "clip2_172",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5286912
                },
                {
                    "name": "clip2_173",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5289472
                },
                {
                    "name": "clip2_174",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5292032
                },
                {
                    "name": "clip2_175",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5294592
                },
                {
                    "name": "clip2_176",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5304832
                },
                {
                    "name": "clip2_177",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5307392
                },
                {
                    "name": "clip2_178",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5309952
                },
                {
                    "name": "clip2_179",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5312512
                },
                {
                    "name": "clip2_180",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5315072
                },
                {
                    "name": "clip2_181",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5317632
                },
                {
                    "name": "clip2_182",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5320192
                },
                {
                    "name": "clip2_183",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5322752
                },
                {
                    "name": "clip2_184",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5325312
                },
                {
                    "name": "clip2_185",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5327872
                },
                {
                    "name": "clip2_186",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5338112
                },
                {
                    "name": "clip2_187",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5340672
                },
                {
                    "name": "clip2_188",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5343232
                },
                {
                    "name": "clip2_189",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5345792
                },
                {
                    "name": "clip2_190",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5348352
                },
                {
                    "name": "clip2_191",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5350912
                },
                {
                    "name": "clip2_192",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5353472
                },
                {
                    "name": "clip2_193",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5356032
                },
                {
                    "name": "clip2_194",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5358592
                },
                {
                    "name": "clip2_195",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5361152
                },
                {
                    "name": "clip2_196",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5371392
                },
                {
                    "name": "clip2_197",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5373952
                },
                {
                    "name": "clip2_198",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5376512
                },
                {
                    "name": "clip2_199",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5379072
                },
                {
                    "name": "clip2_200",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5381632
                },
                {
                    "name": "clip2_201",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5384192
                },
                {
                    "name": "clip2_202",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5386752
                },
                {
                    "name": "clip2_203",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5389312
                },
                {
                    "name": "clip2_204",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5391872
                },
                {
                    "name": "clip2_205",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5394432
                },
                {
                    "name": "clip2_206",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5404672
                },
                {
                    "name": "clip2_207",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5407232
                },
                {
                    "name": "clip2_208",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5409792
                },
                {
                    "name": "clip2_209",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5412352
                },
                {
                    "name": "clip2_210",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5414912
                },
                {
                    "name": "clip2_211",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5417472
                },
                {
                    "name": "clip2_212",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5420032
                },
                {
                    "name": "clip2_213",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5422592
                },
                {
                    "name": "clip2_214",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5425152
                },
                {
                    "name": "clip2_215",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5427712
                },
                {
                    "name": "clip2_216",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5437952
                },
                {
                    "name": "clip2_217",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5440512
                },
                {
                    "name": "clip2_218",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5443072
                },
                {
                    "name": "clip2_219",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5445632
                },
                {
                    "name": "clip2_220",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5448192
                },
                {
                    "name": "clip2_221",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5450752
                },
                {
                    "name": "clip2_222",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5453312
                },
                {
                    "name": "clip2_223",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5455872
                },
                {
                    "name": "clip2_224",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5458432
                },
                {
                    "name": "clip2_225",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5460992
                },
                {
                    "name": "clip2_226",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5471232
                },
                {
                    "name": "clip2_227",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5473792
                },
                {
                    "name": "clip2_228",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5476352
                },
                {
                    "name": "clip2_229",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5478912
                },
                {
                    "name": "clip2_230",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5481472
                },
                {
                    "name": "clip2_231",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5484032
                },
                {
                    "name": "clip2_232",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5486592
                },
                {
                    "name": "clip2_233",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5489152
                },
                {
                    "name": "clip2_234",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5491712
                },
                {
                    "name": "clip2_235",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5494272
                },
                {
                    "name": "clip2_236",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5504512
                },
                {
                    "name": "clip2_237",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5507072
                },
                {
                    "name": "clip2_238",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5509632
                },
                {
                    "name": "clip2_239",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5512192
                },
                {
                    "name": "clip2_240",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5514752
                },
                {
                    "name": "clip2_241",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5517312
                },
                {
                    "name": "clip2_242",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5519872
                },
                {
                    "name": "clip2_243",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5522432
                },
                {
                    "name": "clip2_244",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5524992
                },
                {
                    "name": "clip2_245",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5527552
                },
                {
                    "name": "clip2_246",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5537792
                },
                {
                    "name": "clip2_247",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5540352
                },
                {
                    "name": "clip2_248",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5542912
                },
                {
                    "name": "clip2_249",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5545472
                },
                {
                    "name": "clip2_250",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5548032
                },
                {
                    "name": "clip2_251",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5550592
                },
                {
                    "name": "clip2_252",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5553152
                },
                {
                    "name": "clip2_253",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5555712
                },
                {
                    "name": "clip2_254",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5558272
                },
                {
                    "name": "clip2_255",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5560832
                },
                {
                    "name": "clip2_256",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5571072
                },
                {
                    "name": "clip2_257",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5573632
                },
                {
                    "name": "clip2_258",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5576192
                },
                {
                    "name": "clip2_259",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5578752
                },
                {
                    "name": "clip2_260",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5581312
                },
                {
                    "name": "clip2_261",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5583872
                },
                {
                    "name": "clip2_262",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5586432
                },
                {
                    "name": "clip2_263",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5588992
                },
                {
                    "name": "clip2_264",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5591552
                },
                {
                    "name": "clip2_265",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5594112
                },
                {
                    "name": "clip2_266",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5604352
                },
                {
                    "name": "clip2_267",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5606912
                },
                {
                    "name": "clip2_268",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5609472
                },
                {
                    "name": "clip2_269",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5612032
                },
                {
                    "name": "clip2_270",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5614592
                },
                {
                    "name": "clip2_271",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5617152
                },
                {
                    "name": "clip2_272",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5619712
                },
                {
                    "name": "clip2_273",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5622272
                },
                {
                    "name": "clip2_274",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5624832
                },
                {
                    "name": "clip2_275",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5627392
                },
                {
                    "name": "clip2_276",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5637632
                },
                {
                    "name": "clip2_277",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5640192
                },
                {
                    "name": "clip2_278",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5642752
                },
                {
                    "name": "clip2_279",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5645312
                },
                {
                    "name": "clip2_280",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5647872
                },
                {
                    "name": "clip2_281",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5650432
                },
                {
                    "name": "clip2_282",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5652992
                },
                {
                    "name": "clip2_283",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5655552
                },
                {
                    "name": "clip2_284",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5658112
                },
                {
                    "name": "clip2_285",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5660672
                },
                {
                    "name": "clip2_286",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5670912
                },
                {
                    "name": "clip2_287",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5673472
                },
                {
                    "name": "clip2_288",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5676032
                },
                {
                    "name": "clip2_289",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5678592
                },
                {
                    "name": "clip2_290",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5681152
                },
                {
                    "name": "clip2_291",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5683712
                },
                {
                    "name": "clip2_292",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5686272
                },
                {
                    "name": "clip2_293",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5688832
                },
                {
                    "name": "clip2_294",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5691392
                },
                {
                    "name": "clip2_295",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5693952
                },
                {
                    "name": "clip2_296",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5704192
                },
                {
                    "name": "clip2_297",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5706752
                },
                {
                    "name": "clip2_298",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5709312
                },
                {
                    "name": "clip2_299",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5711872
                },
                {
                    "name": "clip2_300",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5714432
                },
                {
                    "name": "clip2_301",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5716992
                },
                {
                    "name": "clip2_302",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5719552
                },
                {
                    "name": "clip2_303",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5722112
                },
                {
                    "name": "clip2_304",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5724672
                },
                {
                    "name": "clip2_305",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5727232
                },
                {
                    "name": "clip2_306",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5737472
                },
                {
                    "name": "clip2_307",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5740032
                },
                {
                    "name": "clip2_308",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5742592
                },
                {
                    "name": "clip2_309",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5745152
                },
                {
                    "name": "clip2_310",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5747712
                },
                {
                    "name": "clip2_311",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5750272
                },
                {
                    "name": "clip2_312",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5752832
                },
                {
                    "name": "clip2_313",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5755392
                },
                {
                    "name": "clip2_314",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5757952
                },
                {
                    "name": "clip2_315",
                    "shape": [
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 10240,
                    "byteOffset": 5760512
                },
                {
                    "name": "clip2_316",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5770752
                },
                {
                    "name": "clip2_317",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5773312
                },
                {
                    "name": "clip2_318",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5775872
                },
                {
                    "name": "clip2_319",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5778432
                },
                {
                    "name": "clip2_320",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5780992
                },
                {
                    "name": "clip2_321",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5783552
                },
                {
                    "name": "clip2_322",
                    "shape": [
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 2560,
                    "byteOffset": 5786112
                },
                {
                    "name": "clip2_323",
                    "shape": [
                        77,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 197120,
                    "byteOffset": 5788672
                },
                {
                    "name": "clip2_324",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 5985792
                },
                {
                    "name": "clip2_325",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9262592
                },
                {
                    "name": "clip2_326",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 12539392
                },
                {
                    "name": "clip2_327",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 15816192
                },
                {
                    "name": "clip2_328",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19092992
                }
            ]
        },
        {
            "dataPath": "params_shard_189.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_329",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_330",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_331",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_332",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_333",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_190.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_334",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_335",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_336",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_337",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_191.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_338",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_339",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_340",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_341",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_192.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_342",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_343",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_344",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_345",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_346",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_193.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_347",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_348",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_349",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_350",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_351",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_194.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_352",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_353",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_354",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_355",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_195.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_356",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_357",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_358",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_359",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_196.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_360",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_361",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_362",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_363",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_364",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_197.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_365",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_366",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_367",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_368",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_369",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_198.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_370",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_371",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_372",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_373",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_199.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_374",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_375",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_376",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_377",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_200.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_378",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_379",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_380",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_381",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_382",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_201.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_383",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_384",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_385",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_386",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_387",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_202.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_388",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_389",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_390",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_391",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_203.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_392",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_393",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_394",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_395",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_204.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_396",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_397",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_398",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_399",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_400",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_205.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_401",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_402",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_403",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_404",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_405",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_206.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_406",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_407",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_408",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_409",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_207.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_410",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_411",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_412",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_413",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_208.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_414",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_415",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_416",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_417",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_418",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_209.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_419",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_420",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_421",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_422",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_423",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_210.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_424",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_425",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_426",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_427",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_211.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_428",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_429",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_430",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_431",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_212.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_432",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_433",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_434",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_435",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_436",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_213.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_437",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_438",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_439",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_440",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_441",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_214.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_442",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_443",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_444",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_445",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_215.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_446",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_447",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_448",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_449",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_216.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_450",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_451",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_452",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_453",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_454",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_217.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_455",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_456",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_457",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_458",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_459",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_218.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_460",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_461",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_462",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_463",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_219.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_464",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_465",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_466",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_467",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_220.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_468",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_469",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_470",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_471",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_472",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_221.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_473",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_474",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_475",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_476",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_477",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_222.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_478",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_479",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_480",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_481",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_223.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_482",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_483",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_484",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_485",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_224.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_486",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_487",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_488",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_489",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_490",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_225.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_491",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_492",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_493",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_494",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_495",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_226.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_496",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_497",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_498",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                },
                {
                    "name": "clip2_499",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 29491200
                }
            ]
        },
        {
            "dataPath": "params_shard_227.bin",
            "format": "raw-shard",
            "nbytes": 32768000,
            "records": [
                {
                    "name": "clip2_500",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_501",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_502",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_503",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 19660800
                }
            ]
        },
        {
            "dataPath": "params_shard_228.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_504",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_505",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 3276800
                },
                {
                    "name": "clip2_506",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 6553600
                },
                {
                    "name": "clip2_507",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 9830400
                },
                {
                    "name": "clip2_508",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                }
            ]
        },
        {
            "dataPath": "params_shard_229.bin",
            "format": "raw-shard",
            "nbytes": 26214400,
            "records": [
                {
                    "name": "clip2_509",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_510",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_511",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 16384000
                },
                {
                    "name": "clip2_512",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 19660800
                },
                {
                    "name": "clip2_513",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 22937600
                }
            ]
        },
        {
            "dataPath": "params_shard_230.bin",
            "format": "raw-shard",
            "nbytes": 29491200,
            "records": [
                {
                    "name": "clip2_514",
                    "shape": [
                        1280,
                        5120
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 0
                },
                {
                    "name": "clip2_515",
                    "shape": [
                        5120,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 13107200,
                    "byteOffset": 13107200
                },
                {
                    "name": "clip2_516",
                    "shape": [
                        1280,
                        1280
                    ],
                    "dtype": "bfloat16",
                    "format": "raw",
                    "nbytes": 3276800,
                    "byteOffset": 26214400
                }
            ]
        }
    ]
}