diff --git a/.gitattributes b/.gitattributes index 4080f3f71f6fb2b8f12e91ea551513ca5c2ff008..360f2d10f6de7dd8e86741712c6f15e5bfaf78dc 100644 --- a/.gitattributes +++ b/.gitattributes @@ -51,3 +51,29 @@ lingbot/checkpoint-151000/state/__4_0.distcp filter=lfs diff=lfs merge=lfs -text lingbot/checkpoint-151000/state/__5_0.distcp filter=lfs diff=lfs merge=lfs -text lingbot/checkpoint-151000/state/__6_0.distcp filter=lfs diff=lfs merge=lfs -text lingbot/checkpoint-151000/state/__7_0.distcp filter=lfs diff=lfs merge=lfs -text +dm05/checkpoint-86000/state/.metadata filter=lfs diff=lfs merge=lfs -text +dm05/checkpoint-86000/state/__0_0.distcp filter=lfs diff=lfs merge=lfs -text +dm05/checkpoint-86000/state/__0_1.distcp filter=lfs diff=lfs merge=lfs -text +dm05/checkpoint-86000/state/__0_2.distcp filter=lfs diff=lfs merge=lfs -text +dm05/checkpoint-86000/state/__0_3.distcp filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_0/d/874ac71d7dee40a4c6b13abf2ce16b40 filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_0/d/991d9888065ee20f3f716cb9fcae60cd filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_0/d/a7de6921a62e9caae69d7b8cb9463356 filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_0/d/e75b6a2ff750fae5216aa3b74f375676 filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_0/d/ef69036551fe90423f9e8ede05957f29 filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_1/d/185e811af5283feed6e0a924d3c346cf filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_1/d/2765a18abced3097d9cddb036117ea15 filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_1/d/ba6c4676695fac6d07db8322079e764e filter=lfs diff=lfs merge=lfs -text +pi05/120000/params/ocdbt.process_1/d/c357388e78624c3bd26ee0d497d37bfc filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_0/d/78399c32470191978bcca35a8c83fc0e filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_0/d/94544c6c69e3673070d0c89660512d14 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_0/d/e8f29bce7fcbd02230364a5ea6a1019a filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_1/d/2d4f1917cf7734421195df45de938703 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_1/d/3d04da5b957a58495910107c08f62eb0 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_1/d/5ee85dd1362bf7aba4076154adee71a0 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_2/d/7d04930ea03442e748d46f42d74efa97 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_2/d/bdc43f2dba4aecfaacc924950eb72435 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_2/d/fe81f44cea20cac46cd5f97dea11d172 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_3/d/9314deaca83ddf9173de247009462604 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_3/d/dd7f545e8fc2f34890902ddad9662b61 filter=lfs diff=lfs merge=lfs -text +spatial/90000/params/ocdbt.process_3/d/fe58466ddb619f4c9c9a70e35dae662b filter=lfs diff=lfs merge=lfs -text diff --git a/dm05/checkpoint-86000/state/.metadata b/dm05/checkpoint-86000/state/.metadata new file mode 100644 index 0000000000000000000000000000000000000000..d756090f5313eaee536589f4d5ace0a322494777 --- /dev/null +++ b/dm05/checkpoint-86000/state/.metadata @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b84082b061689869352cd725a243e662cbca43952d79ec23ad7e7e102c3be79b +size 503448 diff --git a/dm05/checkpoint-86000/state/__0_0.distcp b/dm05/checkpoint-86000/state/__0_0.distcp new file mode 100644 index 0000000000000000000000000000000000000000..176a314141a8033761282bdd84766899409339db --- /dev/null +++ b/dm05/checkpoint-86000/state/__0_0.distcp @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:647239112132ec8e0f343a90112c9885dc74b26e81862e33c31505f2ed1a0691 +size 5829513555 diff --git a/dm05/checkpoint-86000/state/__0_1.distcp b/dm05/checkpoint-86000/state/__0_1.distcp new file mode 100644 index 0000000000000000000000000000000000000000..f87f4e2172b0b730bf05f33cd8d272e66e092981 --- /dev/null +++ b/dm05/checkpoint-86000/state/__0_1.distcp @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27c27b6eaae5a629b974f9e8c1ea879a3c1f291bff7a0e73eb1af5589103e81a +size 5829671021 diff --git a/dm05/checkpoint-86000/state/__0_2.distcp b/dm05/checkpoint-86000/state/__0_2.distcp new file mode 100644 index 0000000000000000000000000000000000000000..76899c18e739600cfaccbe6d66da591c11b1c9c0 --- /dev/null +++ b/dm05/checkpoint-86000/state/__0_2.distcp @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4b3ff288b2a86a66b877e6229e9c8670f0be67b54abb6e70accdf6b2fae679a5 +size 5829870405 diff --git a/dm05/checkpoint-86000/state/__0_3.distcp b/dm05/checkpoint-86000/state/__0_3.distcp new file mode 100644 index 0000000000000000000000000000000000000000..f9656e289e79b84364d65bb944074008f6759ddb --- /dev/null +++ b/dm05/checkpoint-86000/state/__0_3.distcp @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9e21ebe881e29d1266435980a7c356567ff412031072fbf152b691eaf0da4543 +size 5829880676 diff --git a/dm05/checkpoint-86000/trainer.json b/dm05/checkpoint-86000/trainer.json new file mode 100644 index 0000000000000000000000000000000000000000..54a141b6ece7ff8b54e22ee5472fd3a8d0bbde84 --- /dev/null +++ b/dm05/checkpoint-86000/trainer.json @@ -0,0 +1,497 @@ +{ + "step": 86000, + "scheduler": { + "base_lrs": [ + 5e-05 + ], + "last_epoch": 86000, + "_step_count": 86001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 2.447975151910276e-05 + ], + "lr_lambdas": [ + null + ] + }, + "config": { + "model": "dm05", + "output": "/data/luotianwei/training_runs/v16_h200_continuation_20260924/runs/dm05-v16-h20032-b1792-from83000-20260924-r1", + "steps": 130000, + "micro_batch": 56, + "accumulation": 1, + "workers": 6, + "horizon": 50, + "lr": 5e-05, + "warmup": 10000, + "save_every": 1000, + "validation_every": 1000, + "validation_limit": 0, + "resume": "/data/luotianwei/training_runs/v16_h200_continuation_20260924/runs/dm05-v16-h20016-b896-from82000-20260924-r1/checkpoint-83000", + "transition_receipt": null, + "seed": 42, + "denoise_steps": 10, + "smoke": false, + "wandb": "online", + "wandb_project": "dexomni-v16-baselines", + "wandb_entity": "hongjinwu-tsinghua-university", + "wandb_id": null, + "parallel": "hsdp", + "shard_size": 8, + "numa_bind": false, + "isolate_unused_head": false, + "vision_checkpointing": "native", + "optimized": true, + "benchmark": false, + "profile_step": -1, + "h200_probe_updates": 0, + "global_batch_size": 1792, + "total_workers": 192, + "normalization": { + "data_root": "/data/luotianwei/asset_render_infra/runs/V16_20260916/full_rerender_01/training_package_final18k.building", + "dataset_version": "V16 final18k", + "package_sha256": { + "PACKAGE_COMPLETE.json": "a4d574528c4c478b9a6eb2be9a9baf17acfdb6945b5d8deb4eb0a66a057d3e67", + "source_manifest.jsonl": "62a025c12454ffa80d00e1b13613ce4b6fe5d438bd64a646cd4e54198771a85d", + "meta/splits.json": "8c4de928722371acf2806d9826751dce37a73443712748d779b4715155913cf7", + "meta/info.json": "7e330c3a1fce7e3af26bb4aa2d2c956e9a8a82d225a9538e6e4ca7fe2aa9da51", + "meta/dexjoco_policy_schema.json": "96977d055d566fac68f8e77f8e3c81180f1f9b3e46f6fa73ae8573866b12830e", + "norm_stats.json": "1887f8104bf3d17fa640b2ef78131a2f763a65a594ebc75a612a3998fc576154", + "language/prompt_paraphrases.json": "362652a4642ccd199235bd5ffa87e45588244faf9f6d9d5960dfca914a7cc2b0", + "code/prompt_sampler.py": "ed20fea5d6bde0c4db6d6922130359769d304b43fc5f7483e5fbc6b846e4e3f4", + "code/training_reader.py": "e529a0d6803a084338a12f93430a77a4ee53c945e8bee3e0fa090076943957d8", + "FILE_MANIFEST.jsonl": "24ba881af1b26a6ab2a9ebc79f51e896f7b4d5c6742a78a9e7b1fe90e019038f", + "provenance/TRAINING_PACKAGE_READER_QA.json": "50b819b150b0dddd1a5a566def8dde6ff5170cbcaf377eddcc6530b9d1680f37", + "provenance/FINAL18K_DELIVERY_SCOPE.json": "42be22010291b4298925ef7e2bd3de5455074fbe82b380f055b371123d36373e" + }, + "train_episodes": 15178, + "heldout_episodes": 2822, + "train_frames": 5095027, + "tasks": 54, + "per_task": { + "a01_fruit_to_bucket": { + "train": 427, + "validation": 73 + }, + "a02_object_to_tray": { + "train": 367, + "validation": 133 + }, + "a03_ball_to_bowl": { + "train": 438, + "validation": 62 + }, + "a06_orient_rubiks_cube": { + "train": 432, + "validation": 68 + }, + "b01_pitcher_lid": { + "train": 459, + "validation": 41 + }, + "b03_peg_in_socket": { + "train": 412, + "validation": 88 + }, + "b04_block_stacking": { + "train": 446, + "validation": 54 + }, + "b05_clear_box_lid": { + "train": 442, + "validation": 58 + }, + "b320_as013_utility_knife_body_stabilized_blade_slide": { + "train": 445, + "validation": 55 + }, + "b320_as019_screwdriver_safe_orientation_handover": { + "train": 419, + "validation": 81 + }, + "b320_cup_stabilized_faucet_open": { + "train": 196, + "validation": 104 + }, + "b320_dual_drawer_synchronized_open": { + "validation": 83, + "train": 217 + }, + "b320_dutch_oven_stabilized_stirring": { + "train": 222, + "validation": 78 + }, + "b320_hs005_sink_bottle_stabilized_scrub": { + "train": 94, + "validation": 6 + }, + "b320_low_compression_carton_carry": { + "train": 220, + "validation": 80 + }, + "b320_slim_object_stable_handover": { + "train": 252, + "validation": 48 + }, + "b320_sponge_stabilized_soap_dispense": { + "train": 242, + "validation": 58 + }, + "b320_synchronized_dual_knob_setting_2": { + "train": 236, + "validation": 64 + }, + "b320_synchronized_salt_pepper_placement": { + "train": 281, + "validation": 19 + }, + "b320_two_block_midair_face_mating": { + "train": 253, + "validation": 47 + }, + "bench320_bc012_dual_can_synchronized_bin_load": { + "train": 233, + "validation": 67 + }, + "bench320_fu006_hold_drawer_store_charger": { + "train": 270, + "validation": 30 + }, + "bench320_fu015_laptop_base_stabilized_open": { + "train": 264, + "validation": 36 + }, + "c01_hammer_nail": { + "train": 96, + "validation": 4 + }, + "c02_spoon_marbles": { + "validation": 35, + "train": 265 + }, + "c04_water_plant": { + "train": 391, + "validation": 109 + }, + "c05_move_mouse_click": { + "train": 238, + "validation": 62 + }, + "c06_bimanual_photograph": { + "train": 463, + "validation": 37 + }, + "c10_two_stage_charging": { + "validation": 86, + "train": 414 + }, + "c12_push_blocks_to_line": { + "train": 272, + "validation": 28 + }, + "d02_bimanual_microwave_cook": { + "train": 443, + "validation": 57 + }, + "d03_bimanual_handoff_to_box": { + "validation": 108, + "train": 392 + }, + "d05_scissors_three_cuts": { + "validation": 61, + "train": 439 + }, + "d06_sort_clutter_to_baskets": { + "validation": 45, + "train": 455 + }, + "d09_prepare_coffee": { + "train": 208, + "validation": 92 + }, + "d20_1_5_set_controls": { + "train": 4, + "validation": 4 + }, + "d20_1_load_and_start": { + "train": 58, + "validation": 8 + }, + "d20_2_retrieve_and_plate": { + "train": 24, + "validation": 2 + }, + "g02_repair_close_device": { + "train": 227, + "validation": 73 + }, + "g08_micropipette_box_stabilized_lid_open": { + "validation": 53, + "train": 247 + }, + "g10_folding_ruler_bimanual_unfold": { + "train": 270, + "validation": 30 + }, + "h01_chinese_chess": { + "train": 454, + "validation": 46 + }, + "p001_bimanual_folding_fan_target_angle": { + "train": 89, + "validation": 11 + }, + "p0_02_hold_lever_open_latched_door": { + "train": 253, + "validation": 47 + }, + "ycb01_roll_ball_through_gate": { + "train": 451, + "validation": 49 + }, + "ycb02_select_largest_fitting_ball": { + "train": 442, + "validation": 58 + }, + "ycb03_airplane_propeller_spin": { + "train": 273, + "validation": 27 + }, + "ycb04_cups_denest_order": { + "train": 79, + "validation": 21 + }, + "ycb05_drill_point_and_trigger": { + "train": 85, + "validation": 15 + }, + "ycb06_single_step_table_setting": { + "train": 255, + "validation": 45 + }, + "ycb07_timer_program_start": { + "train": 272, + "validation": 28 + }, + "ycb08_spatula_flip_sponge": { + "train": 78, + "validation": 22 + }, + "ycb09_wine_glass_target_transfer": { + "train": 228, + "validation": 72 + }, + "ycb10_marker_finger_gait": { + "train": 46, + "validation": 54 + } + }, + "validation_samples": 2822, + "normalization": "exact packaged selected-train-only statistics; no refitting", + "normalization_key_mapping": { + "state": "state", + "actions": "action" + }, + "validation_interval_optimizer_updates": 1000, + "validation_sampling": "all 2822 heldout trajectories, one fixed start/middle/end-10 observation each", + "validation_frame_subset_declared": true, + "no_source_data_modified": true, + "mount": "same Aliyun CPFS; /data in GPU pods, /mnt/cpfs in CPU host", + "prior_exposure": "These trajectories existed in the V15 training universe. This split is held out only for a fresh V16 training run that respects it; it is not unseen-data validation for a model already trained on these V15 trajectories.", + "split_sha256": "232b6cde4b104f6da2230d3e4891fb6355f5f2d1c867154675404b7ea84f23b6", + "prompt_revision": "V16_prompt_20260919C_after_20260918B", + "language_snapshot_sha256": { + "episode_prompt_map.json": "5b09cefb3d6ed562dbfdfeab7b1dbf3a936c57a97202d9824151a556f12f93a1", + "source_manifest.jsonl": "62a025c12454ffa80d00e1b13613ce4b6fe5d438bd64a646cd4e54198771a85d", + "prior_prompt_paraphrases.json": "3c90059496eefb4b108e1268c305249baf1b05c6be8c1cf48db7de8b13c2bb1b", + "prompt_paraphrases.json": "362652a4642ccd199235bd5ffa87e45588244faf9f6d9d5960dfca914a7cc2b0", + "validation_texts.json": "9e247d61455aeab6b1e1f412001a0ca25573d6424a0d08ca692636e93345cba7", + "prior_READY.json": "47980b130165b623db0a31dd0289dd91d3ec8130a1f481f11eb6e943ca50b6a4", + "prior_source_manifest.jsonl": "5154aa38e4762635400cabbba2b1df80d439124533687dc1fc1bdcf61c89e105", + "info.json": "7e330c3a1fce7e3af26bb4aa2d2c956e9a8a82d225a9538e6e4ca7fe2aa9da51" + }, + "language_derivatives_sha256": { + "tokenized_prompt_bank.json": "977fad69db991b56d201e14388827fe6e0a674fbefb640d2d34ddb07d713c4c8", + "processor_input_checks.json": "5cdc471b5e3336fb7baf71ecd837bbdc7ba8719760022a8ee68f4ab260e96637" + }, + "prior_ready_sha256": "47980b130165b623db0a31dd0289dd91d3ec8130a1f481f11eb6e943ca50b6a4" + }, + "dataset_version": "V16 final18k", + "authoritative_package": "/data/luotianwei/asset_render_infra/runs/V16_20260916/full_rerender_01/training_package_final18k.building", + "prompt_revision": "V16_prompt_20260919C_after_20260918B", + "validation_language": { + "prompt_revision": "V16_prompt_20260919C_after_20260918B", + "validation_text_manifest": "/data/luotianwei/training_runs/v16_dm05_2node_20260918/revisions/prompt_r3_20260919/data/v16_final18k/language_snapshot/validation_texts.json", + "validation_text_sha256": "9e247d61455aeab6b1e1f412001a0ca25573d6424a0d08ca692636e93345cba7", + "language_bank_sha256": "362652a4642ccd199235bd5ffa87e45588244faf9f6d9d5960dfca914a7cc2b0" + }, + "split_sha256": "232b6cde4b104f6da2230d3e4891fb6355f5f2d1c867154675404b7ea84f23b6", + "nccl": { + "NCCL_NET": "IB", + "NCCL_IB_HCA": "=mlx5_0:1,mlx5_1:1,mlx5_2:1,mlx5_3:1,mlx5_4:1,mlx5_5:1,mlx5_6:1,mlx5_7:1", + "NCCL_IB_GID_INDEX": null, + "NCCL_NET_GDR_LEVEL": null, + "NCCL_SOCKET_IFNAME": "bond0", + "LD_PRELOAD": null, + "NCCL_PROTO": null, + "NCCL_ALGO": null + }, + "integration_sha256": { + "train_v16.py": "d93e00e3171ce5e0ac03f108cf64ddd15799db0d19df647f9f9c1e28c83fb000", + "policies.py": "d7df7b4febb450087dd4c1dc7710bce37239de5f1d6d81bfb3630a73a40af295", + "v15_data.py": "250bfa40fb31593b751fa2e7676483e0390b8de29af4f97af7c4556228f4f17d", + "action_metrics.py": "c805e94f2998645e97b059dad648bd4efb6dc9cd72d4ae865157ef260c3b1ee0", + "run_training.sh": "b9728c6feb80fed1ff5eaebda717a8f2374cb023719f7423b75bcef296704ba2", + "optimized_runtime.py": "4e7c21803c2f9c9f37a846f029e3655472e3d689cb3982c6512448c5e1aa57ea", + "perf_attention.py": "5c18b0a0e386a88fcce279c3bb0dab673f445ea9c6073e8091652dcbd1f59d67", + "runtime_cache.py": "8dbfa2d16dcdcd98fcaf13942b727c80eb10d2aa342861115c93b4db338d58de", + "hierarchical_comms.py": "763b1541f4d78660a2dfaca41f4a17e0bba1d99bf003f5d15f2f3aa8264e3d9f", + "prompt_revision.py": "f0cb234a865d3006cefd4219c44aba3a26d9ecc1c36f435382a130b465173baa" + }, + "world_size": 32, + "rank0_cpu_affinity": null, + "formal_training": true, + "launch_recipe": { + "parallel": "hsdp", + "world": 32, + "micro_batch": 56, + "shard_size": 8, + "numa_bind": false, + "accumulation": 1, + "workers": 6, + "isolate_unused_head": false, + "vision_checkpointing": "native", + "nccl_proto": null, + "optimized": true, + "cuda_allocator": "expandable_segments:True" + }, + "parent_checkpoint": "/data/luotianwei/training_runs/v16_h200_continuation_20260924/runs/dm05-v16-h20016-b896-from82000-20260924-r1/checkpoint-83000", + "parent_optimizer_step": 83000, + "parent_run": "/data/luotianwei/training_runs/v16_h200_continuation_20260924/runs/dm05-v16-h20016-b896-from82000-20260924-r1", + "parent_wandb_run_id": "rpihjzfe", + "continuation_revision": "h200_20260924_dm4node_b1792", + "first_new_prompt_optimizer_step": 20001, + "prompt_revision_changed": false, + "batch_change_authorized": true, + "previous_global_batch": 896, + "learning_rate_scaling": "none; preserve original optimizer and scheduler", + "dataset_mirror": { + "passed": true, + "all_manifest_files_sha256_verified": true, + "files": 90042, + "bytes": 110510326989, + "seconds": 24.13396289999946, + "manifest_sha256": "62a025c12454ffa80d00e1b13613ce4b6fe5d438bd64a646cd4e54198771a85d", + "checksums": { + "source_manifest.jsonl": "62a025c12454ffa80d00e1b13613ce4b6fe5d438bd64a646cd4e54198771a85d", + "FILE_MANIFEST.jsonl": "24ba881af1b26a6ab2a9ebc79f51e896f7b4d5c6742a78a9e7b1fe90e019038f", + "PACKAGE_COMPLETE.json": "a4d574528c4c478b9a6eb2be9a9baf17acfdb6945b5d8deb4eb0a66a057d3e67" + }, + "authoritative": "/data/luotianwei/asset_render_infra/runs/V16_20260916/full_rerender_01/training_package_final18k.building", + "mirror": "/mnt/luotianwei/dataset_exports/v16_final18k_20260917/dataset" + }, + "augmentation": { + "status": "unchanged existing TrainingTransformPipeline", + "new_implementation": false + }, + "parent_checkpoint_trainer_sha256": "ac3513c65eba8c591084de24b9a1dbf75a808e02cf7994fe2891aa0829ce7b46", + "resume_boundary_validation": "full fixed held-out set before first update; then every 1000 updates", + "actual_entrypoint_sha256": "68042f2d400410e9672a39661b1b1e4099af1c3b81535c2909ccf708696c39e8", + "actual_runtime_files_sha256": { + "runtime_dm4node/common/continuation.py": "0e0b651f899187206c931400389f97c8c1570538c23b553cf22b69c7aec6b527", + "runtime_dm4node/dm05/train_v16.py": "68042f2d400410e9672a39661b1b1e4099af1c3b81535c2909ccf708696c39e8" + }, + "runtime_nccl_version": 22809, + "performance_recipe": { + "reshard_after_forward": false, + "gc_every_optimizer_updates": 50, + "compile_blocks": false, + "trim_only_padding": true, + "cache_attention_mask": false, + "gradient_reduce_dtype": "float32", + "loss_reduction": "mean of per-sample masked means" + } + }, + "complete": true, + "data_cursor": { + "epoch": 1, + "global_offset": 4634112, + "world_size": 32, + "dataset_samples": 5095027, + "optimizer_step": 86000, + "drop_last": true + }, + "optimizer_uninitialized_parameters": [ + "native.model.action_expert.layers.0.input_layernorm.weight", + "native.model.action_expert.layers.0.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.1.input_layernorm.weight", + "native.model.action_expert.layers.1.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.10.input_layernorm.weight", + "native.model.action_expert.layers.10.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.11.input_layernorm.weight", + "native.model.action_expert.layers.11.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.12.input_layernorm.weight", + "native.model.action_expert.layers.12.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.13.input_layernorm.weight", + "native.model.action_expert.layers.13.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.14.input_layernorm.weight", + "native.model.action_expert.layers.14.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.15.input_layernorm.weight", + "native.model.action_expert.layers.15.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.16.input_layernorm.weight", + "native.model.action_expert.layers.16.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.17.input_layernorm.weight", + "native.model.action_expert.layers.17.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.18.input_layernorm.weight", + "native.model.action_expert.layers.18.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.19.input_layernorm.weight", + "native.model.action_expert.layers.19.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.2.input_layernorm.weight", + "native.model.action_expert.layers.2.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.20.input_layernorm.weight", + "native.model.action_expert.layers.20.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.21.input_layernorm.weight", + "native.model.action_expert.layers.21.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.22.input_layernorm.weight", + "native.model.action_expert.layers.22.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.23.input_layernorm.weight", + "native.model.action_expert.layers.23.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.24.input_layernorm.weight", + "native.model.action_expert.layers.24.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.25.input_layernorm.weight", + "native.model.action_expert.layers.25.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.26.input_layernorm.weight", + "native.model.action_expert.layers.26.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.27.input_layernorm.weight", + "native.model.action_expert.layers.27.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.28.input_layernorm.weight", + "native.model.action_expert.layers.28.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.29.input_layernorm.weight", + "native.model.action_expert.layers.29.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.3.input_layernorm.weight", + "native.model.action_expert.layers.3.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.30.input_layernorm.weight", + "native.model.action_expert.layers.30.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.31.input_layernorm.weight", + "native.model.action_expert.layers.31.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.32.input_layernorm.weight", + "native.model.action_expert.layers.32.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.33.input_layernorm.weight", + "native.model.action_expert.layers.33.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.4.input_layernorm.weight", + "native.model.action_expert.layers.4.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.5.input_layernorm.weight", + "native.model.action_expert.layers.5.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.6.input_layernorm.weight", + "native.model.action_expert.layers.6.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.7.input_layernorm.weight", + "native.model.action_expert.layers.7.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.8.input_layernorm.weight", + "native.model.action_expert.layers.8.pre_feedforward_layernorm.weight", + "native.model.action_expert.layers.9.input_layernorm.weight", + "native.model.action_expert.layers.9.pre_feedforward_layernorm.weight", + "native.model.action_expert.norm.weight", + "native.model.vlm.lm_head.weight" + ] +} diff --git a/motus/step-100000/COMPLETE.json b/motus/step-100000/COMPLETE.json new file mode 100644 index 0000000000000000000000000000000000000000..c058a5d102f2195449fa64eff8941922366a4464 --- /dev/null +++ b/motus/step-100000/COMPLETE.json @@ -0,0 +1,5 @@ +{ + "step": 100000, + "world_size": 32, + "data_fingerprint": "ef446b6d34a1d58565d9882ae6a76c6bd7714417d5bf0126e3c8a60083390834" +} diff --git a/motus/step-100000/mp_rank_00_model_states.pt b/motus/step-100000/mp_rank_00_model_states.pt new file mode 100644 index 0000000000000000000000000000000000000000..7be81ef9ddb350a06d1f3ce4cf2c787845e2b032 --- /dev/null +++ b/motus/step-100000/mp_rank_00_model_states.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd51ff7f7a180cd7dcf39c2405a67e0e105e4134957ea1212a91f17f54693775 +size 20302340934 diff --git a/pi05/120000/_CHECKPOINT_METADATA b/pi05/120000/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..f7191dd8d61b23c7feaf407c5cdb0327bb5189b7 --- /dev/null +++ b/pi05/120000/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": {"assets": "openpi.training.checkpoints.CallbackHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1790088444647358274, "commit_timestamp_nsecs": 1790088458233042886, "custom_metadata": {}} \ No newline at end of file diff --git a/pi05/120000/assets/training_package_final18k.building/dexjoco_policy_schema.json b/pi05/120000/assets/training_package_final18k.building/dexjoco_policy_schema.json new file mode 100644 index 0000000000000000000000000000000000000000..b029329c1e562aaa56b46918e071139d62177340 --- /dev/null +++ b/pi05/120000/assets/training_package_final18k.building/dexjoco_policy_schema.json @@ -0,0 +1,20 @@ +{ + "schema_id": "dexjoco.policy.world_palm_command_eef52.v1", + "state_dim": 52, + "action_dim": 52, + "side_order": [ + "right", + "left" + ], + "pose_frame": "simulation_world", + "per_side_layout": [ + "palm_xyz_m3", + "palm_rotvec_rad3", + "hand_control_rad20" + ], + "state_semantics": "current_measured", + "action_semantics": "absolute_controller_command", + "inactive_side_target": "measured_state_hold", + "fps": 30, + "use_delta_state_actions": false +} diff --git a/pi05/120000/assets/training_package_final18k.building/norm_stats.json b/pi05/120000/assets/training_package_final18k.building/norm_stats.json new file mode 100644 index 0000000000000000000000000000000000000000..10ab4b1c1fa0e1877a81e4041cd87e170d162ca0 --- /dev/null +++ b/pi05/120000/assets/training_package_final18k.building/norm_stats.json @@ -0,0 +1,440 @@ +{ + "norm_stats": { + "state": { + "mean": [ + -0.3756132374448428, + -0.1922035737454245, + 1.0734230522126986, + -0.10580894246740222, + 1.5268773253785781, + 0.4333860134570782, + 1.132603953344494, + 0.08388715424914854, + 0.07523190984348331, + -0.07488959831063875, + 0.7732500842174495, + -0.12185213894010583, + 0.03850897705568312, + 0.2327035147078383, + 0.7632519618412967, + -0.2910936825517061, + 0.16769780230497838, + 0.41669230984074335, + 0.6387215124508895, + -0.24341684851721052, + 0.36322233178079255, + 0.462197209783378, + 0.32384134316880764, + -0.22829207266160093, + 0.33395296461319696, + 0.6350905085498076, + -0.3829340032214411, + 0.23747290056934375, + 1.1343590026068826, + 0.026452533005930295, + 1.4843445695776927, + -0.2585795523639295, + 0.7469146075438257, + 0.0672836664647167, + 0.1955226226353155, + -0.20780992198090484, + 0.5752212242661446, + 0.04070584363737646, + 0.0683460504290883, + 0.05577568087165422, + 0.558050560327393, + 0.12704396839113233, + 0.1267647391843182, + 0.1517304261074897, + 0.4239144173999666, + 0.09345724438730785, + 0.26116501126327085, + 0.21259837105925616, + 0.16146965412911107, + 0.07877759725253462, + 0.23133980464121412, + 0.27850359237622985 + ], + "std": [ + 0.09304206459434612, + 0.1141805585115126, + 0.0841446408990242, + 0.5204488459290729, + 0.3817662704118041, + 0.49694798042127974, + 0.2489560013555542, + 0.1641344552146221, + 0.1781477649495512, + 0.2502631711226777, + 0.34873854317364805, + 0.1306700593090313, + 0.18972288032456258, + 0.32330288143528735, + 0.40195007500792546, + 0.07905433881083665, + 0.3187277782065752, + 0.3565283211489738, + 0.37094963430240535, + 0.0953331216749368, + 0.4659978581296099, + 0.352689106002158, + 0.2854612791786066, + 0.11234707986089751, + 0.39803934189144835, + 0.521816813241966, + 0.08009408212943486, + 0.09401033845091089, + 0.1904600388290421, + 0.3992187480807428, + 0.2960483004012528, + 0.486371433709671, + 0.47586127059713124, + 0.191694118524718, + 0.3896953068700562, + 0.22268162681507173, + 0.4599319823567688, + 0.14423681858923207, + 0.2559992251829431, + 0.23752660054358588, + 0.46289004101273157, + 0.14470454899903715, + 0.32026994267311665, + 0.26623261962977257, + 0.3915513086266434, + 0.14511725082098642, + 0.4226707422409446, + 0.26943846307128516, + 0.20132351473565963, + 0.1318417820071871, + 0.36393383229390625, + 0.3850392568094395 + ], + "q01": [ + -0.6303133940696717, + -0.49039829075336455, + 0.9590315091609954, + -1.4498432207107543, + 0.4342478609085083, + -0.6164907681941986, + 0.46926967859268187, + -0.13878458738327026, + -0.2688474714756012, + -0.4699000120162964, + -0.0409120986610651, + -0.37005338072776794, + -0.18901705354452134, + -0.04946111008524894, + -0.07391509041190147, + -0.37005823850631714, + -0.14579924881458284, + 0.045585820749402046, + -0.1353547328710556, + -0.37003010511398315, + -0.09601925179362297, + 0.0493484428524971, + -0.16259999573230743, + -0.37001505494117737, + -0.03281373284757137, + -0.04305015780031681, + -0.620149359703064, + -0.018492647893726826, + 0.959199697971344, + -1.2192575716972351, + 0.5459224581718445, + -1.7410719466209412, + 0.04749998822808266, + -0.13877438008785248, + -0.4641999900341034, + -0.4699001610279083, + -0.13390560775995253, + -0.3347871708869934, + -0.2800824296474457, + -0.20414662837982178, + -0.16438406705856323, + -0.26173519909381865, + -0.2052921387553215, + -0.07768250808119774, + -0.15539845824241638, + -0.26413850247859955, + -0.10864684417843819, + -1.618995774776977e-8, + -0.16259999573230743, + -0.27257111310958865, + -0.03925309255719185, + -0.019343462586402894 + ], + "q99": [ + -0.1429787877202033, + 0.06926057368516925, + 1.31552481174469, + 1.419685802459719, + 2.313767518997193, + 1.8823706173896817, + 1.6030604839324951, + 0.5273734545707705, + 0.8339820194244385, + 0.6540826928615578, + 1.4454282951354984, + 0.19733155071735392, + 0.9226136159896852, + 1.3771842885017398, + 1.4400742053985596, + -0.03186169348657102, + 1.2215302157402042, + 1.5744069504737854, + 1.289541120529175, + 0.0, + 1.54863703250885, + 1.5618776345252992, + 0.9544583165645605, + 0.042008595317602204, + 1.548729658126831, + 1.5735000371932983, + -0.14856379091739613, + 0.49161147832870555, + 1.5470348596572876, + 1.0734670352935813, + 2.1224486541748053, + 0.7093607556819924, + 1.5999280118942263, + 0.7695496964454651, + 1.3125319123268129, + 0.3716325366497045, + 1.54079332113266, + 0.37005192041397095, + 1.423793301582338, + 1.4689472913742065, + 1.531768712997437, + 0.3700435161590576, + 1.5511980056762695, + 1.5744975805282593, + 1.291005187034607, + 0.3700040578842163, + 1.5486994981765747, + 1.5621692895889288, + 0.6914429044723511, + 0.36999934911727905, + 1.5489757061004639, + 1.573495626449585 + ] + }, + "actions": { + "mean": [ + -0.3774107377951146, + -0.18965618535706702, + 1.0641560658690188, + -0.11103215016979268, + 1.5383117967473399, + 0.4532484255267634, + 1.2039889892376208, + 0.11678257676051611, + 0.11588039181884575, + -0.06093050913636807, + 0.8149401311327196, + -0.1018223126727827, + 0.05742328690068333, + 0.24123074688621046, + 0.7829393645883668, + -0.2958270739398715, + 0.1805258156932764, + 0.4246148250852619, + 0.6504853811836465, + -0.2460646884389835, + 0.3741211964375254, + 0.4678848982449643, + 0.32235360482634756, + -0.23765578555728065, + 0.3379110187884529, + 0.643152251649382, + -0.3848594706130866, + 0.2357238783200948, + 1.1263265817951122, + 0.021872026007404226, + 1.4861157052289846, + -0.2788444682078895, + 0.7843823958258818, + 0.07701920838345469, + 0.214831416196619, + -0.2102144486385382, + 0.6057784770199638, + 0.02948739070176912, + 0.08135993517052935, + 0.059747916171314565, + 0.5733317954554376, + 0.12655270298818608, + 0.1368120192430504, + 0.15605925890463526, + 0.4295585910023934, + 0.0956009204345276, + 0.2693357191396455, + 0.21579352706767746, + 0.15918600261297375, + 0.08171245596112303, + 0.2341379805309993, + 0.2808800596485234 + ], + "std": [ + 0.09641069340875866, + 0.11552087570514671, + 0.0939393278278454, + 0.563722423693591, + 0.4875573067842306, + 0.5399865121557139, + 0.27652388260340527, + 0.14684859822064758, + 0.1684866418140672, + 0.25952557539633136, + 0.36590928210454793, + 0.11058576469559811, + 0.20063887028322033, + 0.3341425757094886, + 0.40964437380556257, + 0.09441367991960796, + 0.3287868205736134, + 0.36528511447502743, + 0.375255072797315, + 0.10586658220070251, + 0.47840825798977804, + 0.3581175632426619, + 0.29215476188906697, + 0.12186112363913805, + 0.40243501014928296, + 0.5321386156550624, + 0.08237170280205165, + 0.09558065705192924, + 0.19745989690203336, + 0.4276374823074401, + 0.37072525521428906, + 0.5016658236076725, + 0.5209300966264223, + 0.1871772373501775, + 0.38356398656347473, + 0.23027786966750088, + 0.4873308715528858, + 0.13015109075698988, + 0.26894123904290335, + 0.2429862761253798, + 0.48034368134821526, + 0.14761885142008427, + 0.33496951850571866, + 0.2746012234457569, + 0.39989538866620117, + 0.14963057317974632, + 0.4382793620788446, + 0.2756912622411717, + 0.20470206291873239, + 0.13310030509721554, + 0.37136047040924147, + 0.3905423055831995 + ], + "q01": [ + -0.6369176208972931, + -0.49610799849033355, + 0.878734827041626, + -1.517451400756836, + 0.1415246000885964, + -0.8078330540657044, + 0.5683906280994415, + -0.16589868068695068, + -0.17180271923542023, + -0.49319911003112793, + -0.027466807067394235, + -0.36470742404460904, + -0.1524206021428108, + -0.03850268520414829, + -0.07340786144137382, + -0.4949833154678345, + -0.13442210614681244, + 0.055256085097789766, + -0.13676236003637313, + -0.49380611479282377, + -0.09227014303207398, + 0.060445761680603026, + -0.3264355117082596, + -0.49498969316482544, + -0.02686372883617878, + -0.041792757064104076, + -0.6257761108875275, + -0.025668210461735723, + 0.8586105704307556, + -1.258314883708954, + 0.2663422566652298, + -1.7825496745109557, + -0.04479999467730522, + -0.16589999198913574, + -0.4931999146938324, + -0.4931999742984772, + -0.12625114262104034, + -0.3150788855552673, + -0.27238276064395905, + -0.19124494552612303, + -0.19424962133169174, + -0.24421980768442153, + -0.19036661237478256, + -0.07502289950847625, + -0.25037343442440035, + -0.25319877922534945, + -0.1027567994594574, + -1.367732659574017e-8, + -0.27963358044624326, + -0.2525194877386093, + -0.03889536760747433, + -0.017797669395804405 + ], + "q99": [ + -0.11421960830688455, + 0.07550657570362115, + 1.3191913795471195, + 1.5217537164688126, + 2.481211943626404, + 1.9685343456268405, + 1.6507995128631592, + 0.5159857547283176, + 0.8874582099914556, + 0.7049283862113955, + 1.4419074940681458, + 0.1731965500116351, + 0.996877969503403, + 1.463679189682007, + 1.4402226376533511, + -0.06131668716669042, + 1.2631767010688784, + 1.627199411392212, + 1.2886699962615968, + -0.014802797697484493, + 1.6271992921829224, + 1.6056890130043031, + 0.9410188329219819, + 0.024919682629406827, + 1.5642336201667788, + 1.6271997690200806, + -0.12243296459317206, + 0.4934544855356225, + 1.5470348596572876, + 1.1484377813339277, + 2.2727585124969503, + 0.7677231836319001, + 1.6507998704910278, + 0.7607841765880589, + 1.306426877975464, + 0.3887728536128999, + 1.5609785485267642, + 0.36016589820384987, + 1.4833825039863717, + 1.4693725228309635, + 1.536626744270325, + 0.4935007607936861, + 1.6271998882293701, + 1.6271998882293701, + 1.2918108701705933, + 0.48732446014881187, + 1.6271998882293701, + 1.6237806773185748, + 0.6809232234954834, + 0.41420475661754635, + 1.6271997690200806, + 1.6271998882293701 + ] + } + } +} \ No newline at end of file diff --git a/pi05/120000/params/_METADATA b/pi05/120000/params/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..14d836611efde27fdbb29957fff436120e28b2b7 --- /dev/null +++ b/pi05/120000/params/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 269]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 576, 538]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 269, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 8, 72, 144]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72, 16, 72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [14, 14, 3, 72]}}, "('params', 'PaliGemma', 'img', 'head', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'PaliGemma', 'img', 'head', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [576, 256]}}, "('params', 'PaliGemma', 'img', 'pos_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 16, 1152]}}, "('params', 'PaliGemma', 'llm', 'embedder', 'input_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16072, 2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [192]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 384]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 8, 256, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 8, 256, 128]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 2, 1, 256, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 2, 1, 128, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 8, 256, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 8, 128, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 2, 2048, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 2048, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 2, 1024, 512]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 512, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 128]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 192]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 1024, 384]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 128]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 192]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [9, 1024, 384]}}, "('params', 'action_in_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('params', 'action_in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [52, 64]}}, "('params', 'action_out_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [52]}}, "('params', 'action_out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64, 52]}}, "('params', 'time_mlp_in', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_in", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('params', 'time_mlp_in', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_in", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 128]}}, "('params', 'time_mlp_out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('params', 'time_mlp_out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 128]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/pi05/120000/params/_sharding b/pi05/120000/params/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..70167297ea88b11762819b26847f5a71c32ce67c --- /dev/null +++ b/pi05/120000/params/_sharding @@ -0,0 +1 @@ +{"cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null, null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null, null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null, null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcucG9zX2VtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZW1iZWRkZXIuaW5wdXRfZW1iZWRkaW5nLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [\"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5nYXRpbmdfZWluc3VtLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5saW5lYXIudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmdhdGluZ19laW5zdW0udmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmxpbmVhci52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW0udy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW1fMS53LnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX291dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX291dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX2luLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX2luLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [2, 8], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}, {\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}, {\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}"} \ No newline at end of file diff --git a/pi05/120000/params/array_metadatas/process_0 b/pi05/120000/params/array_metadatas/process_0 new file mode 100644 index 0000000000000000000000000000000000000000..a043c01ea383a2d9397d18e37fbdbc8d0984556d --- /dev/null +++ b/pi05/120000/params/array_metadatas/process_0 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 269], "chunk_shape": [27, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 576, 538], "chunk_shape": [27, 576, 538], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 1152], "chunk_shape": [27, 269, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 8, 72, 144], "chunk_shape": [27, 8, 72, 144], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 72], "chunk_shape": [14, 14, 3, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [576, 256], "chunk_shape": [576, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 16, 1152], "chunk_shape": [1, 16, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [16072, 2048], "chunk_shape": [16072, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.bias.value", "write_shape": [192], "chunk_shape": [192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.kernel.value", "write_shape": [512, 384], "chunk_shape": [512, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [9, 8, 256, 256], "chunk_shape": [9, 8, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [9, 8, 256, 128], "chunk_shape": [9, 8, 256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [9, 2, 1, 256, 256], "chunk_shape": [9, 2, 1, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [9, 2, 1, 128, 256], "chunk_shape": [9, 2, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [9, 8, 256, 256], "chunk_shape": [9, 8, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [9, 8, 128, 256], "chunk_shape": [9, 8, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [9, 2, 2048, 2048], "chunk_shape": [9, 2, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [9, 2048, 2048], "chunk_shape": [9, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [9, 2, 1024, 512], "chunk_shape": [9, 2, 1024, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [9, 512, 1024], "chunk_shape": [9, 512, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.kernel.value", "write_shape": [9, 1024, 384], "chunk_shape": [9, 1024, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.kernel.value", "write_shape": [9, 1024, 384], "chunk_shape": [9, 1024, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [52, 64], "chunk_shape": [52, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [52], "chunk_shape": [52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [64, 52], "chunk_shape": [64, 52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.kernel.value", "write_shape": [512, 128], "chunk_shape": [512, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.kernel.value", "write_shape": [512, 128], "chunk_shape": [512, 128], "ext_metadata": null}}]} \ No newline at end of file diff --git a/pi05/120000/params/array_metadatas/process_1 b/pi05/120000/params/array_metadatas/process_1 new file mode 100644 index 0000000000000000000000000000000000000000..a043c01ea383a2d9397d18e37fbdbc8d0984556d --- /dev/null +++ b/pi05/120000/params/array_metadatas/process_1 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 269], "chunk_shape": [27, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 576, 538], "chunk_shape": [27, 576, 538], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 1152], "chunk_shape": [27, 269, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 8, 72, 144], "chunk_shape": [27, 8, 72, 144], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 72], "chunk_shape": [14, 14, 3, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [576, 256], "chunk_shape": [576, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 16, 1152], "chunk_shape": [1, 16, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [16072, 2048], "chunk_shape": [16072, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.bias.value", "write_shape": [192], "chunk_shape": [192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.kernel.value", "write_shape": [512, 384], "chunk_shape": [512, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [9, 8, 256, 256], "chunk_shape": [9, 8, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [9, 8, 256, 128], "chunk_shape": [9, 8, 256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [9, 2, 1, 256, 256], "chunk_shape": [9, 2, 1, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [9, 2, 1, 128, 256], "chunk_shape": [9, 2, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [9, 8, 256, 256], "chunk_shape": [9, 8, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [9, 8, 128, 256], "chunk_shape": [9, 8, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [9, 2, 2048, 2048], "chunk_shape": [9, 2, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [9, 2048, 2048], "chunk_shape": [9, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [9, 2, 1024, 512], "chunk_shape": [9, 2, 1024, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [9, 512, 1024], "chunk_shape": [9, 512, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.kernel.value", "write_shape": [9, 1024, 384], "chunk_shape": [9, 1024, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.kernel.value", "write_shape": [9, 1024, 384], "chunk_shape": [9, 1024, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [52, 64], "chunk_shape": [52, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [52], "chunk_shape": [52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [64, 52], "chunk_shape": [64, 52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.kernel.value", "write_shape": [512, 128], "chunk_shape": [512, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.kernel.value", "write_shape": [512, 128], "chunk_shape": [512, 128], "ext_metadata": null}}]} \ No newline at end of file diff --git a/pi05/120000/params/d/babee4c35e9a59eebe34b4c3cc4d7dbd b/pi05/120000/params/d/babee4c35e9a59eebe34b4c3cc4d7dbd new file mode 100644 index 0000000000000000000000000000000000000000..d11eca52a3b35d2dc246c244351df7ac6e4859d3 Binary files /dev/null and b/pi05/120000/params/d/babee4c35e9a59eebe34b4c3cc4d7dbd differ diff --git a/pi05/120000/params/manifest.ocdbt b/pi05/120000/params/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..7136e25ee53eb11eeb697f76ea6e77726665dab5 Binary files /dev/null and b/pi05/120000/params/manifest.ocdbt differ diff --git a/pi05/120000/params/ocdbt.process_0/d/6d87d81da5d30393d70f4cad385b1578 b/pi05/120000/params/ocdbt.process_0/d/6d87d81da5d30393d70f4cad385b1578 new file mode 100644 index 0000000000000000000000000000000000000000..72dd9f73ebe9883b372f7912e184701a459d2de1 Binary files /dev/null and b/pi05/120000/params/ocdbt.process_0/d/6d87d81da5d30393d70f4cad385b1578 differ diff --git a/pi05/120000/params/ocdbt.process_0/d/874ac71d7dee40a4c6b13abf2ce16b40 b/pi05/120000/params/ocdbt.process_0/d/874ac71d7dee40a4c6b13abf2ce16b40 new file mode 100644 index 0000000000000000000000000000000000000000..577e069376462ab2278a832d0d0af6ce502f676f --- /dev/null +++ b/pi05/120000/params/ocdbt.process_0/d/874ac71d7dee40a4c6b13abf2ce16b40 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ac121352f7dfc9d0368d976be4b2c74bd2b212219380243a0d358253312cdad0 +size 2378133817 diff --git a/pi05/120000/params/ocdbt.process_0/d/991d9888065ee20f3f716cb9fcae60cd b/pi05/120000/params/ocdbt.process_0/d/991d9888065ee20f3f716cb9fcae60cd new file mode 100644 index 0000000000000000000000000000000000000000..d8f8779c4612594e61822010418625b45e498f09 --- /dev/null +++ b/pi05/120000/params/ocdbt.process_0/d/991d9888065ee20f3f716cb9fcae60cd @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a138d70f942068058809f7271a598abd0dfa59cd4b29a83eff008cb598a437f1 +size 988280700 diff --git a/pi05/120000/params/ocdbt.process_0/d/a7de6921a62e9caae69d7b8cb9463356 b/pi05/120000/params/ocdbt.process_0/d/a7de6921a62e9caae69d7b8cb9463356 new file mode 100644 index 0000000000000000000000000000000000000000..9d5a6a99aec469abae69db3d858b02bd52f765dc --- /dev/null +++ b/pi05/120000/params/ocdbt.process_0/d/a7de6921a62e9caae69d7b8cb9463356 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:502af57c6664b3a4f63fd4e55fb94f4a359df94788e261c3515a63b6064e72bb +size 890431199 diff --git a/pi05/120000/params/ocdbt.process_0/d/e75b6a2ff750fae5216aa3b74f375676 b/pi05/120000/params/ocdbt.process_0/d/e75b6a2ff750fae5216aa3b74f375676 new file mode 100644 index 0000000000000000000000000000000000000000..bf6dc101fa0915b2fbaa561c2bd7b7e9463f3595 --- /dev/null +++ b/pi05/120000/params/ocdbt.process_0/d/e75b6a2ff750fae5216aa3b74f375676 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:225c9642e1ff4e18b6b4da53a3d2178a576f0c5b78ee338b08e7362119b46c02 +size 1955722 diff --git a/pi05/120000/params/ocdbt.process_0/d/ed240a079df18ee1160f6865a92a1439 b/pi05/120000/params/ocdbt.process_0/d/ed240a079df18ee1160f6865a92a1439 new file mode 100644 index 0000000000000000000000000000000000000000..62feaf6d292a9d5a315f6a10e6ecd8b3b240e0a9 Binary files /dev/null and b/pi05/120000/params/ocdbt.process_0/d/ed240a079df18ee1160f6865a92a1439 differ diff --git a/pi05/120000/params/ocdbt.process_0/d/ef69036551fe90423f9e8ede05957f29 b/pi05/120000/params/ocdbt.process_0/d/ef69036551fe90423f9e8ede05957f29 new file mode 100644 index 0000000000000000000000000000000000000000..1e2539bbda4380fdff9c2b8e51407d1bc2f1c49e --- /dev/null +++ b/pi05/120000/params/ocdbt.process_0/d/ef69036551fe90423f9e8ede05957f29 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8485c4505b1ab219dda62d41880a94ffb15ab85fa48812a63ae6cf959278ab78 +size 1959569970 diff --git a/pi05/120000/params/ocdbt.process_0/d/fef413d6b31ac378bd060ad0314a18cd b/pi05/120000/params/ocdbt.process_0/d/fef413d6b31ac378bd060ad0314a18cd new file mode 100644 index 0000000000000000000000000000000000000000..5053fc3a01396b67e6c61fb2e47ead40a85a1b20 Binary files /dev/null and b/pi05/120000/params/ocdbt.process_0/d/fef413d6b31ac378bd060ad0314a18cd differ diff --git a/pi05/120000/params/ocdbt.process_0/manifest.ocdbt b/pi05/120000/params/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..566dc2cfae15b4d91b65bad9141e2f9d6456ab8c Binary files /dev/null and b/pi05/120000/params/ocdbt.process_0/manifest.ocdbt differ diff --git a/pi05/120000/params/ocdbt.process_1/d/185e811af5283feed6e0a924d3c346cf b/pi05/120000/params/ocdbt.process_1/d/185e811af5283feed6e0a924d3c346cf new file mode 100644 index 0000000000000000000000000000000000000000..0258ef4482fe17bdf8da75e2070a439ae3981c73 --- /dev/null +++ b/pi05/120000/params/ocdbt.process_1/d/185e811af5283feed6e0a924d3c346cf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bec1d20c2563b7d5d75cd822c695897ed38acf2e9399f5ea7190c962faec25f3 +size 1069944538 diff --git a/pi05/120000/params/ocdbt.process_1/d/2765a18abced3097d9cddb036117ea15 b/pi05/120000/params/ocdbt.process_1/d/2765a18abced3097d9cddb036117ea15 new file mode 100644 index 0000000000000000000000000000000000000000..7f9230f1e78b3c9af7995aa4b99d7197e2bc17a7 --- /dev/null +++ b/pi05/120000/params/ocdbt.process_1/d/2765a18abced3097d9cddb036117ea15 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eb63d03a32de86ad3e3a0fc8e2cb8b456bc3f27a7abdd626743de962555f0718 +size 2098084875 diff --git a/pi05/120000/params/ocdbt.process_1/d/ba6c4676695fac6d07db8322079e764e b/pi05/120000/params/ocdbt.process_1/d/ba6c4676695fac6d07db8322079e764e new file mode 100644 index 0000000000000000000000000000000000000000..83143e1c7a5045b57de2125c71735b77e0898615 --- /dev/null +++ b/pi05/120000/params/ocdbt.process_1/d/ba6c4676695fac6d07db8322079e764e @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c7785f8428a97bb67d9c4c2538016c161208ead510cf6023e7d2d93c5f14acfb +size 2239151773 diff --git a/pi05/120000/params/ocdbt.process_1/d/c357388e78624c3bd26ee0d497d37bfc b/pi05/120000/params/ocdbt.process_1/d/c357388e78624c3bd26ee0d497d37bfc new file mode 100644 index 0000000000000000000000000000000000000000..3875a888315efa87729b7e175f6c3086629d2b05 --- /dev/null +++ b/pi05/120000/params/ocdbt.process_1/d/c357388e78624c3bd26ee0d497d37bfc @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d2b7f8c9821d0b07bf21029d1e8be0b5271bd39cda6df2b9faa70239db2f4bcf +size 810512080 diff --git a/pi05/120000/params/ocdbt.process_1/d/c9e0a1d3483cce605d5e817c9a2e5594 b/pi05/120000/params/ocdbt.process_1/d/c9e0a1d3483cce605d5e817c9a2e5594 new file mode 100644 index 0000000000000000000000000000000000000000..b32b9dcd2537a7566b8c1cb5b4e08349827d7b1e Binary files /dev/null and b/pi05/120000/params/ocdbt.process_1/d/c9e0a1d3483cce605d5e817c9a2e5594 differ diff --git a/pi05/120000/params/ocdbt.process_1/manifest.ocdbt b/pi05/120000/params/ocdbt.process_1/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..61b3cfac65326be4c308c779dce1b882a95c2cbf Binary files /dev/null and b/pi05/120000/params/ocdbt.process_1/manifest.ocdbt differ diff --git a/spatial/90000/_CHECKPOINT_METADATA b/spatial/90000/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..04721b0d0193feec0d92b83919b4be0fe87f3a7d --- /dev/null +++ b/spatial/90000/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": {"assets": "openpi.training.checkpoints.CallbackHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1790303312415647513, "commit_timestamp_nsecs": 1790303330730872522, "custom_metadata": {}} \ No newline at end of file diff --git a/spatial/90000/assets/v16_eef52/norm_stats.json b/spatial/90000/assets/v16_eef52/norm_stats.json new file mode 100644 index 0000000000000000000000000000000000000000..80bf4c5a73f90c7bd1c482e7dad8d0c624d3fdbc --- /dev/null +++ b/spatial/90000/assets/v16_eef52/norm_stats.json @@ -0,0 +1,440 @@ +{ + "norm_stats": { + "state": { + "mean": [ + -0.37493317884985355, + -0.19214926965532358, + 1.073924416642189, + -0.10123716720994118, + 1.5204711570046308, + 0.4372921326565244, + 1.1315033984615281, + 0.08495136470113773, + 0.07564528639992668, + -0.07465158073151765, + 0.7744987712702465, + -0.1258867261094454, + 0.04250661448977076, + 0.2347008680635789, + 0.7603606899067282, + -0.2925327381560343, + 0.17125480447862984, + 0.4166827567097831, + 0.6352519667991475, + -0.24707432649524802, + 0.36496552409835864, + 0.4619743012675064, + 0.3210227588964232, + -0.2310879726747223, + 0.3349583466008356, + 0.6342009989417076, + -0.38318271908981727, + 0.2367559395184455, + 1.1363971105909347, + 0.02797872719342557, + 1.48013873815997, + -0.262102352427792, + 0.7566701079949177, + 0.06618824201054642, + 0.19556439696384115, + -0.20728487371188353, + 0.5826199787040975, + 0.04425560327133216, + 0.072311895310848, + 0.05623284639066142, + 0.5654256355474193, + 0.1300542724144109, + 0.12910657664844316, + 0.15465966677500978, + 0.4311632284261141, + 0.09708080675048263, + 0.2681796828751213, + 0.21585575145773095, + 0.1650758918629359, + 0.08163102266165731, + 0.23556799246562235, + 0.28681041516863653 + ], + "std": [ + 0.09214291247851196, + 0.1146965198876074, + 0.08397843159588082, + 0.522619034467312, + 0.38439627596147014, + 0.49732564408110275, + 0.2471619375320197, + 0.16364254790777663, + 0.17817177049134325, + 0.24841382107208024, + 0.3481596402435832, + 0.13106209847227937, + 0.1956673664442704, + 0.3260485101527757, + 0.4006551616168581, + 0.07880818321998188, + 0.3224221234211086, + 0.3584502209061131, + 0.3700396372615921, + 0.09444697262622495, + 0.46711689223794683, + 0.3538755605719868, + 0.2850818298573228, + 0.11047583407306832, + 0.3984847073325966, + 0.5222004660839564, + 0.0817131312697536, + 0.0969877251750872, + 0.1909037392988457, + 0.40781527277180185, + 0.3126174226166087, + 0.4892208002319383, + 0.47495802301521006, + 0.19088283496815825, + 0.39366486452353383, + 0.21998156328412896, + 0.46245919570363886, + 0.14378798719105634, + 0.2598726777318054, + 0.23404233718240913, + 0.4649632482091461, + 0.1450312983764252, + 0.3210599523812679, + 0.2642499624482593, + 0.3936792295988475, + 0.1452639267999304, + 0.42787948733200193, + 0.26947032031052515, + 0.20196063698875627, + 0.13211751394617227, + 0.3682051097490605, + 0.3897070846827109 + ], + "q01": [ + -0.6208170282840729, + -0.4867764663696289, + 0.9593588572740555, + -1.4626132988929748, + 0.4261292818188667, + -0.6191595703363418, + 0.4830762362480164, + -0.13878323152661323, + -0.26790595293045044, + -0.4699000120162964, + -0.038699508160352704, + -0.3700518310070038, + -0.18993005231022836, + -0.05070611760020256, + -0.07528877288103103, + -0.3700555267930031, + -0.14427189350128175, + 0.04666521765291691, + -0.1407039423286915, + -0.37002819776535034, + -0.09577793903648853, + 0.0502321557328105, + -0.16259999573230743, + -0.3700132966041565, + -0.033374550491571425, + -0.04328557576984167, + -0.6225471317768096, + -0.024140829406678676, + 0.9589344900846482, + -1.2371728372573854, + 0.4694429349899292, + -1.729389420747757, + 0.04749998822808266, + -0.1387753339111805, + -0.4641999900341034, + -0.46990010172128677, + -0.1581012237071991, + -0.3290718254446983, + -0.2974558874964714, + -0.20096000000834466, + -0.16438935741782187, + -0.25930082112550734, + -0.21966602340340613, + -0.0773624663054943, + -0.15539880096912384, + -0.26036013156175614, + -0.11633960314095021, + -3.974363593783892e-8, + -0.16259999573230743, + -0.2725770589709282, + -0.04078961752355099, + -0.018900213446468115 + ], + "q99": [ + -0.14407621502876272, + 0.06851677611470239, + 1.3132777643203735, + 1.430105987787247, + 2.307256724834443, + 1.8628496241569563, + 1.603004696369171, + 0.5220825231075292, + 0.8371079432964369, + 0.6382268297672308, + 1.4471813261508946, + 0.19025001704692854, + 0.9249057108163845, + 1.3730177426338208, + 1.440089032649994, + -0.03290390912443383, + 1.2209309709072116, + 1.5744413232803345, + 1.2891837537288668, + 0.0, + 1.548639417886734, + 1.5633095633983611, + 0.9508493846654892, + 0.03451127812266397, + 1.548564341068268, + 1.5734999179840088, + -0.1511378169059751, + 0.4993625739216805, + 1.5470348596572876, + 1.0896608853340153, + 2.1388903212547303, + 0.7447108435630806, + 1.5988503336906437, + 0.8050316560268408, + 1.3152864038944256, + 0.3445987305045155, + 1.5359327471256272, + 0.37005117565393447, + 1.3837730526924137, + 1.3500248003005995, + 1.5332987415790558, + 0.37004399329423904, + 1.5511969327926636, + 1.5744670665264129, + 1.2943953859806065, + 0.37000373005867004, + 1.5486996173858643, + 1.5279693484306396, + 0.6926574909687043, + 0.3699995279312134, + 1.5489805936813354, + 1.5734963417053223 + ] + }, + "actions": { + "mean": [ + -0.376626310359482, + -0.1896794869715017, + 1.0648562972015143, + -0.10482095168287195, + 1.5297014324008298, + 0.45670789624141594, + 1.202218663212303, + 0.11828392059331577, + 0.11605526081759913, + -0.060935248274749776, + 0.8156495378104216, + -0.10538599961239335, + 0.06141180933068519, + 0.24326971011709353, + 0.7794533919698591, + -0.2979807640868894, + 0.18363962829539562, + 0.42424180993256455, + 0.6465447396441776, + -0.2504673332712931, + 0.3755653989470177, + 0.46749053364538196, + 0.31927322372965633, + -0.2407074871401445, + 0.33870997128534885, + 0.6419578476305398, + -0.38519563826825703, + 0.2347552502921017, + 1.128162579612881, + 0.02282570789627493, + 1.4786149423760955, + -0.2815865832160827, + 0.7948596988695057, + 0.07643057306849928, + 0.21523796120914815, + -0.20955082425225374, + 0.6126353083470936, + 0.03270213509483886, + 0.08516661267224432, + 0.06013077903298965, + 0.5808203849219091, + 0.1292611605930498, + 0.1393399274975366, + 0.1590154204739216, + 0.43629029647408674, + 0.0992215753844696, + 0.2762483523335857, + 0.21882903300477555, + 0.16232854603030022, + 0.0845716416155762, + 0.2382605466091275, + 0.28931980039472494 + ], + "std": [ + 0.09560598287600974, + 0.11611520096487093, + 0.0937020269608546, + 0.5648954223954181, + 0.48806980992043464, + 0.5395921950804928, + 0.27459251423523257, + 0.14676193698804024, + 0.16892566153647134, + 0.2576225289085267, + 0.3651308272584289, + 0.11139296342745225, + 0.20665577105032237, + 0.3370318453580047, + 0.4081629450988825, + 0.09433223534485435, + 0.33182218836588595, + 0.3671281602184461, + 0.37412804122256454, + 0.10560341601364016, + 0.47921616653065974, + 0.35939740692787003, + 0.29163873601321155, + 0.1200962481761208, + 0.40265078965724044, + 0.5321702767021225, + 0.08402046786813083, + 0.09845117783340505, + 0.19821515665692974, + 0.4352978834727, + 0.3930351512873156, + 0.5047210113866738, + 0.5193521281735162, + 0.18672818893678325, + 0.3873173213393463, + 0.22774631272484888, + 0.4887464863917981, + 0.12907470531189552, + 0.2730756972055369, + 0.23931972625270703, + 0.48247024156884183, + 0.14776450080705147, + 0.3356605548408224, + 0.27261334014539124, + 0.4019271874858559, + 0.14978439107806918, + 0.44334886843558086, + 0.2752480818160851, + 0.204783442405884, + 0.13321289238448752, + 0.37533695397780287, + 0.3952954945348143 + ], + "q01": [ + -0.6284304916858673, + -0.4935945984721184, + 0.8788384944200516, + -1.5263540470600128, + 0.13587358564138413, + -0.810822919011116, + 0.5762723177671433, + -0.1658976972103119, + -0.17228451177477835, + -0.49319905042648315, + -0.026840169746428725, + -0.369540833234787, + -0.15237827181816102, + -0.0397159118950367, + -0.07345753587782383, + -0.49499121338129043, + -0.13379494681954385, + 0.056022809073328975, + -0.141864345818758, + -0.49421037077903746, + -0.09221774794161319, + 0.06080367930233479, + -0.32654102832078935, + -0.49498996168375015, + -0.02734675496816635, + -0.04199452336877584, + -0.6296214866638183, + -0.030295154359191657, + 0.8543106687068939, + -1.2677503991127015, + 0.08933606140315532, + -1.7730140936374665, + -0.04479999467730522, + -0.16589999198913574, + -0.4931999444961548, + -0.4931999742984772, + -0.14921643495559692, + -0.30577933192253115, + -0.2816772013902664, + -0.18585201889276504, + -0.1952996338903904, + -0.2435160630941391, + -0.21160183802247048, + -0.07491627089679241, + -0.25988535195589063, + -0.2528540062904358, + -0.11482699774205685, + -1.4544694355578258e-8, + -0.29116781055927277, + -0.25899167090654374, + -0.04005385063588619, + -0.017335239853709935 + ], + "q99": [ + -0.11468523986637588, + 0.07557138144969966, + 1.31752610206604, + 1.5296190762519868, + 2.470554656982422, + 1.946338435411454, + 1.6507995128631592, + 0.5138123136758804, + 0.8935712379217154, + 0.6885862237215054, + 1.4421009695529938, + 0.16915209099650436, + 0.9969354510307319, + 1.4618491971492775, + 1.4391019654273989, + -0.06469714924693079, + 1.259181746244431, + 1.627199411392212, + 1.2873228704929354, + -0.018076832238584414, + 1.6271990537643433, + 1.6182453036308289, + 0.9359641551971438, + 0.018660953342914947, + 1.5595788753032689, + 1.6271997690200806, + -0.1287964740395541, + 0.5003932386636735, + 1.5470348596572876, + 1.1703107023239143, + 2.273779118061066, + 0.8127716386318234, + 1.6507998704910278, + 0.7921539020538335, + 1.312040345668793, + 0.3720327082276346, + 1.5511150670051574, + 0.35186077058315296, + 1.4273963451385505, + 1.378677935600281, + 1.537896139621735, + 0.48847823560237985, + 1.6271998882293701, + 1.6271990549564361, + 1.2952128291130067, + 0.4889179104566576, + 1.6271998882293701, + 1.550438606739048, + 0.6778001552820212, + 0.41327032834291466, + 1.6271997690200806, + 1.6271998882293701 + ] + } + } +} \ No newline at end of file diff --git a/spatial/90000/params/_METADATA b/spatial/90000/params/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..9a6e8dbd04a9a2292594cad28a9f16e18f5dd881 --- /dev/null +++ b/spatial/90000/params/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 269]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 288, 1076]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 269, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4, 72, 288]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 72, 16, 72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [14, 14, 3, 72]}}, "('params', 'PaliGemma', 'img', 'head', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'PaliGemma', 'img', 'head', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [288, 512]}}, "('params', 'PaliGemma', 'img', 'pos_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 16, 1152]}}, "('params', 'PaliGemma', 'llm', 'embedder', 'input_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16072, 2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [192]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 768]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 256, 512]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 256, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 128, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 64, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 512, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 256, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 512, 4096]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 256, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 256, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 128]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 192]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 256, 768]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 128]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 192]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 256, 768]}}, "('params', 'action_in_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('params', 'action_in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [52, 64]}}, "('params', 'action_out_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [52]}}, "('params', 'action_out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64, 52]}}, "('params', 'align_projector', 'fc1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "align_projector", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'align_projector', 'fc1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "align_projector", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 512]}}, "('params', 'align_projector', 'fc2', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "align_projector", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'align_projector', 'fc2', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "align_projector", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 512]}}, "('params', 'align_projector', 'vlm_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "align_projector", "key_type": 2}, {"key": "vlm_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'align_projector', 'vlm_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "align_projector", "key_type": 2}, {"key": "vlm_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('params', 'time_mlp_in', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_in", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('params', 'time_mlp_in', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_in", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('params', 'time_mlp_out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('params', 'time_mlp_out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_mlp_out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/spatial/90000/params/_sharding b/spatial/90000/params/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..aca8e9cc389f9f2c58c3a3eaea3eb07bf60e1d32 --- /dev/null +++ b/spatial/90000/params/_sharding @@ -0,0 +1 @@ +{"cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null, null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null, null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null, null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcucG9zX2VtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZW1iZWRkZXIuaW5wdXRfZW1iZWRkaW5nLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [\"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5nYXRpbmdfZWluc3VtLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5saW5lYXIudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmdhdGluZ19laW5zdW0udmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmxpbmVhci52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW0udy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW1fMS53LnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\", null], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFsaWduX3Byb2plY3Rvci52bG1fbm9ybS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFsaWduX3Byb2plY3Rvci52bG1fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFsaWduX3Byb2plY3Rvci5mYzEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFsaWduX3Byb2plY3Rvci5mYzEua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFsaWduX3Byb2plY3Rvci5mYzIuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLmFsaWduX3Byb2plY3Rvci5mYzIua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX291dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX291dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX2luLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}","cGFyYW1zLnRpbWVfbWxwX2luLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [4, 4], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [null, \"fsdp\"], \"device_mesh\": {\"mesh\": [[{\"id\": 0}, {\"id\": 1}, {\"id\": 2}, {\"id\": 3}], [{\"id\": 4}, {\"id\": 5}, {\"id\": 6}, {\"id\": 7}], [{\"id\": 8}, {\"id\": 9}, {\"id\": 10}, {\"id\": 11}], [{\"id\": 12}, {\"id\": 13}, {\"id\": 14}, {\"id\": 15}]]}}"} \ No newline at end of file diff --git a/spatial/90000/params/array_metadatas/process_0 b/spatial/90000/params/array_metadatas/process_0 new file mode 100644 index 0000000000000000000000000000000000000000..bab866b8ae96a9812b32d230ea0e98b95576ff24 --- /dev/null +++ b/spatial/90000/params/array_metadatas/process_0 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 269], "chunk_shape": [27, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 288, 1076], "chunk_shape": [27, 288, 1076], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 1152], "chunk_shape": [27, 269, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 4, 72, 288], "chunk_shape": [27, 4, 72, 288], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 72], "chunk_shape": [14, 14, 3, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [288, 512], "chunk_shape": [288, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 16, 1152], "chunk_shape": [1, 16, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [16072, 2048], "chunk_shape": [16072, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.bias.value", "write_shape": [192], "chunk_shape": [192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.kernel.value", "write_shape": [256, 768], "chunk_shape": [256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 2, 256, 512], "chunk_shape": [18, 2, 256, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 128, 256], "chunk_shape": [18, 2, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 64, 256], "chunk_shape": [18, 2, 1, 64, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 2, 512, 256], "chunk_shape": [18, 2, 512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 512, 4096], "chunk_shape": [18, 2, 512, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 1024, 2048], "chunk_shape": [18, 1024, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 256, 1024], "chunk_shape": [18, 2, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 256, 1024], "chunk_shape": [18, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [52, 64], "chunk_shape": [52, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [52], "chunk_shape": [52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [64, 52], "chunk_shape": [64, 52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}]} \ No newline at end of file diff --git a/spatial/90000/params/array_metadatas/process_1 b/spatial/90000/params/array_metadatas/process_1 new file mode 100644 index 0000000000000000000000000000000000000000..bab866b8ae96a9812b32d230ea0e98b95576ff24 --- /dev/null +++ b/spatial/90000/params/array_metadatas/process_1 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 269], "chunk_shape": [27, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 288, 1076], "chunk_shape": [27, 288, 1076], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 1152], "chunk_shape": [27, 269, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 4, 72, 288], "chunk_shape": [27, 4, 72, 288], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 72], "chunk_shape": [14, 14, 3, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [288, 512], "chunk_shape": [288, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 16, 1152], "chunk_shape": [1, 16, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [16072, 2048], "chunk_shape": [16072, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.bias.value", "write_shape": [192], "chunk_shape": [192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.kernel.value", "write_shape": [256, 768], "chunk_shape": [256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 2, 256, 512], "chunk_shape": [18, 2, 256, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 128, 256], "chunk_shape": [18, 2, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 64, 256], "chunk_shape": [18, 2, 1, 64, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 2, 512, 256], "chunk_shape": [18, 2, 512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 512, 4096], "chunk_shape": [18, 2, 512, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 1024, 2048], "chunk_shape": [18, 1024, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 256, 1024], "chunk_shape": [18, 2, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 256, 1024], "chunk_shape": [18, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [52, 64], "chunk_shape": [52, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [52], "chunk_shape": [52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [64, 52], "chunk_shape": [64, 52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}]} \ No newline at end of file diff --git a/spatial/90000/params/array_metadatas/process_2 b/spatial/90000/params/array_metadatas/process_2 new file mode 100644 index 0000000000000000000000000000000000000000..bab866b8ae96a9812b32d230ea0e98b95576ff24 --- /dev/null +++ b/spatial/90000/params/array_metadatas/process_2 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 269], "chunk_shape": [27, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 288, 1076], "chunk_shape": [27, 288, 1076], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 1152], "chunk_shape": [27, 269, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 4, 72, 288], "chunk_shape": [27, 4, 72, 288], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 72], "chunk_shape": [14, 14, 3, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [288, 512], "chunk_shape": [288, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 16, 1152], "chunk_shape": [1, 16, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [16072, 2048], "chunk_shape": [16072, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.bias.value", "write_shape": [192], "chunk_shape": [192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.kernel.value", "write_shape": [256, 768], "chunk_shape": [256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 2, 256, 512], "chunk_shape": [18, 2, 256, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 128, 256], "chunk_shape": [18, 2, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 64, 256], "chunk_shape": [18, 2, 1, 64, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 2, 512, 256], "chunk_shape": [18, 2, 512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 512, 4096], "chunk_shape": [18, 2, 512, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 1024, 2048], "chunk_shape": [18, 1024, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 256, 1024], "chunk_shape": [18, 2, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 256, 1024], "chunk_shape": [18, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [52, 64], "chunk_shape": [52, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [52], "chunk_shape": [52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [64, 52], "chunk_shape": [64, 52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}]} \ No newline at end of file diff --git a/spatial/90000/params/array_metadatas/process_3 b/spatial/90000/params/array_metadatas/process_3 new file mode 100644 index 0000000000000000000000000000000000000000..bab866b8ae96a9812b32d230ea0e98b95576ff24 --- /dev/null +++ b/spatial/90000/params/array_metadatas/process_3 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 269], "chunk_shape": [27, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 288, 1076], "chunk_shape": [27, 288, 1076], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 1152], "chunk_shape": [27, 269, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 72], "chunk_shape": [27, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 4, 72, 288], "chunk_shape": [27, 4, 72, 288], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 1, 72], "chunk_shape": [27, 1, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 72, 16, 72], "chunk_shape": [27, 72, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [72], "chunk_shape": [72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 72], "chunk_shape": [14, 14, 3, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [288, 512], "chunk_shape": [288, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 16, 1152], "chunk_shape": [1, 16, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [16072, 2048], "chunk_shape": [16072, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.bias.value", "write_shape": [192], "chunk_shape": [192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.Dense_0.kernel.value", "write_shape": [256, 768], "chunk_shape": [256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 2, 256, 512], "chunk_shape": [18, 2, 256, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 128, 256], "chunk_shape": [18, 2, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 64, 256], "chunk_shape": [18, 2, 1, 64, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 2, 512, 256], "chunk_shape": [18, 2, 512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 2, 256, 256], "chunk_shape": [18, 2, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 512, 4096], "chunk_shape": [18, 2, 512, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 1024, 2048], "chunk_shape": [18, 1024, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 256, 1024], "chunk_shape": [18, 2, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 256, 1024], "chunk_shape": [18, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 128], "chunk_shape": [18, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.bias.value", "write_shape": [18, 192], "chunk_shape": [18, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.Dense_0.kernel.value", "write_shape": [18, 256, 768], "chunk_shape": [18, 256, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [52, 64], "chunk_shape": [52, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [52], "chunk_shape": [52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [64, 52], "chunk_shape": [64, 52], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc1.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.fc2.kernel.value", "write_shape": [512, 512], "chunk_shape": [512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.bias.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.align_projector.vlm_norm.scale.value", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_in.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.bias.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_mlp_out.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}]} \ No newline at end of file diff --git a/spatial/90000/params/d/5758f6bde544bca225f7e10c13eb0f2a b/spatial/90000/params/d/5758f6bde544bca225f7e10c13eb0f2a new file mode 100644 index 0000000000000000000000000000000000000000..5cd1b01eb2fffbf5f0144afd73de8d522451a31f Binary files /dev/null and b/spatial/90000/params/d/5758f6bde544bca225f7e10c13eb0f2a differ diff --git a/spatial/90000/params/manifest.ocdbt b/spatial/90000/params/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..855566df542201445ff0484bf697dfb5e202d5c8 Binary files /dev/null and b/spatial/90000/params/manifest.ocdbt differ diff --git a/spatial/90000/params/ocdbt.process_0/d/60663c37c011f8adfefb2389a28677cf b/spatial/90000/params/ocdbt.process_0/d/60663c37c011f8adfefb2389a28677cf new file mode 100644 index 0000000000000000000000000000000000000000..80bb5c82b6a5aeb40eaa9c076bbb7740845b5ec2 Binary files /dev/null and b/spatial/90000/params/ocdbt.process_0/d/60663c37c011f8adfefb2389a28677cf differ diff --git a/spatial/90000/params/ocdbt.process_0/d/63b7410b868fe558e0fc3ea9ded5b7f2 b/spatial/90000/params/ocdbt.process_0/d/63b7410b868fe558e0fc3ea9ded5b7f2 new file mode 100644 index 0000000000000000000000000000000000000000..5053fc3a01396b67e6c61fb2e47ead40a85a1b20 Binary files /dev/null and b/spatial/90000/params/ocdbt.process_0/d/63b7410b868fe558e0fc3ea9ded5b7f2 differ diff --git a/spatial/90000/params/ocdbt.process_0/d/78399c32470191978bcca35a8c83fc0e b/spatial/90000/params/ocdbt.process_0/d/78399c32470191978bcca35a8c83fc0e new file mode 100644 index 0000000000000000000000000000000000000000..3f6fcdf05426446501f1e6be3ece658a3de64c13 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_0/d/78399c32470191978bcca35a8c83fc0e @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4606ce520891aeb1776a4412cf06a390f6fb926d1ce97e0cec8b1af3afaead7c +size 501993056 diff --git a/spatial/90000/params/ocdbt.process_0/d/94544c6c69e3673070d0c89660512d14 b/spatial/90000/params/ocdbt.process_0/d/94544c6c69e3673070d0c89660512d14 new file mode 100644 index 0000000000000000000000000000000000000000..2967e876db2bf89fb389f452ecefc06e05b506f1 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_0/d/94544c6c69e3673070d0c89660512d14 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:79563fb80bd280ae3666ea5913ecd77b5acc90cb5d06d9b8dc600e2247bb6b5e +size 1120144218 diff --git a/spatial/90000/params/ocdbt.process_0/d/a89343218106dd050342c985a3aca09c b/spatial/90000/params/ocdbt.process_0/d/a89343218106dd050342c985a3aca09c new file mode 100644 index 0000000000000000000000000000000000000000..0bcd1e8e3fb34baf9ca02bb21dbd2369114f21f0 Binary files /dev/null and b/spatial/90000/params/ocdbt.process_0/d/a89343218106dd050342c985a3aca09c differ diff --git a/spatial/90000/params/ocdbt.process_0/d/e8f29bce7fcbd02230364a5ea6a1019a b/spatial/90000/params/ocdbt.process_0/d/e8f29bce7fcbd02230364a5ea6a1019a new file mode 100644 index 0000000000000000000000000000000000000000..28b0706628d0348e471f114420d8c11b7b492e06 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_0/d/e8f29bce7fcbd02230364a5ea6a1019a @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:686dbcb15e55a2d73d1735d55c62dad76447063d795181bc6c98f6cde063d141 +size 1495766977 diff --git a/spatial/90000/params/ocdbt.process_0/manifest.ocdbt b/spatial/90000/params/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..1cbbff5d5735e240d435254b637de59ba1d0052f Binary files /dev/null and b/spatial/90000/params/ocdbt.process_0/manifest.ocdbt differ diff --git a/spatial/90000/params/ocdbt.process_1/d/2d4f1917cf7734421195df45de938703 b/spatial/90000/params/ocdbt.process_1/d/2d4f1917cf7734421195df45de938703 new file mode 100644 index 0000000000000000000000000000000000000000..84bc775d0335dfe4007d590ea66c29c88258c567 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_1/d/2d4f1917cf7734421195df45de938703 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e43c44f69d560bd675ddd1a0af63bc44962e76f4e3e3c12af08a1180609f825 +size 2169544957 diff --git a/spatial/90000/params/ocdbt.process_1/d/3d04da5b957a58495910107c08f62eb0 b/spatial/90000/params/ocdbt.process_1/d/3d04da5b957a58495910107c08f62eb0 new file mode 100644 index 0000000000000000000000000000000000000000..9a64163208183950c5fd0d942dba0946eaf14c08 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_1/d/3d04da5b957a58495910107c08f62eb0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5b83cdf876ef0e123d9fe2c093772e15544f812ae9b97f6f0c5c6b084ee393b1 +size 343115183 diff --git a/spatial/90000/params/ocdbt.process_1/d/4806e3f1f00cbd411a2988a502d1e48c b/spatial/90000/params/ocdbt.process_1/d/4806e3f1f00cbd411a2988a502d1e48c new file mode 100644 index 0000000000000000000000000000000000000000..632d337eefaf9d048240c3d4257011bdf03ca3a6 Binary files /dev/null and b/spatial/90000/params/ocdbt.process_1/d/4806e3f1f00cbd411a2988a502d1e48c differ diff --git a/spatial/90000/params/ocdbt.process_1/d/5ee85dd1362bf7aba4076154adee71a0 b/spatial/90000/params/ocdbt.process_1/d/5ee85dd1362bf7aba4076154adee71a0 new file mode 100644 index 0000000000000000000000000000000000000000..4f1d6945d6c68d9e066f9ff43ab78ca222fb2898 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_1/d/5ee85dd1362bf7aba4076154adee71a0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd521773fa0d876989cf4c2588d19da1bea543ee5562248a75efb3a624f99170 +size 605284149 diff --git a/spatial/90000/params/ocdbt.process_1/manifest.ocdbt b/spatial/90000/params/ocdbt.process_1/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..121c5e228a311467c62d9e293a20e523cf43411f Binary files /dev/null and b/spatial/90000/params/ocdbt.process_1/manifest.ocdbt differ diff --git a/spatial/90000/params/ocdbt.process_2/d/66dfccfde44065dac392e616eb1d885f b/spatial/90000/params/ocdbt.process_2/d/66dfccfde44065dac392e616eb1d885f new file mode 100644 index 0000000000000000000000000000000000000000..a5508c9f4006922993b02a67943769edb530399c Binary files /dev/null and b/spatial/90000/params/ocdbt.process_2/d/66dfccfde44065dac392e616eb1d885f differ diff --git a/spatial/90000/params/ocdbt.process_2/d/7d04930ea03442e748d46f42d74efa97 b/spatial/90000/params/ocdbt.process_2/d/7d04930ea03442e748d46f42d74efa97 new file mode 100644 index 0000000000000000000000000000000000000000..66d76478c856375b7e0a9a2876de7aa076d5917e --- /dev/null +++ b/spatial/90000/params/ocdbt.process_2/d/7d04930ea03442e748d46f42d74efa97 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cb4a61553660ff679cbf09fbdd1736ea737629795a02b4e3996ed869a3a236a2 +size 2169494302 diff --git a/spatial/90000/params/ocdbt.process_2/d/bdc43f2dba4aecfaacc924950eb72435 b/spatial/90000/params/ocdbt.process_2/d/bdc43f2dba4aecfaacc924950eb72435 new file mode 100644 index 0000000000000000000000000000000000000000..04b46fb105d59b1cec9072c354ab9d0e2c8b8a03 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_2/d/bdc43f2dba4aecfaacc924950eb72435 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:38fcaac1b7947e6a97f46ed71f06dfc408098b199aa2ab235c452638a990451d +size 240349801 diff --git a/spatial/90000/params/ocdbt.process_2/d/fe81f44cea20cac46cd5f97dea11d172 b/spatial/90000/params/ocdbt.process_2/d/fe81f44cea20cac46cd5f97dea11d172 new file mode 100644 index 0000000000000000000000000000000000000000..e74f3e396795cfa37f727c1473b6fb184613ade1 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_2/d/fe81f44cea20cac46cd5f97dea11d172 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5e722cda7e266a8273b4a2953a08968808c1e8f0988eeac516137441734e2b54 +size 708072290 diff --git a/spatial/90000/params/ocdbt.process_2/manifest.ocdbt b/spatial/90000/params/ocdbt.process_2/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..c064bac1dd0147d10c176f4922f8c59fa6dacb08 Binary files /dev/null and b/spatial/90000/params/ocdbt.process_2/manifest.ocdbt differ diff --git a/spatial/90000/params/ocdbt.process_3/d/9314deaca83ddf9173de247009462604 b/spatial/90000/params/ocdbt.process_3/d/9314deaca83ddf9173de247009462604 new file mode 100644 index 0000000000000000000000000000000000000000..6e5c26a8aeecb441ccebc09a802c4720b2e41cd1 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_3/d/9314deaca83ddf9173de247009462604 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8f8ff30a069fa5702af43956dd4cda66dd2f7f5780be40b1f3253ef7467b7c5a +size 1122712261 diff --git a/spatial/90000/params/ocdbt.process_3/d/dd7f545e8fc2f34890902ddad9662b61 b/spatial/90000/params/ocdbt.process_3/d/dd7f545e8fc2f34890902ddad9662b61 new file mode 100644 index 0000000000000000000000000000000000000000..c4d6571a1865c68f8789061495ddb4d20d57d561 --- /dev/null +++ b/spatial/90000/params/ocdbt.process_3/d/dd7f545e8fc2f34890902ddad9662b61 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3afad72d5a483606f2916f1840ef561bd17948f1879d101bb201be2747a72ca8 +size 314870173 diff --git a/spatial/90000/params/ocdbt.process_3/d/fe58466ddb619f4c9c9a70e35dae662b b/spatial/90000/params/ocdbt.process_3/d/fe58466ddb619f4c9c9a70e35dae662b new file mode 100644 index 0000000000000000000000000000000000000000..bcd68dd5c2166583dc728fd0eab7602d10c1a4dc --- /dev/null +++ b/spatial/90000/params/ocdbt.process_3/d/fe58466ddb619f4c9c9a70e35dae662b @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ef087b39d56b0ec7df160f53681d4daab46eed014655b52fe5579b604126a32b +size 1680245951 diff --git a/spatial/90000/params/ocdbt.process_3/manifest.ocdbt b/spatial/90000/params/ocdbt.process_3/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..193c5ed0ece6e0a1da6ecdb6b7fe3517330ef411 Binary files /dev/null and b/spatial/90000/params/ocdbt.process_3/manifest.ocdbt differ