diff --git "a/web-sd-shards-v1-5/ndarray-cache.json" "b/web-sd-shards-v1-5/ndarray-cache.json" new file mode 100644--- /dev/null +++ "b/web-sd-shards-v1-5/ndarray-cache.json" @@ -0,0 +1,12058 @@ +{ + "metadata": { + "unetParamSize": 718, + "vaeParamSize": 140, + "clipParamSize": 196 + }, + "records": [ + { + "dataPath": "params_shard_0.bin", + "format": "raw-shard", + "nbytes": 32285440, + "records": [ + { + "name": "unet_0", + "shape": [ + 320, + 4, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 23040, + "byteOffset": 0 + }, + { + "name": "unet_1", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 23040 + }, + { + "name": "unet_2", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 23680 + }, + { + "name": "unet_3", + "shape": [ + 4, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 23040, + "byteOffset": 24320 + }, + { + "name": "unet_4", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 47360 + }, + { + "name": "unet_5", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 48000 + }, + { + "name": "unet_6", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 48640 + }, + { + "name": "unet_7", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 253440 + }, + { + "name": "unet_8", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 458240 + }, + { + "name": "unet_9", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 458880 + }, + { + "name": "unet_10", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 459520 + }, + { + "name": "unet_11", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 462080 + }, + { + "name": "unet_12", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 464640 + }, + { + "name": "unet_13", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 465280 + }, + { + "name": "unet_14", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 465920 + }, + { + "name": "unet_15", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 466560 + }, + { + "name": "unet_16", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 467200 + }, + { + "name": "unet_17", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 467840 + }, + { + "name": "unet_18", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 468480 + }, + { + "name": "unet_19", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 469120 + }, + { + "name": "unet_20", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 469760 + }, + { + "name": "unet_21", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 470400 + }, + { + "name": "unet_22", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 675200 + }, + { + "name": "unet_23", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 880000 + }, + { + "name": "unet_24", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 880640 + }, + { + "name": "unet_25", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 881280 + }, + { + "name": "unet_26", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 883840 + }, + { + "name": "unet_27", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 886400 + }, + { + "name": "unet_28", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 887040 + }, + { + "name": "unet_29", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 887680 + }, + { + "name": "unet_30", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 888320 + }, + { + "name": "unet_31", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 888960 + }, + { + "name": "unet_32", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 889600 + }, + { + "name": "unet_33", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 890240 + }, + { + "name": "unet_34", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 890880 + }, + { + "name": "unet_35", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 2734080 + }, + { + "name": "unet_36", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 4577280 + }, + { + "name": "unet_37", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 6420480 + }, + { + "name": "unet_38", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 6421120 + }, + { + "name": "unet_39", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 6421760 + }, + { + "name": "unet_40", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 6422400 + }, + { + "name": "unet_41", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 6423040 + }, + { + "name": "unet_42", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 6423680 + }, + { + "name": "unet_43", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 8266880 + }, + { + "name": "unet_44", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 10110080 + }, + { + "name": "unet_45", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 10110720 + }, + { + "name": "unet_46", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 10111360 + }, + { + "name": "unet_47", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 10112000 + }, + { + "name": "unet_48", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 10112640 + }, + { + "name": "unet_49", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 10113280 + }, + { + "name": "unet_50", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 10114560 + }, + { + "name": "unet_51", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 10115840 + }, + { + "name": "unet_52", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 10935040 + }, + { + "name": "unet_53", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11754240 + }, + { + "name": "unet_54", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11755520 + }, + { + "name": "unet_55", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 11756800 + }, + { + "name": "unet_56", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 11761920 + }, + { + "name": "unet_57", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11767040 + }, + { + "name": "unet_58", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11768320 + }, + { + "name": "unet_59", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11769600 + }, + { + "name": "unet_60", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11770880 + }, + { + "name": "unet_61", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11772160 + }, + { + "name": "unet_62", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11773440 + }, + { + "name": "unet_63", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11774720 + }, + { + "name": "unet_64", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11776000 + }, + { + "name": "unet_65", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 11777280 + }, + { + "name": "unet_66", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 11778560 + }, + { + "name": "unet_67", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 12597760 + }, + { + "name": "unet_68", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13416960 + }, + { + "name": "unet_69", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13418240 + }, + { + "name": "unet_70", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 13419520 + }, + { + "name": "unet_71", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 13424640 + }, + { + "name": "unet_72", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13429760 + }, + { + "name": "unet_73", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13431040 + }, + { + "name": "unet_74", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13432320 + }, + { + "name": "unet_75", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13433600 + }, + { + "name": "unet_76", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13434880 + }, + { + "name": "unet_77", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13436160 + }, + { + "name": "unet_78", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13437440 + }, + { + "name": "unet_79", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 13438720 + }, + { + "name": "unet_80", + "shape": [ + 640, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3686400, + "byteOffset": 20811520 + }, + { + "name": "unet_81", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 24497920 + }, + { + "name": "unet_82", + "shape": [ + 640, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 409600, + "byteOffset": 31870720 + }, + { + "name": "unet_83", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 32280320 + }, + { + "name": "unet_84", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 32280960 + }, + { + "name": "unet_85", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 32281600 + }, + { + "name": "unet_86", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 32282880 + }, + { + "name": "unet_87", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 32284160 + } + ] + }, + { + "dataPath": "params_shard_1.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_125", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_2.bin", + "format": "raw-shard", + "nbytes": 27956480, + "records": [ + { + "name": "unet_88", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 0 + }, + { + "name": "unet_89", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 7372800 + }, + { + "name": "unet_90", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14745600 + }, + { + "name": "unet_91", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14746880 + }, + { + "name": "unet_92", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14748160 + }, + { + "name": "unet_93", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14749440 + }, + { + "name": "unet_94", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14750720 + }, + { + "name": "unet_95", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 14752000 + }, + { + "name": "unet_96", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 14754560 + }, + { + "name": "unet_97", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 14757120 + }, + { + "name": "unet_98", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 18033920 + }, + { + "name": "unet_99", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21310720 + }, + { + "name": "unet_100", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21313280 + }, + { + "name": "unet_101", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 21315840 + }, + { + "name": "unet_102", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 21326080 + }, + { + "name": "unet_103", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21336320 + }, + { + "name": "unet_104", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21338880 + }, + { + "name": "unet_105", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21341440 + }, + { + "name": "unet_106", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21344000 + }, + { + "name": "unet_107", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21346560 + }, + { + "name": "unet_108", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21349120 + }, + { + "name": "unet_109", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21351680 + }, + { + "name": "unet_110", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21354240 + }, + { + "name": "unet_111", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 21356800 + }, + { + "name": "unet_112", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 21359360 + }, + { + "name": "unet_113", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 24636160 + }, + { + "name": "unet_114", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27912960 + }, + { + "name": "unet_115", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27915520 + }, + { + "name": "unet_116", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 27918080 + }, + { + "name": "unet_117", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 27928320 + }, + { + "name": "unet_118", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27938560 + }, + { + "name": "unet_119", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27941120 + }, + { + "name": "unet_120", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27943680 + }, + { + "name": "unet_121", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27946240 + }, + { + "name": "unet_122", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27948800 + }, + { + "name": "unet_123", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27951360 + }, + { + "name": "unet_124", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27953920 + } + ] + }, + { + "dataPath": "params_shard_3.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_127", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_4.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_134", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_5.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_135", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_6.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_141", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_7.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_142", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_8.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_148", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_9.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_149", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_10.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_170", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_11.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_171", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_12.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_177", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_13.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_178", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_14.bin", + "format": "raw-shard", + "nbytes": 58982400, + "records": [ + { + "name": "unet_186", + "shape": [ + 1280, + 2560, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 58982400, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_15.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_187", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_16.bin", + "format": "raw-shard", + "nbytes": 58982400, + "records": [ + { + "name": "unet_194", + "shape": [ + 1280, + 2560, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 58982400, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_17.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_195", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_18.bin", + "format": "raw-shard", + "nbytes": 29637120, + "records": [ + { + "name": "unet_126", + "shape": [ + 1280, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 14745600, + "byteOffset": 0 + }, + { + "name": "unet_128", + "shape": [ + 1280, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 14745600 + }, + { + "name": "unet_129", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 16384000 + }, + { + "name": "unet_130", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 16385280 + }, + { + "name": "unet_131", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16386560 + }, + { + "name": "unet_132", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16389120 + }, + { + "name": "unet_133", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16391680 + }, + { + "name": "unet_136", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16394240 + }, + { + "name": "unet_137", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16396800 + }, + { + "name": "unet_138", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16399360 + }, + { + "name": "unet_139", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16401920 + }, + { + "name": "unet_140", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16404480 + }, + { + "name": "unet_143", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16407040 + }, + { + "name": "unet_144", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16409600 + }, + { + "name": "unet_145", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16412160 + }, + { + "name": "unet_146", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16414720 + }, + { + "name": "unet_147", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16417280 + }, + { + "name": "unet_150", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16419840 + }, + { + "name": "unet_151", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16422400 + }, + { + "name": "unet_152", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16424960 + }, + { + "name": "unet_153", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16427520 + }, + { + "name": "unet_154", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16430080 + }, + { + "name": "unet_155", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16432640 + }, + { + "name": "unet_156", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16435200 + }, + { + "name": "unet_157", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16437760 + }, + { + "name": "unet_158", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19714560 + }, + { + "name": "unet_159", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 22991360 + }, + { + "name": "unet_160", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 22993920 + }, + { + "name": "unet_161", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 22996480 + }, + { + "name": "unet_162", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 23006720 + }, + { + "name": "unet_163", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23016960 + }, + { + "name": "unet_164", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23019520 + }, + { + "name": "unet_165", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23022080 + }, + { + "name": "unet_166", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23024640 + }, + { + "name": "unet_167", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23027200 + }, + { + "name": "unet_168", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23029760 + }, + { + "name": "unet_169", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23032320 + }, + { + "name": "unet_172", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23034880 + }, + { + "name": "unet_173", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23037440 + }, + { + "name": "unet_174", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23040000 + }, + { + "name": "unet_175", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23042560 + }, + { + "name": "unet_176", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23045120 + }, + { + "name": "unet_179", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23047680 + }, + { + "name": "unet_180", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23050240 + }, + { + "name": "unet_181", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23052800 + }, + { + "name": "unet_182", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23055360 + }, + { + "name": "unet_183", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23057920 + }, + { + "name": "unet_184", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23060480 + }, + { + "name": "unet_185", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23063040 + }, + { + "name": "unet_188", + "shape": [ + 1280, + 2560, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6553600, + "byteOffset": 23065600 + }, + { + "name": "unet_189", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 29619200 + }, + { + "name": "unet_190", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 29624320 + }, + { + "name": "unet_191", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 29629440 + }, + { + "name": "unet_192", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 29632000 + }, + { + "name": "unet_193", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 29634560 + } + ] + }, + { + "dataPath": "params_shard_19.bin", + "format": "raw-shard", + "nbytes": 58982400, + "records": [ + { + "name": "unet_202", + "shape": [ + 1280, + 2560, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 58982400, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_20.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_203", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_21.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_210", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_22.bin", + "format": "raw-shard", + "nbytes": 58982400, + "records": [ + { + "name": "unet_256", + "shape": [ + 1280, + 2560, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 58982400, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_23.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_257", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_24.bin", + "format": "raw-shard", + "nbytes": 32949760, + "records": [ + { + "name": "unet_196", + "shape": [ + 1280, + 2560, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6553600, + "byteOffset": 0 + }, + { + "name": "unet_197", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 6553600 + }, + { + "name": "unet_198", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 6558720 + }, + { + "name": "unet_199", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6563840 + }, + { + "name": "unet_200", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6566400 + }, + { + "name": "unet_201", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6568960 + }, + { + "name": "unet_204", + "shape": [ + 1280, + 2560, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6553600, + "byteOffset": 6571520 + }, + { + "name": "unet_205", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 13125120 + }, + { + "name": "unet_206", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 13130240 + }, + { + "name": "unet_207", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13135360 + }, + { + "name": "unet_208", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13137920 + }, + { + "name": "unet_209", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13140480 + }, + { + "name": "unet_211", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13143040 + }, + { + "name": "unet_212", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13145600 + }, + { + "name": "unet_213", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13148160 + }, + { + "name": "unet_214", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16424960 + }, + { + "name": "unet_215", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19701760 + }, + { + "name": "unet_216", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19704320 + }, + { + "name": "unet_217", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 19706880 + }, + { + "name": "unet_218", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 19717120 + }, + { + "name": "unet_219", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19727360 + }, + { + "name": "unet_220", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19729920 + }, + { + "name": "unet_221", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19732480 + }, + { + "name": "unet_222", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19735040 + }, + { + "name": "unet_223", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19737600 + }, + { + "name": "unet_224", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19740160 + }, + { + "name": "unet_225", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19742720 + }, + { + "name": "unet_226", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19745280 + }, + { + "name": "unet_227", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19747840 + }, + { + "name": "unet_228", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19750400 + }, + { + "name": "unet_229", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 23027200 + }, + { + "name": "unet_230", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26304000 + }, + { + "name": "unet_231", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26306560 + }, + { + "name": "unet_232", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 26309120 + }, + { + "name": "unet_233", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 26319360 + }, + { + "name": "unet_234", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26329600 + }, + { + "name": "unet_235", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26332160 + }, + { + "name": "unet_236", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26334720 + }, + { + "name": "unet_237", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26337280 + }, + { + "name": "unet_238", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26339840 + }, + { + "name": "unet_239", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26342400 + }, + { + "name": "unet_240", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26344960 + }, + { + "name": "unet_241", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26347520 + }, + { + "name": "unet_242", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26350080 + }, + { + "name": "unet_243", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 26352640 + }, + { + "name": "unet_244", + "shape": [ + 1280, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 29629440 + }, + { + "name": "unet_245", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32906240 + }, + { + "name": "unet_246", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32908800 + }, + { + "name": "unet_247", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 32911360 + }, + { + "name": "unet_248", + "shape": [ + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 10240, + "byteOffset": 32921600 + }, + { + "name": "unet_249", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32931840 + }, + { + "name": "unet_250", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32934400 + }, + { + "name": "unet_251", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32936960 + }, + { + "name": "unet_252", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32939520 + }, + { + "name": "unet_253", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32942080 + }, + { + "name": "unet_254", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32944640 + }, + { + "name": "unet_255", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32947200 + } + ] + }, + { + "dataPath": "params_shard_25.bin", + "format": "raw-shard", + "nbytes": 58982400, + "records": [ + { + "name": "unet_264", + "shape": [ + 1280, + 2560, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 58982400, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_26.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_265", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_27.bin", + "format": "raw-shard", + "nbytes": 44236800, + "records": [ + { + "name": "unet_272", + "shape": [ + 1280, + 1920, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 44236800, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_28.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_273", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_29.bin", + "format": "raw-shard", + "nbytes": 29491200, + "records": [ + { + "name": "unet_280", + "shape": [ + 1280, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 29491200, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_30.bin", + "format": "raw-shard", + "nbytes": 22118400, + "records": [ + { + "name": "unet_326", + "shape": [ + 640, + 1920, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 22118400, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_31.bin", + "format": "raw-shard", + "nbytes": 32903680, + "records": [ + { + "name": "unet_258", + "shape": [ + 1280, + 2560, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6553600, + "byteOffset": 0 + }, + { + "name": "unet_259", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 6553600 + }, + { + "name": "unet_260", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 6558720 + }, + { + "name": "unet_261", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6563840 + }, + { + "name": "unet_262", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6566400 + }, + { + "name": "unet_263", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6568960 + }, + { + "name": "unet_266", + "shape": [ + 1280, + 2560, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6553600, + "byteOffset": 6571520 + }, + { + "name": "unet_267", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 13125120 + }, + { + "name": "unet_268", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 13130240 + }, + { + "name": "unet_269", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13135360 + }, + { + "name": "unet_270", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13137920 + }, + { + "name": "unet_271", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13140480 + }, + { + "name": "unet_274", + "shape": [ + 1280, + 1920, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4915200, + "byteOffset": 13143040 + }, + { + "name": "unet_275", + "shape": [ + 1920 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3840, + "byteOffset": 18058240 + }, + { + "name": "unet_276", + "shape": [ + 1920 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3840, + "byteOffset": 18062080 + }, + { + "name": "unet_277", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 18065920 + }, + { + "name": "unet_278", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 18068480 + }, + { + "name": "unet_279", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 18071040 + }, + { + "name": "unet_281", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 18073600 + }, + { + "name": "unet_282", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 18074880 + }, + { + "name": "unet_283", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 18076160 + }, + { + "name": "unet_284", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 18895360 + }, + { + "name": "unet_285", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19714560 + }, + { + "name": "unet_286", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19715840 + }, + { + "name": "unet_287", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 19717120 + }, + { + "name": "unet_288", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 19722240 + }, + { + "name": "unet_289", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19727360 + }, + { + "name": "unet_290", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19728640 + }, + { + "name": "unet_291", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19729920 + }, + { + "name": "unet_292", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19731200 + }, + { + "name": "unet_293", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19732480 + }, + { + "name": "unet_294", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19733760 + }, + { + "name": "unet_295", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19735040 + }, + { + "name": "unet_296", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19736320 + }, + { + "name": "unet_297", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19737600 + }, + { + "name": "unet_298", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 19738880 + }, + { + "name": "unet_299", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 20558080 + }, + { + "name": "unet_300", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21377280 + }, + { + "name": "unet_301", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21378560 + }, + { + "name": "unet_302", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 21379840 + }, + { + "name": "unet_303", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 21384960 + }, + { + "name": "unet_304", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21390080 + }, + { + "name": "unet_305", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21391360 + }, + { + "name": "unet_306", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21392640 + }, + { + "name": "unet_307", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21393920 + }, + { + "name": "unet_308", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21395200 + }, + { + "name": "unet_309", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21396480 + }, + { + "name": "unet_310", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21397760 + }, + { + "name": "unet_311", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21399040 + }, + { + "name": "unet_312", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21400320 + }, + { + "name": "unet_313", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 21401600 + }, + { + "name": "unet_314", + "shape": [ + 640, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 22220800 + }, + { + "name": "unet_315", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23040000 + }, + { + "name": "unet_316", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23041280 + }, + { + "name": "unet_317", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 23042560 + }, + { + "name": "unet_318", + "shape": [ + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5120, + "byteOffset": 23047680 + }, + { + "name": "unet_319", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23052800 + }, + { + "name": "unet_320", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23054080 + }, + { + "name": "unet_321", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23055360 + }, + { + "name": "unet_322", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23056640 + }, + { + "name": "unet_323", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23057920 + }, + { + "name": "unet_324", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23059200 + }, + { + "name": "unet_325", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23060480 + }, + { + "name": "unet_327", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 23061760 + }, + { + "name": "unet_328", + "shape": [ + 640, + 1920, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2457600, + "byteOffset": 30434560 + }, + { + "name": "unet_329", + "shape": [ + 1920 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3840, + "byteOffset": 32892160 + }, + { + "name": "unet_330", + "shape": [ + 1920 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3840, + "byteOffset": 32896000 + }, + { + "name": "unet_331", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 32899840 + }, + { + "name": "unet_332", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 32901120 + }, + { + "name": "unet_333", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 32902400 + } + ] + }, + { + "dataPath": "params_shard_32.bin", + "format": "raw-shard", + "nbytes": 23765760, + "records": [ + { + "name": "unet_334", + "shape": [ + 640, + 1280, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 14745600, + "byteOffset": 0 + }, + { + "name": "unet_335", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 14745600 + }, + { + "name": "unet_336", + "shape": [ + 640, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 22118400 + }, + { + "name": "unet_337", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23756800 + }, + { + "name": "unet_338", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 23759360 + }, + { + "name": "unet_339", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23761920 + }, + { + "name": "unet_340", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23763200 + }, + { + "name": "unet_341", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 23764480 + } + ] + }, + { + "dataPath": "params_shard_33.bin", + "format": "raw-shard", + "nbytes": 28306560, + "records": [ + { + "name": "unet_342", + "shape": [ + 640, + 960, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 11059200, + "byteOffset": 0 + }, + { + "name": "unet_343", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 11059200 + }, + { + "name": "unet_344", + "shape": [ + 640, + 960, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1228800, + "byteOffset": 18432000 + }, + { + "name": "unet_345", + "shape": [ + 960 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1920, + "byteOffset": 19660800 + }, + { + "name": "unet_346", + "shape": [ + 960 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1920, + "byteOffset": 19662720 + }, + { + "name": "unet_347", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19664640 + }, + { + "name": "unet_348", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19665920 + }, + { + "name": "unet_349", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19667200 + }, + { + "name": "unet_350", + "shape": [ + 640, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 7372800, + "byteOffset": 19668480 + }, + { + "name": "unet_351", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27041280 + }, + { + "name": "unet_352", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27041920 + }, + { + "name": "unet_353", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 27042560 + }, + { + "name": "unet_354", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 27247360 + }, + { + "name": "unet_355", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27452160 + }, + { + "name": "unet_356", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27452800 + }, + { + "name": "unet_357", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27453440 + }, + { + "name": "unet_358", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27456000 + }, + { + "name": "unet_359", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27458560 + }, + { + "name": "unet_360", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27459200 + }, + { + "name": "unet_361", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27459840 + }, + { + "name": "unet_362", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27460480 + }, + { + "name": "unet_363", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27461120 + }, + { + "name": "unet_364", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27461760 + }, + { + "name": "unet_365", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27462400 + }, + { + "name": "unet_366", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27463040 + }, + { + "name": "unet_367", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27463680 + }, + { + "name": "unet_368", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 27464320 + }, + { + "name": "unet_369", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 27669120 + }, + { + "name": "unet_370", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27873920 + }, + { + "name": "unet_371", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27874560 + }, + { + "name": "unet_372", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27875200 + }, + { + "name": "unet_373", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 27877760 + }, + { + "name": "unet_374", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27880320 + }, + { + "name": "unet_375", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27880960 + }, + { + "name": "unet_376", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27881600 + }, + { + "name": "unet_377", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27882240 + }, + { + "name": "unet_378", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27882880 + }, + { + "name": "unet_379", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27883520 + }, + { + "name": "unet_380", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27884160 + }, + { + "name": "unet_381", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27884800 + }, + { + "name": "unet_382", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 27885440 + }, + { + "name": "unet_383", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 27886080 + }, + { + "name": "unet_384", + "shape": [ + 320, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 28090880 + }, + { + "name": "unet_385", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28295680 + }, + { + "name": "unet_386", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28296320 + }, + { + "name": "unet_387", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 28296960 + }, + { + "name": "unet_388", + "shape": [ + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 28299520 + }, + { + "name": "unet_389", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28302080 + }, + { + "name": "unet_390", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28302720 + }, + { + "name": "unet_391", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28303360 + }, + { + "name": "unet_392", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28304000 + }, + { + "name": "unet_393", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28304640 + }, + { + "name": "unet_394", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28305280 + }, + { + "name": "unet_395", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 28305920 + } + ] + }, + { + "dataPath": "params_shard_34.bin", + "format": "raw-shard", + "nbytes": 33320320, + "records": [ + { + "name": "unet_396", + "shape": [ + 320, + 960, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 5529600, + "byteOffset": 0 + }, + { + "name": "unet_397", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 5529600 + }, + { + "name": "unet_398", + "shape": [ + 320, + 960, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 614400, + "byteOffset": 7372800 + }, + { + "name": "unet_399", + "shape": [ + 960 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1920, + "byteOffset": 7987200 + }, + { + "name": "unet_400", + "shape": [ + 960 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1920, + "byteOffset": 7989120 + }, + { + "name": "unet_401", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 7991040 + }, + { + "name": "unet_402", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 7991680 + }, + { + "name": "unet_403", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 7992320 + }, + { + "name": "unet_404", + "shape": [ + 320, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3686400, + "byteOffset": 7992960 + }, + { + "name": "unet_405", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 11679360 + }, + { + "name": "unet_406", + "shape": [ + 320, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 409600, + "byteOffset": 13522560 + }, + { + "name": "unet_407", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13932160 + }, + { + "name": "unet_408", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13933440 + }, + { + "name": "unet_409", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 13934720 + }, + { + "name": "unet_410", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 13935360 + }, + { + "name": "unet_411", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 13936000 + }, + { + "name": "unet_412", + "shape": [ + 320, + 640, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3686400, + "byteOffset": 13936640 + }, + { + "name": "unet_413", + "shape": [ + 320, + 320, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1843200, + "byteOffset": 17623040 + }, + { + "name": "unet_414", + "shape": [ + 320, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 409600, + "byteOffset": 19466240 + }, + { + "name": "unet_415", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19875840 + }, + { + "name": "unet_416", + "shape": [ + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19877120 + }, + { + "name": "unet_417", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 19878400 + }, + { + "name": "unet_418", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 19879040 + }, + { + "name": "unet_419", + "shape": [ + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 19879680 + }, + { + "name": "unet_420", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 19880320 + }, + { + "name": "unet_421", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 20699520 + }, + { + "name": "unet_422", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 23976320 + }, + { + "name": "unet_423", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 23976960 + }, + { + "name": "unet_424", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 23977600 + }, + { + "name": "unet_425", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 24796800 + }, + { + "name": "unet_426", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 24797440 + }, + { + "name": "unet_427", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 24798080 + }, + { + "name": "unet_428", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 25002880 + }, + { + "name": "unet_429", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 25207680 + }, + { + "name": "unet_430", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 25412480 + }, + { + "name": "unet_431", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 25617280 + }, + { + "name": "unet_432", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 25822080 + }, + { + "name": "unet_433", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 26313600 + }, + { + "name": "unet_434", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 26805120 + }, + { + "name": "unet_435", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 27009920 + }, + { + "name": "unet_436", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 27829120 + }, + { + "name": "unet_437", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 28648320 + }, + { + "name": "unet_438", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 29467520 + }, + { + "name": "unet_439", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 29468160 + }, + { + "name": "unet_440", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 29468800 + }, + { + "name": "unet_441", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 30288000 + }, + { + "name": "unet_442", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 30288640 + }, + { + "name": "unet_443", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 30289280 + }, + { + "name": "unet_444", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 30494080 + }, + { + "name": "unet_445", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 30698880 + }, + { + "name": "unet_446", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 30903680 + }, + { + "name": "unet_447", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 31108480 + }, + { + "name": "unet_448", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 31313280 + }, + { + "name": "unet_449", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 31804800 + }, + { + "name": "unet_450", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 32296320 + }, + { + "name": "unet_451", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 32501120 + } + ] + }, + { + "dataPath": "params_shard_35.bin", + "format": "raw-shard", + "nbytes": 31796480, + "records": [ + { + "name": "unet_452", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 0 + }, + { + "name": "unet_453", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 819200 + }, + { + "name": "unet_454", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 1638400 + }, + { + "name": "unet_455", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 1639040 + }, + { + "name": "unet_456", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 1639680 + }, + { + "name": "unet_457", + "shape": [ + 1280, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 1640960 + }, + { + "name": "unet_458", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 3279360 + }, + { + "name": "unet_459", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 3280640 + }, + { + "name": "unet_460", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 3281920 + }, + { + "name": "unet_461", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 3283200 + }, + { + "name": "unet_462", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 4102400 + }, + { + "name": "unet_463", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 4921600 + }, + { + "name": "unet_464", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 5740800 + }, + { + "name": "unet_465", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 6560000 + }, + { + "name": "unet_466", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 7379200 + }, + { + "name": "unet_467", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 8362240 + }, + { + "name": "unet_468", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 9345280 + }, + { + "name": "unet_469", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 10164480 + }, + { + "name": "unet_470", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13441280 + }, + { + "name": "unet_471", + "shape": [ + 2560, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16718080 + }, + { + "name": "unet_472", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19994880 + }, + { + "name": "unet_473", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 19996160 + }, + { + "name": "unet_474", + "shape": [ + 1280, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 19997440 + }, + { + "name": "unet_475", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21635840 + }, + { + "name": "unet_476", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 21637120 + }, + { + "name": "unet_477", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 21638400 + }, + { + "name": "unet_478", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 22457600 + }, + { + "name": "unet_479", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 23276800 + }, + { + "name": "unet_480", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 24096000 + }, + { + "name": "unet_481", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 24915200 + }, + { + "name": "unet_482", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 25734400 + }, + { + "name": "unet_483", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 26717440 + }, + { + "name": "unet_484", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 27700480 + }, + { + "name": "unet_485", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 28519680 + } + ] + }, + { + "dataPath": "params_shard_36.bin", + "format": "raw-shard", + "nbytes": 33436160, + "records": [ + { + "name": "unet_486", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 0 + }, + { + "name": "unet_487", + "shape": [ + 2560, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 3276800 + }, + { + "name": "unet_488", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 6553600 + }, + { + "name": "unet_489", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 6554880 + }, + { + "name": "unet_490", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6556160 + }, + { + "name": "unet_491", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 6558720 + }, + { + "name": "unet_492", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 9835520 + }, + { + "name": "unet_493", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 9838080 + }, + { + "name": "unet_494", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 9840640 + }, + { + "name": "unet_495", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 9843200 + }, + { + "name": "unet_496", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13120000 + }, + { + "name": "unet_497", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16396800 + }, + { + "name": "unet_498", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19673600 + }, + { + "name": "unet_499", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 22950400 + }, + { + "name": "unet_500", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 26227200 + }, + { + "name": "unet_501", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 28193280 + }, + { + "name": "unet_502", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 30159360 + } + ] + }, + { + "dataPath": "params_shard_37.bin", + "format": "raw-shard", + "nbytes": 26214400, + "records": [ + { + "name": "unet_503", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_504", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 13107200 + } + ] + }, + { + "dataPath": "params_shard_38.bin", + "format": "raw-shard", + "nbytes": 32778240, + "records": [ + { + "name": "unet_505", + "shape": [ + 5120, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_506", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13107200 + }, + { + "name": "unet_507", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13109760 + }, + { + "name": "unet_508", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13112320 + }, + { + "name": "unet_509", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16389120 + }, + { + "name": "unet_510", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16391680 + }, + { + "name": "unet_511", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16394240 + }, + { + "name": "unet_512", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19671040 + }, + { + "name": "unet_513", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 22947840 + }, + { + "name": "unet_514", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 26224640 + }, + { + "name": "unet_515", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 29501440 + } + ] + }, + { + "dataPath": "params_shard_39.bin", + "format": "raw-shard", + "nbytes": 33423360, + "records": [ + { + "name": "unet_516", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 0 + }, + { + "name": "unet_517", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 1966080 + }, + { + "name": "unet_518", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 3932160 + }, + { + "name": "unet_519", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 7208960 + }, + { + "name": "unet_520", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 20316160 + } + ] + }, + { + "dataPath": "params_shard_40.bin", + "format": "raw-shard", + "nbytes": 32791040, + "records": [ + { + "name": "unet_521", + "shape": [ + 5120, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_522", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13107200 + }, + { + "name": "unet_523", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13109760 + }, + { + "name": "unet_524", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13112320 + }, + { + "name": "unet_525", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13114880 + }, + { + "name": "unet_526", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16391680 + }, + { + "name": "unet_527", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16394240 + }, + { + "name": "unet_528", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16396800 + }, + { + "name": "unet_529", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19673600 + }, + { + "name": "unet_530", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 19676160 + }, + { + "name": "unet_531", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19678720 + }, + { + "name": "unet_532", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 22955520 + }, + { + "name": "unet_533", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 22958080 + }, + { + "name": "unet_534", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 22960640 + }, + { + "name": "unet_535", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 26237440 + }, + { + "name": "unet_536", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 29514240 + } + ] + }, + { + "dataPath": "params_shard_41.bin", + "format": "raw-shard", + "nbytes": 26869760, + "records": [ + { + "name": "unet_537", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 0 + }, + { + "name": "unet_538", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 3276800 + }, + { + "name": "unet_539", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 6553600 + }, + { + "name": "unet_540", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 8519680 + }, + { + "name": "unet_541", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 10485760 + }, + { + "name": "unet_542", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 13762560 + } + ] + }, + { + "dataPath": "params_shard_42.bin", + "format": "raw-shard", + "nbytes": 32785920, + "records": [ + { + "name": "unet_543", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_544", + "shape": [ + 5120, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 13107200 + }, + { + "name": "unet_545", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26214400 + }, + { + "name": "unet_546", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 26216960 + }, + { + "name": "unet_547", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 26219520 + }, + { + "name": "unet_548", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 29496320 + }, + { + "name": "unet_549", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 29498880 + }, + { + "name": "unet_550", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 29501440 + }, + { + "name": "unet_551", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32778240 + }, + { + "name": "unet_552", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32780800 + }, + { + "name": "unet_553", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 32783360 + } + ] + }, + { + "dataPath": "params_shard_43.bin", + "format": "raw-shard", + "nbytes": 33448960, + "records": [ + { + "name": "unet_554", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 0 + }, + { + "name": "unet_555", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 3276800 + }, + { + "name": "unet_556", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 3279360 + }, + { + "name": "unet_557", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 3281920 + }, + { + "name": "unet_558", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 3284480 + }, + { + "name": "unet_559", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6561280 + }, + { + "name": "unet_560", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6563840 + }, + { + "name": "unet_561", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6566400 + }, + { + "name": "unet_562", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 6568960 + }, + { + "name": "unet_563", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 6571520 + }, + { + "name": "unet_564", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 9848320 + }, + { + "name": "unet_565", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 9850880 + }, + { + "name": "unet_566", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 9853440 + }, + { + "name": "unet_567", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 9856000 + }, + { + "name": "unet_568", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13132800 + }, + { + "name": "unet_569", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16409600 + }, + { + "name": "unet_570", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19686400 + }, + { + "name": "unet_571", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 22963200 + }, + { + "name": "unet_572", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 26240000 + }, + { + "name": "unet_573", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 28206080 + }, + { + "name": "unet_574", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 30172160 + } + ] + }, + { + "dataPath": "params_shard_44.bin", + "format": "raw-shard", + "nbytes": 26214400, + "records": [ + { + "name": "unet_575", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_576", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 13107200 + } + ] + }, + { + "dataPath": "params_shard_45.bin", + "format": "raw-shard", + "nbytes": 32780800, + "records": [ + { + "name": "unet_577", + "shape": [ + 5120, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_578", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13107200 + }, + { + "name": "unet_579", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13109760 + }, + { + "name": "unet_580", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13112320 + }, + { + "name": "unet_581", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16389120 + }, + { + "name": "unet_582", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16391680 + }, + { + "name": "unet_583", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16394240 + }, + { + "name": "unet_584", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16396800 + }, + { + "name": "unet_585", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19673600 + }, + { + "name": "unet_586", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 22950400 + }, + { + "name": "unet_587", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 26227200 + }, + { + "name": "unet_588", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 29504000 + } + ] + }, + { + "dataPath": "params_shard_46.bin", + "format": "raw-shard", + "nbytes": 33423360, + "records": [ + { + "name": "unet_589", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 0 + }, + { + "name": "unet_590", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 1966080 + }, + { + "name": "unet_591", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 3932160 + }, + { + "name": "unet_592", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 7208960 + }, + { + "name": "unet_593", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 20316160 + } + ] + }, + { + "dataPath": "params_shard_47.bin", + "format": "raw-shard", + "nbytes": 32780800, + "records": [ + { + "name": "unet_594", + "shape": [ + 5120, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_595", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13107200 + }, + { + "name": "unet_596", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13109760 + }, + { + "name": "unet_597", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13112320 + }, + { + "name": "unet_598", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16389120 + }, + { + "name": "unet_599", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16391680 + }, + { + "name": "unet_600", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 16394240 + }, + { + "name": "unet_601", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 16396800 + }, + { + "name": "unet_602", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 19673600 + }, + { + "name": "unet_603", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 22950400 + }, + { + "name": "unet_604", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 26227200 + }, + { + "name": "unet_605", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 29504000 + } + ] + }, + { + "dataPath": "params_shard_48.bin", + "format": "raw-shard", + "nbytes": 33423360, + "records": [ + { + "name": "unet_606", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 0 + }, + { + "name": "unet_607", + "shape": [ + 768, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1966080, + "byteOffset": 1966080 + }, + { + "name": "unet_608", + "shape": [ + 1280, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 3932160 + }, + { + "name": "unet_609", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 7208960 + }, + { + "name": "unet_610", + "shape": [ + 1280, + 5120 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 20316160 + } + ] + }, + { + "dataPath": "params_shard_49.bin", + "format": "raw-shard", + "nbytes": 33112320, + "records": [ + { + "name": "unet_611", + "shape": [ + 5120, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 13107200, + "byteOffset": 0 + }, + { + "name": "unet_612", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13107200 + }, + { + "name": "unet_613", + "shape": [ + 1, + 1280, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2560, + "byteOffset": 13109760 + }, + { + "name": "unet_614", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 13112320 + }, + { + "name": "unet_615", + "shape": [ + 1280, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 13113600 + }, + { + "name": "unet_616", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14752000 + }, + { + "name": "unet_617", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14753280 + }, + { + "name": "unet_618", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 14754560 + }, + { + "name": "unet_619", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 14755840 + }, + { + "name": "unet_620", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 15575040 + }, + { + "name": "unet_621", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 16394240 + }, + { + "name": "unet_622", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 17213440 + }, + { + "name": "unet_623", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 18032640 + }, + { + "name": "unet_624", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 18851840 + }, + { + "name": "unet_625", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 19834880 + }, + { + "name": "unet_626", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 20817920 + }, + { + "name": "unet_627", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 21637120 + }, + { + "name": "unet_628", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 24913920 + }, + { + "name": "unet_629", + "shape": [ + 2560, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 28190720 + }, + { + "name": "unet_630", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 31467520 + }, + { + "name": "unet_631", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 31468800 + }, + { + "name": "unet_632", + "shape": [ + 1280, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 31470080 + }, + { + "name": "unet_633", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 33108480 + }, + { + "name": "unet_634", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 33109760 + }, + { + "name": "unet_635", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 33111040 + } + ] + }, + { + "dataPath": "params_shard_50.bin", + "format": "raw-shard", + "nbytes": 31791360, + "records": [ + { + "name": "unet_636", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 0 + }, + { + "name": "unet_637", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 819200 + }, + { + "name": "unet_638", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 1638400 + }, + { + "name": "unet_639", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 2457600 + }, + { + "name": "unet_640", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 3276800 + }, + { + "name": "unet_641", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 4096000 + }, + { + "name": "unet_642", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 5079040 + }, + { + "name": "unet_643", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 6062080 + }, + { + "name": "unet_644", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 6881280 + }, + { + "name": "unet_645", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 10158080 + }, + { + "name": "unet_646", + "shape": [ + 2560, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 13434880 + }, + { + "name": "unet_647", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 16711680 + }, + { + "name": "unet_648", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 16712960 + }, + { + "name": "unet_649", + "shape": [ + 1280, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1638400, + "byteOffset": 16714240 + }, + { + "name": "unet_650", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 18352640 + }, + { + "name": "unet_651", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 18353920 + }, + { + "name": "unet_652", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 18355200 + }, + { + "name": "unet_653", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 18356480 + }, + { + "name": "unet_654", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 19175680 + }, + { + "name": "unet_655", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 19994880 + }, + { + "name": "unet_656", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 20814080 + }, + { + "name": "unet_657", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 21633280 + }, + { + "name": "unet_658", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 22452480 + }, + { + "name": "unet_659", + "shape": [ + 768, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 983040, + "byteOffset": 23435520 + }, + { + "name": "unet_660", + "shape": [ + 640, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 24418560 + }, + { + "name": "unet_661", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 25237760 + }, + { + "name": "unet_662", + "shape": [ + 640, + 2560 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 28514560 + } + ] + }, + { + "dataPath": "params_shard_51.bin", + "format": "raw-shard", + "nbytes": 29246600, + "records": [ + { + "name": "unet_663", + "shape": [ + 2560, + 640 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 3276800, + "byteOffset": 0 + }, + { + "name": "unet_664", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 3276800 + }, + { + "name": "unet_665", + "shape": [ + 1, + 640, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1280, + "byteOffset": 3278080 + }, + { + "name": "unet_666", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 3279360 + }, + { + "name": "unet_667", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 3280000 + }, + { + "name": "unet_668", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 4099200 + }, + { + "name": "unet_669", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 4099840 + }, + { + "name": "unet_670", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 4100480 + }, + { + "name": "unet_671", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 4101120 + }, + { + "name": "unet_672", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 4305920 + }, + { + "name": "unet_673", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 4510720 + }, + { + "name": "unet_674", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 4715520 + }, + { + "name": "unet_675", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 4920320 + }, + { + "name": "unet_676", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 5125120 + }, + { + "name": "unet_677", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 5616640 + }, + { + "name": "unet_678", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 6108160 + }, + { + "name": "unet_679", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 6312960 + }, + { + "name": "unet_680", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 7132160 + }, + { + "name": "unet_681", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 7951360 + }, + { + "name": "unet_682", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 8770560 + }, + { + "name": "unet_683", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 8771200 + }, + { + "name": "unet_684", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 8771840 + }, + { + "name": "unet_685", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 9591040 + }, + { + "name": "unet_686", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 9591680 + }, + { + "name": "unet_687", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 9592320 + }, + { + "name": "unet_688", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 9592960 + }, + { + "name": "unet_689", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 9797760 + }, + { + "name": "unet_690", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 10002560 + }, + { + "name": "unet_691", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 10207360 + }, + { + "name": "unet_692", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 10412160 + }, + { + "name": "unet_693", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 10616960 + }, + { + "name": "unet_694", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 11108480 + }, + { + "name": "unet_695", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 11600000 + }, + { + "name": "unet_696", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 11804800 + }, + { + "name": "unet_697", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 12624000 + }, + { + "name": "unet_698", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 13443200 + }, + { + "name": "unet_699", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 14262400 + }, + { + "name": "unet_700", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 14263040 + }, + { + "name": "unet_701", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 14263680 + }, + { + "name": "unet_702", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 15082880 + }, + { + "name": "unet_703", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 15083520 + }, + { + "name": "unet_704", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 15084160 + }, + { + "name": "unet_705", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 15084800 + }, + { + "name": "unet_706", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 15289600 + }, + { + "name": "unet_707", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 15494400 + }, + { + "name": "unet_708", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 15699200 + }, + { + "name": "unet_709", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 15904000 + }, + { + "name": "unet_710", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 16108800 + }, + { + "name": "unet_711", + "shape": [ + 768, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 491520, + "byteOffset": 16600320 + }, + { + "name": "unet_712", + "shape": [ + 320, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 204800, + "byteOffset": 17091840 + }, + { + "name": "unet_713", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 17296640 + }, + { + "name": "unet_714", + "shape": [ + 320, + 1280 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 18115840 + }, + { + "name": "unet_715", + "shape": [ + 1280, + 320 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 819200, + "byteOffset": 18935040 + }, + { + "name": "unet_716", + "shape": [ + 1, + 320, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 640, + "byteOffset": 19754240 + }, + { + "name": "unet_717", + "shape": [ + 1, + 4, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 8, + "byteOffset": 19754880 + }, + { + "name": "vae_0", + "shape": [ + 512, + 4, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 36864, + "byteOffset": 19754888 + }, + { + "name": "vae_1", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 19791752 + }, + { + "name": "vae_2", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 19792008 + }, + { + "name": "vae_3", + "shape": [ + 3, + 128, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6912, + "byteOffset": 19792264 + }, + { + "name": "vae_4", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 19799176 + }, + { + "name": "vae_5", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 19800200 + }, + { + "name": "vae_6", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 19801224 + }, + { + "name": "vae_7", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 19802248 + }, + { + "name": "vae_8", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 19803272 + }, + { + "name": "vae_9", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 19804296 + }, + { + "name": "vae_10", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 19805320 + }, + { + "name": "vae_11", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 24523912 + }, + { + "name": "vae_12", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 29242504 + }, + { + "name": "vae_13", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 29243528 + }, + { + "name": "vae_14", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 29244552 + }, + { + "name": "vae_15", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 29245576 + } + ] + }, + { + "dataPath": "params_shard_52.bin", + "format": "raw-shard", + "nbytes": 33042432, + "records": [ + { + "name": "vae_16", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 0 + }, + { + "name": "vae_17", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 4718592 + }, + { + "name": "vae_18", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 9437184 + }, + { + "name": "vae_19", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 9438208 + }, + { + "name": "vae_20", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 9439232 + }, + { + "name": "vae_21", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 9440256 + }, + { + "name": "vae_22", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 9441280 + }, + { + "name": "vae_23", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 14159872 + }, + { + "name": "vae_24", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18878464 + }, + { + "name": "vae_25", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18879488 + }, + { + "name": "vae_26", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18880512 + }, + { + "name": "vae_27", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18881536 + }, + { + "name": "vae_28", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 18882560 + }, + { + "name": "vae_29", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 23601152 + }, + { + "name": "vae_30", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28319744 + }, + { + "name": "vae_31", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28320768 + }, + { + "name": "vae_32", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28321792 + }, + { + "name": "vae_33", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28322816 + }, + { + "name": "vae_34", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 28323840 + } + ] + }, + { + "dataPath": "params_shard_53.bin", + "format": "raw-shard", + "nbytes": 33042432, + "records": [ + { + "name": "vae_35", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 0 + }, + { + "name": "vae_36", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4718592 + }, + { + "name": "vae_37", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4719616 + }, + { + "name": "vae_38", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4720640 + }, + { + "name": "vae_39", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4721664 + }, + { + "name": "vae_40", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 4722688 + }, + { + "name": "vae_41", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 9441280 + }, + { + "name": "vae_42", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 14159872 + }, + { + "name": "vae_43", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18878464 + }, + { + "name": "vae_44", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18879488 + }, + { + "name": "vae_45", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18880512 + }, + { + "name": "vae_46", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 18881536 + }, + { + "name": "vae_47", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 18882560 + }, + { + "name": "vae_48", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 23601152 + }, + { + "name": "vae_49", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28319744 + }, + { + "name": "vae_50", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28320768 + }, + { + "name": "vae_51", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28321792 + }, + { + "name": "vae_52", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 28322816 + }, + { + "name": "vae_53", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 28323840 + } + ] + }, + { + "dataPath": "params_shard_54.bin", + "format": "raw-shard", + "nbytes": 75890688, + "records": [ + { + "name": "clip_0", + "shape": [ + 49408, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 75890688, + "byteOffset": 0 + } + ] + }, + { + "dataPath": "params_shard_55.bin", + "format": "raw-shard", + "nbytes": 33201966, + "records": [ + { + "name": "vae_54", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 0 + }, + { + "name": "vae_55", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4718592 + }, + { + "name": "vae_56", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4719616 + }, + { + "name": "vae_57", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4720640 + }, + { + "name": "vae_58", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 4721664 + }, + { + "name": "vae_59", + "shape": [ + 512, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 4722688 + }, + { + "name": "vae_60", + "shape": [ + 256, + 512, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 2359296, + "byteOffset": 9441280 + }, + { + "name": "vae_61", + "shape": [ + 256, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 11800576 + }, + { + "name": "vae_62", + "shape": [ + 256, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 262144, + "byteOffset": 12980224 + }, + { + "name": "vae_63", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 13242368 + }, + { + "name": "vae_64", + "shape": [ + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 13243392 + }, + { + "name": "vae_65", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 13244416 + }, + { + "name": "vae_66", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 13244928 + }, + { + "name": "vae_67", + "shape": [ + 256, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 13245440 + }, + { + "name": "vae_68", + "shape": [ + 256, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 14425088 + }, + { + "name": "vae_69", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 15604736 + }, + { + "name": "vae_70", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 15605248 + }, + { + "name": "vae_71", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 15605760 + }, + { + "name": "vae_72", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 15606272 + }, + { + "name": "vae_73", + "shape": [ + 256, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 15606784 + }, + { + "name": "vae_74", + "shape": [ + 256, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 16786432 + }, + { + "name": "vae_75", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 17966080 + }, + { + "name": "vae_76", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 17966592 + }, + { + "name": "vae_77", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 17967104 + }, + { + "name": "vae_78", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 17967616 + }, + { + "name": "vae_79", + "shape": [ + 256, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 17968128 + }, + { + "name": "vae_80", + "shape": [ + 128, + 256, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 589824, + "byteOffset": 19147776 + }, + { + "name": "vae_81", + "shape": [ + 128, + 128, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 294912, + "byteOffset": 19737600 + }, + { + "name": "vae_82", + "shape": [ + 128, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 65536, + "byteOffset": 20032512 + }, + { + "name": "vae_83", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 20098048 + }, + { + "name": "vae_84", + "shape": [ + 256 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 20098560 + }, + { + "name": "vae_85", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 20099072 + }, + { + "name": "vae_86", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 20099328 + }, + { + "name": "vae_87", + "shape": [ + 128, + 128, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 294912, + "byteOffset": 20099584 + }, + { + "name": "vae_88", + "shape": [ + 128, + 128, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 294912, + "byteOffset": 20394496 + }, + { + "name": "vae_89", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 20689408 + }, + { + "name": "vae_90", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 20689664 + }, + { + "name": "vae_91", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 20689920 + }, + { + "name": "vae_92", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 20690176 + }, + { + "name": "vae_93", + "shape": [ + 128, + 128, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 294912, + "byteOffset": 20690432 + }, + { + "name": "vae_94", + "shape": [ + 128, + 128, + 3, + 3 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 294912, + "byteOffset": 20985344 + }, + { + "name": "vae_95", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 21280256 + }, + { + "name": "vae_96", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 21280512 + }, + { + "name": "vae_97", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 21280768 + }, + { + "name": "vae_98", + "shape": [ + 128 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 21281024 + }, + { + "name": "vae_99", + "shape": [ + 4, + 4, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 32, + "byteOffset": 21281280 + }, + { + "name": "vae_100", + "shape": [ + 1, + 4, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 8, + "byteOffset": 21281312 + }, + { + "name": "vae_101", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 21281320 + }, + { + "name": "vae_102", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 21282344 + }, + { + "name": "vae_103", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 21283368 + }, + { + "name": "vae_104", + "shape": [ + 512, + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 524288, + "byteOffset": 21284392 + }, + { + "name": "vae_105", + "shape": [ + 512, + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 524288, + "byteOffset": 21808680 + }, + { + "name": "vae_106", + "shape": [ + 512, + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 524288, + "byteOffset": 22332968 + }, + { + "name": "vae_107", + "shape": [ + 512, + 512 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 524288, + "byteOffset": 22857256 + }, + { + "name": "vae_108", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23381544 + }, + { + "name": "vae_109", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23382568 + }, + { + "name": "vae_110", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23383592 + }, + { + "name": "vae_111", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23384616 + }, + { + "name": "vae_112", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23385640 + }, + { + "name": "vae_113", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23386664 + }, + { + "name": "vae_114", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23387688 + }, + { + "name": "vae_115", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23388712 + }, + { + "name": "vae_116", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23389736 + }, + { + "name": "vae_117", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23390760 + }, + { + "name": "vae_118", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23391784 + }, + { + "name": "vae_119", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23392808 + }, + { + "name": "vae_120", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23393832 + }, + { + "name": "vae_121", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23394856 + }, + { + "name": "vae_122", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23395880 + }, + { + "name": "vae_123", + "shape": [ + 1, + 512, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1024, + "byteOffset": 23396904 + }, + { + "name": "vae_124", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23397928 + }, + { + "name": "vae_125", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23398440 + }, + { + "name": "vae_126", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23398952 + }, + { + "name": "vae_127", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23399464 + }, + { + "name": "vae_128", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23399976 + }, + { + "name": "vae_129", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23400488 + }, + { + "name": "vae_130", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23401000 + }, + { + "name": "vae_131", + "shape": [ + 1, + 256, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 512, + "byteOffset": 23401512 + }, + { + "name": "vae_132", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23402024 + }, + { + "name": "vae_133", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23402280 + }, + { + "name": "vae_134", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23402536 + }, + { + "name": "vae_135", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23402792 + }, + { + "name": "vae_136", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23403048 + }, + { + "name": "vae_137", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23403304 + }, + { + "name": "vae_138", + "shape": [ + 1, + 128, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 256, + "byteOffset": 23403560 + }, + { + "name": "vae_139", + "shape": [ + 1, + 3, + 1, + 1 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6, + "byteOffset": 23403816 + }, + { + "name": "clip_1", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23403822 + }, + { + "name": "clip_2", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23405358 + }, + { + "name": "clip_3", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23406894 + }, + { + "name": "clip_4", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23408430 + }, + { + "name": "clip_5", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23409966 + }, + { + "name": "clip_6", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23416110 + }, + { + "name": "clip_7", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23417646 + }, + { + "name": "clip_8", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23419182 + }, + { + "name": "clip_9", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23420718 + }, + { + "name": "clip_10", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23422254 + }, + { + "name": "clip_11", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23423790 + }, + { + "name": "clip_12", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23425326 + }, + { + "name": "clip_13", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23426862 + }, + { + "name": "clip_14", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23428398 + }, + { + "name": "clip_15", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23429934 + }, + { + "name": "clip_16", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23436078 + }, + { + "name": "clip_17", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23437614 + }, + { + "name": "clip_18", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23439150 + }, + { + "name": "clip_19", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23440686 + }, + { + "name": "clip_20", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23442222 + }, + { + "name": "clip_21", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23443758 + }, + { + "name": "clip_22", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23445294 + }, + { + "name": "clip_23", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23446830 + }, + { + "name": "clip_24", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23448366 + }, + { + "name": "clip_25", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23449902 + }, + { + "name": "clip_26", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23456046 + }, + { + "name": "clip_27", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23457582 + }, + { + "name": "clip_28", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23459118 + }, + { + "name": "clip_29", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23460654 + }, + { + "name": "clip_30", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23462190 + }, + { + "name": "clip_31", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23463726 + }, + { + "name": "clip_32", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23465262 + }, + { + "name": "clip_33", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23466798 + }, + { + "name": "clip_34", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23468334 + }, + { + "name": "clip_35", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23469870 + }, + { + "name": "clip_36", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23476014 + }, + { + "name": "clip_37", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23477550 + }, + { + "name": "clip_38", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23479086 + }, + { + "name": "clip_39", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23480622 + }, + { + "name": "clip_40", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23482158 + }, + { + "name": "clip_41", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23483694 + }, + { + "name": "clip_42", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23485230 + }, + { + "name": "clip_43", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23486766 + }, + { + "name": "clip_44", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23488302 + }, + { + "name": "clip_45", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23489838 + }, + { + "name": "clip_46", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23495982 + }, + { + "name": "clip_47", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23497518 + }, + { + "name": "clip_48", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23499054 + }, + { + "name": "clip_49", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23500590 + }, + { + "name": "clip_50", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23502126 + }, + { + "name": "clip_51", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23503662 + }, + { + "name": "clip_52", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23505198 + }, + { + "name": "clip_53", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23506734 + }, + { + "name": "clip_54", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23508270 + }, + { + "name": "clip_55", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23509806 + }, + { + "name": "clip_56", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23515950 + }, + { + "name": "clip_57", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23517486 + }, + { + "name": "clip_58", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23519022 + }, + { + "name": "clip_59", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23520558 + }, + { + "name": "clip_60", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23522094 + }, + { + "name": "clip_61", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23523630 + }, + { + "name": "clip_62", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23525166 + }, + { + "name": "clip_63", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23526702 + }, + { + "name": "clip_64", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23528238 + }, + { + "name": "clip_65", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23529774 + }, + { + "name": "clip_66", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23535918 + }, + { + "name": "clip_67", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23537454 + }, + { + "name": "clip_68", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23538990 + }, + { + "name": "clip_69", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23540526 + }, + { + "name": "clip_70", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23542062 + }, + { + "name": "clip_71", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23543598 + }, + { + "name": "clip_72", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23545134 + }, + { + "name": "clip_73", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23546670 + }, + { + "name": "clip_74", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23548206 + }, + { + "name": "clip_75", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23549742 + }, + { + "name": "clip_76", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23555886 + }, + { + "name": "clip_77", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23557422 + }, + { + "name": "clip_78", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23558958 + }, + { + "name": "clip_79", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23560494 + }, + { + "name": "clip_80", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23562030 + }, + { + "name": "clip_81", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23563566 + }, + { + "name": "clip_82", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23565102 + }, + { + "name": "clip_83", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23566638 + }, + { + "name": "clip_84", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23568174 + }, + { + "name": "clip_85", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23569710 + }, + { + "name": "clip_86", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23575854 + }, + { + "name": "clip_87", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23577390 + }, + { + "name": "clip_88", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23578926 + }, + { + "name": "clip_89", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23580462 + }, + { + "name": "clip_90", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23581998 + }, + { + "name": "clip_91", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23583534 + }, + { + "name": "clip_92", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23585070 + }, + { + "name": "clip_93", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23586606 + }, + { + "name": "clip_94", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23588142 + }, + { + "name": "clip_95", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23589678 + }, + { + "name": "clip_96", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23595822 + }, + { + "name": "clip_97", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23597358 + }, + { + "name": "clip_98", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23598894 + }, + { + "name": "clip_99", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23600430 + }, + { + "name": "clip_100", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23601966 + }, + { + "name": "clip_101", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23603502 + }, + { + "name": "clip_102", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23605038 + }, + { + "name": "clip_103", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23606574 + }, + { + "name": "clip_104", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23608110 + }, + { + "name": "clip_105", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23609646 + }, + { + "name": "clip_106", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23615790 + }, + { + "name": "clip_107", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23617326 + }, + { + "name": "clip_108", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23618862 + }, + { + "name": "clip_109", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23620398 + }, + { + "name": "clip_110", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23621934 + }, + { + "name": "clip_111", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23623470 + }, + { + "name": "clip_112", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23625006 + }, + { + "name": "clip_113", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23626542 + }, + { + "name": "clip_114", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23628078 + }, + { + "name": "clip_115", + "shape": [ + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 6144, + "byteOffset": 23629614 + }, + { + "name": "clip_116", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23635758 + }, + { + "name": "clip_117", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23637294 + }, + { + "name": "clip_118", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23638830 + }, + { + "name": "clip_119", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23640366 + }, + { + "name": "clip_120", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23641902 + }, + { + "name": "clip_121", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23643438 + }, + { + "name": "clip_122", + "shape": [ + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1536, + "byteOffset": 23644974 + }, + { + "name": "clip_123", + "shape": [ + 77, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 118272, + "byteOffset": 23646510 + }, + { + "name": "clip_124", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 23764782 + }, + { + "name": "clip_125", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 24944430 + }, + { + "name": "clip_126", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 26124078 + }, + { + "name": "clip_127", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 27303726 + }, + { + "name": "clip_128", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 28483374 + } + ] + }, + { + "dataPath": "params_shard_56.bin", + "format": "raw-shard", + "nbytes": 33030144, + "records": [ + { + "name": "clip_129", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 0 + }, + { + "name": "clip_130", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 4718592 + }, + { + "name": "clip_131", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 5898240 + }, + { + "name": "clip_132", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 7077888 + }, + { + "name": "clip_133", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 8257536 + }, + { + "name": "clip_134", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 9437184 + }, + { + "name": "clip_135", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 14155776 + }, + { + "name": "clip_136", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 18874368 + }, + { + "name": "clip_137", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 20054016 + }, + { + "name": "clip_138", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 21233664 + }, + { + "name": "clip_139", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 22413312 + }, + { + "name": "clip_140", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 23592960 + }, + { + "name": "clip_141", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 28311552 + } + ] + }, + { + "dataPath": "params_shard_57.bin", + "format": "raw-shard", + "nbytes": 33030144, + "records": [ + { + "name": "clip_142", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 0 + }, + { + "name": "clip_143", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 1179648 + }, + { + "name": "clip_144", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 2359296 + }, + { + "name": "clip_145", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 3538944 + }, + { + "name": "clip_146", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 4718592 + }, + { + "name": "clip_147", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 9437184 + }, + { + "name": "clip_148", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 14155776 + }, + { + "name": "clip_149", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 15335424 + }, + { + "name": "clip_150", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 16515072 + }, + { + "name": "clip_151", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 17694720 + }, + { + "name": "clip_152", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 18874368 + }, + { + "name": "clip_153", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 23592960 + }, + { + "name": "clip_154", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 28311552 + }, + { + "name": "clip_155", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 29491200 + }, + { + "name": "clip_156", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 30670848 + }, + { + "name": "clip_157", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 31850496 + } + ] + }, + { + "dataPath": "params_shard_58.bin", + "format": "raw-shard", + "nbytes": 33030144, + "records": [ + { + "name": "clip_158", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 0 + }, + { + "name": "clip_159", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 4718592 + }, + { + "name": "clip_160", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 9437184 + }, + { + "name": "clip_161", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 10616832 + }, + { + "name": "clip_162", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 11796480 + }, + { + "name": "clip_163", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 12976128 + }, + { + "name": "clip_164", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 14155776 + }, + { + "name": "clip_165", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 18874368 + }, + { + "name": "clip_166", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 23592960 + }, + { + "name": "clip_167", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 24772608 + }, + { + "name": "clip_168", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 25952256 + }, + { + "name": "clip_169", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 27131904 + }, + { + "name": "clip_170", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 28311552 + } + ] + }, + { + "dataPath": "params_shard_59.bin", + "format": "raw-shard", + "nbytes": 33030144, + "records": [ + { + "name": "clip_171", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 0 + }, + { + "name": "clip_172", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 4718592 + }, + { + "name": "clip_173", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 5898240 + }, + { + "name": "clip_174", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 7077888 + }, + { + "name": "clip_175", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 8257536 + }, + { + "name": "clip_176", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 9437184 + }, + { + "name": "clip_177", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 14155776 + }, + { + "name": "clip_178", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 18874368 + }, + { + "name": "clip_179", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 20054016 + }, + { + "name": "clip_180", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 21233664 + }, + { + "name": "clip_181", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 22413312 + }, + { + "name": "clip_182", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 23592960 + }, + { + "name": "clip_183", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 28311552 + } + ] + }, + { + "dataPath": "params_shard_60.bin", + "format": "raw-shard", + "nbytes": 28311552, + "records": [ + { + "name": "clip_184", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 0 + }, + { + "name": "clip_185", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 1179648 + }, + { + "name": "clip_186", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 2359296 + }, + { + "name": "clip_187", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 3538944 + }, + { + "name": "clip_188", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 4718592 + }, + { + "name": "clip_189", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 9437184 + }, + { + "name": "clip_190", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 14155776 + }, + { + "name": "clip_191", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 15335424 + }, + { + "name": "clip_192", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 16515072 + }, + { + "name": "clip_193", + "shape": [ + 768, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 1179648, + "byteOffset": 17694720 + }, + { + "name": "clip_194", + "shape": [ + 768, + 3072 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 18874368 + }, + { + "name": "clip_195", + "shape": [ + 3072, + 768 + ], + "dtype": "float32", + "format": "f32-to-bf16", + "nbytes": 4718592, + "byteOffset": 23592960 + } + ] + } + ] +} \ No newline at end of file