diff --git "a/web-sd-shards-xl/ndarray-cache-b16.json" "b/web-sd-shards-xl/ndarray-cache-b16.json" new file mode 100644--- /dev/null +++ "b/web-sd-shards-xl/ndarray-cache-b16.json" @@ -0,0 +1,28495 @@ +{ + "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 + } + ] + } + ] +} \ No newline at end of file