Instructions to use nvidia/Cosmos3-Nano-Sim-Bimanual with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Cosmos
How to use nvidia/Cosmos3-Nano-Sim-Bimanual with Cosmos:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
- Notebooks
- Google Colab
- Kaggle
Commit ·
441c3ef
0
Parent(s):
Initial model release
Browse filesCo-authored-by: zhengfei-nvidia <zhengfei-nvidia@users.noreply.huggingface.co>
Co-authored-by: yaoxuhf <yaoxuhf@users.noreply.huggingface.co>
Co-authored-by: qmiao <qmiao@users.noreply.huggingface.co>
Co-authored-by: yzhautouskay <yzhautouskay@users.noreply.huggingface.co>
Co-authored-by: mbalaNV <mbalaNV@users.noreply.huggingface.co>
This view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +65 -0
- README.md +571 -0
- assets/output/cosmos3-nano-sim-bimanual-quickstart.mp4 +3 -0
- assets/sim_bimanual/agibot/actions/agibot_001_normalized_unifiedv1.json +0 -0
- assets/sim_bimanual/agibot/actions/agibot_034_normalized_unifiedv1.json +0 -0
- assets/sim_bimanual/agibot/actions/agibot_mixed_007_normalized_mixed_action.json +0 -0
- assets/sim_bimanual/agibot/actions/agibot_mixed_028_normalized_mixed_action.json +0 -0
- assets/sim_bimanual/agibot/actions/agibot_original_long60_019_normalized_unifiedv1.json +0 -0
- assets/sim_bimanual/agibot/images/agibot_001_initial.png +3 -0
- assets/sim_bimanual/agibot/images/agibot_034_initial.png +3 -0
- assets/sim_bimanual/agibot/images/agibot_mixed_007_initial.png +3 -0
- assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png +3 -0
- assets/sim_bimanual/agibot/images/agibot_original_long60_019_initial.png +3 -0
- assets/sim_bimanual/yam/actions/yam_019_normalized_unifiedv1.json +0 -0
- assets/sim_bimanual/yam/actions/yam_023_normalized_unifiedv1.json +0 -0
- assets/sim_bimanual/yam/images/yam_019_initial.png +3 -0
- assets/sim_bimanual/yam/images/yam_023_initial.png +3 -0
- assets/training/agibot/actions/agibot_001_normalized_unifiedv1.json +0 -0
- assets/training/agibot/actions/agibot_006_normalized_unifiedv1.json +0 -0
- assets/training/agibot/actions/agibot_034_normalized_unifiedv1.json +0 -0
- assets/training/agibot/gt/agibot_001_gt.mp4 +3 -0
- assets/training/agibot/gt/agibot_006_gt.mp4 +3 -0
- assets/training/agibot/gt/agibot_034_gt.mp4 +3 -0
- assets/training/agibot/images/agibot_001_initial.png +3 -0
- assets/training/agibot/images/agibot_006_initial.png +3 -0
- assets/training/agibot/images/agibot_034_initial.png +3 -0
- bias.md +11 -0
- chat_template.json +3 -0
- checkpoint.json +249 -0
- config.json +424 -0
- cosmos3_nano_sim_bimanual_action_sources/4d38b6321a92741c93e8cf1e96cba18979399ddb5c037708d3aaeffd666faa10.json +41 -0
- cosmos3_nano_sim_bimanual_action_sources/action_schema.json +208 -0
- cosmos3_nano_sim_bimanual_action_sources/ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0.json +162 -0
- cosmos3_nano_sim_bimanual_action_sources/cosmos3_nano_sim_bimanual_action_schema.schema.json +300 -0
- cosmos3_nano_sim_bimanual_action_sources/resolved_action_config.json +97 -0
- cosmos3_nano_sim_bimanual_artifact.json +0 -0
- cosmos3_nano_sim_bimanual_expected_diffusers_weights.json +0 -0
- cosmos3_nano_sim_bimanual_expected_weights.json +0 -0
- cosmos_dreams_action_sources/4d38b6321a92741c93e8cf1e96cba18979399ddb5c037708d3aaeffd666faa10.json +41 -0
- cosmos_dreams_action_sources/action_schema.json +95 -0
- cosmos_dreams_action_sources/cosmos_dreams_action_schema.schema.json +201 -0
- cosmos_dreams_action_sources/resolved_action_config.json +16 -0
- cosmos_dreams_artifact.json +0 -0
- cosmos_dreams_expected_diffusers_weights.json +0 -0
- cosmos_dreams_expected_weights.json +0 -0
- cosmos_dreams_manifest.json +131 -0
- examples/inference/README.md +96 -0
- examples/inference/bimanual_frustum.json +14 -0
- examples/inference/sim_bimanual_agibot_long60_hf.jsonl +1 -0
- examples/inference/sim_bimanual_mixed_action_hf.jsonl +2 -0
.gitattributes
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
text_tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
assets/sim_bimanual/agibot/images/agibot_001_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
assets/sim_bimanual/agibot/images/agibot_006_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
assets/sim_bimanual/agibot/images/agibot_034_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
assets/sim_bimanual/agibot/images/agibot_idx0000009_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
assets/sim_bimanual/agibot/images/agibot_mixed_005_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
assets/sim_bimanual/agibot/images/agibot_mixed_007_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 44 |
+
assets/sim_bimanual/yam/images/yam_005_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 45 |
+
assets/sim_bimanual/yam/images/yam_019_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 46 |
+
assets/sim_bimanual/yam/images/yam_023_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
assets/sim_bimanual/yam/images/yam_idx0000014_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
assets/training/agibot/gt/agibot_001_gt.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
assets/training/agibot/gt/agibot_006_gt.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 50 |
+
assets/training/agibot/gt/agibot_034_gt.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
assets/training/agibot/gt/agibot_mixed_005_gt.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 52 |
+
assets/training/agibot/gt/agibot_mixed_007_gt.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 53 |
+
assets/training/agibot/gt/agibot_mixed_028_gt.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 54 |
+
assets/training/agibot/images/agibot_001_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 55 |
+
assets/training/agibot/images/agibot_006_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 56 |
+
assets/training/agibot/images/agibot_034_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 57 |
+
assets/training/agibot/images/agibot_mixed_005_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 58 |
+
assets/training/agibot/images/agibot_mixed_007_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 59 |
+
assets/training/agibot/images/agibot_mixed_028_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 60 |
+
assets/sim_bimanual/agibot/images/agibot_original_long60_012_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 61 |
+
assets/output/cosmos3-nano-sim-bimanual-quickstart.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 62 |
+
assets/sim_bimanual/agibot/images/agibot_original_long60_019_initial.png filter=lfs diff=lfs merge=lfs -text
|
| 63 |
+
images/cosmos3-nano-sim-bimanual-graph.png filter=lfs diff=lfs merge=lfs -text
|
| 64 |
+
images/cosmos3-nano-sim-bimanual-overview.gif filter=lfs diff=lfs merge=lfs -text
|
| 65 |
+
images/cosmos3-nano-sim-bimanual-overview.mp4 filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,571 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
license_name: openmdw1.1-license
|
| 4 |
+
license_link: >-
|
| 5 |
+
https://openmdw.ai/license/1-1/
|
| 6 |
+
library_name: cosmos
|
| 7 |
+
tags:
|
| 8 |
+
- nvidia
|
| 9 |
+
- cosmos
|
| 10 |
+
- cosmos3
|
| 11 |
+
- omnimodel
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
# **Interactive World Simulation with Cosmos for Physical AI**
|
| 15 |
+
|
| 16 |
+
**[Cosmos3 Examples Collection](https://huggingface.co/collections/nvidia/cosmos3-examples)** | **[Code](https://github.com/nvidia/cosmos)** | **[Technical Report](https://research.nvidia.com/labs/cosmos-lab/cosmos-simulation/technical-report.pdf)** (coming soon) | **[Website](https://research.nvidia.com/labs/cosmos-lab/cosmos-simulation/)**
|
| 17 |
+
|
| 18 |
+
# Model Overview: Cosmos3-Nano-Sim-Bimanual
|
| 19 |
+
|
| 20 |
+

|
| 21 |
+
|
| 22 |
+
The highlighted checkpoint, Cosmos3-Nano-Sim-Bimanual, is a post-trained, causal-distilled variant of the Cosmos3-Nano base model.
|
| 23 |
+
|
| 24 |
+
## Description
|
| 25 |
+
|
| 26 |
+
<p align="center">
|
| 27 |
+
<video src="https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/images/cosmos3-nano-sim-bimanual-overview.mp4" controls autoplay loop muted playsinline width="640" aria-label="Generated results only: AgiBot 008 and 006, seed 2026, on the top row; YAM 001, seed 2025, and YAM 008, seed 2027, on the bottom row. First 15 seconds at original speed."></video>
|
| 28 |
+
<br>
|
| 29 |
+
</p>
|
| 30 |
+
|
| 31 |
+
Cosmos3-Nano-Sim-Bimanual simulates how an environment evolves in response to a bimanual robot’s actions. It supports interleaved camera navigation and robot manipulation within a single action sequence in both open-loop and closed-loop settings, enabling continuous simulation across changes in camera pose and robot arm or gripper state. Given an initial image and a sequence of control inputs, the model generates video in small latent-frame chunks conditioned on cached history and incoming robot actions or camera trajectories. Control inputs can be supplied at any point during a rollout by a policy model, such as a vision-language-action (VLA) model or a world-action model, or by a human-operated joystick. These capabilities support robotic policy evaluation and verification, reinforcement learning (RL) training, and synthetic demonstration data generation without teleoperation when control inputs are supplied by a policy model.
|
| 32 |
+
|
| 33 |
+
Cosmos3-Nano-Sim-Bimanual was developed by NVIDIA as part of the Cosmos example model family.
|
| 34 |
+
|
| 35 |
+
For a quick start, see the [Cosmos3-Nano-Sim-Bimanual skill](https://github.com/NVIDIA/cosmos-framework/tree/main/.agents/skills/cosmos3-nano-sim-bimanual).
|
| 36 |
+
|
| 37 |
+
This model is ready for commercial or non-commercial use.
|
| 38 |
+
|
| 39 |
+
## License
|
| 40 |
+
|
| 41 |
+
This model is released under [OpenMDW1.1](https://openmdw.ai/license/1-1/).
|
| 42 |
+
|
| 43 |
+
## Deployment Geography
|
| 44 |
+
|
| 45 |
+
Global
|
| 46 |
+
|
| 47 |
+
## Use Case
|
| 48 |
+
|
| 49 |
+
Physical AI for robotics: policy evaluation and verification, reinforcement learning (RL) training, and synthetic demonstration data generation with interleaved camera navigation and bimanual manipulation actions. Policy-generated control inputs enable demonstration generation without teleoperation.
|
| 50 |
+
|
| 51 |
+
## Release Date
|
| 52 |
+
|
| 53 |
+
- Hugging Face 10/20/2026 via [https://huggingface.co/collections/nvidia/cosmos3-examples](https://huggingface.co/collections/nvidia/cosmos3-examples)
|
| 54 |
+
- GitHub 10/20/2026 via [https://github.com/nvidia/cosmos](https://github.com/nvidia/cosmos)
|
| 55 |
+
|
| 56 |
+
## Model Architecture
|
| 57 |
+
|
| 58 |
+
**Architecture Type:** Transformer
|
| 59 |
+
|
| 60 |
+
**Network Architecture:** Mixture-of-Transformers (MoT)
|
| 61 |
+
|
| 62 |
+
Cosmos3-Nano-Sim-Bimanual is derived from [Cosmos3-Nano](https://huggingface.co/nvidia/Cosmos3-Nano) through post-training on a mixture of robot manipulation videos paired with robot actions and camera navigation videos paired with camera trajectories, followed by causal distillation. It generates video in small latent-frame chunks conditioned on cached history and incoming control inputs. The model supports interleaved robot manipulation and camera navigation within a single action sequence, maintaining scene continuity across robot motion and viewpoint changes. Additional post-training on user-provided data can adapt the model to different robot embodiments and arm configurations; see the [Cosmos3-Nano-Sim-Bimanual skill](https://github.com/NVIDIA/cosmos-framework/tree/main/.agents/skills/cosmos3-nano-sim-bimanual) for details.
|
| 63 |
+
|
| 64 |
+
This model was developed using the [Cosmos Framework](https://github.com/NVIDIA/cosmos-framework).
|
| 65 |
+
|
| 66 |
+
**Number of trainable model parameters:** 16B
|
| 67 |
+
|
| 68 |
+
## Inputs
|
| 69 |
+
|
| 70 |
+
- **Input Type(s):** Image, robot actions, camera trajectories, text (optional scene description)
|
| 71 |
+
- **Input Format(s):**
|
| 72 |
+
- **Image:** JPG, PNG, JPEG, or WebP (initial frame)
|
| 73 |
+
- **Robot actions:** JSON containing a sequence of robot state or control values with shape `(T, D)`, where `T` is the number of action steps and `D` is the embodiment-specific action dimensionality
|
| 74 |
+
- **Camera trajectories:** Sequence of camera poses (for example, a 9D camera-motion representation)
|
| 75 |
+
- **Text:** String
|
| 76 |
+
- **Input Parameters:**
|
| 77 |
+
- **Image:** Two-dimensional (2D)
|
| 78 |
+
- **Robot actions / Camera trajectories:** Two-dimensional sequences (time steps × control dimensions)
|
| 79 |
+
- **Text:** One-dimensional (1D)
|
| 80 |
+
- **Other Properties Related to Input:**
|
| 81 |
+
- Actions are aligned with the generated video.
|
| 82 |
+
- Supported embodiment: AgiBot
|
| 83 |
+
|
| 84 |
+
## Outputs
|
| 85 |
+
|
| 86 |
+
- **Output Type(s):** Video
|
| 87 |
+
- **Output Format(s):**
|
| 88 |
+
- **Video:** MP4 or streamed video frames
|
| 89 |
+
- **Output Parameters:**
|
| 90 |
+
- **Video:** Three-dimensional (3D)
|
| 91 |
+
- **Other Properties Related to Output:**
|
| 92 |
+
- Generates video at 480p with an output frame rate of 30 frames per second (fps).
|
| 93 |
+
- Output video frame rates describe playback timing; inference throughput is reported separately.
|
| 94 |
+
|
| 95 |
+
Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA's hardware (e.g., GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions.
|
| 96 |
+
|
| 97 |
+
## Software Integration
|
| 98 |
+
|
| 99 |
+
**Runtime Engine(s):**
|
| 100 |
+
|
| 101 |
+
- [PyTorch](https://pytorch.org/)
|
| 102 |
+
- [vLLM-Omni](https://github.com/vllm-project/vllm-omni)
|
| 103 |
+
|
| 104 |
+
**Supported Hardware Microarchitecture Compatibility:**
|
| 105 |
+
|
| 106 |
+
- NVIDIA Blackwell
|
| 107 |
+
- NVIDIA Hopper
|
| 108 |
+
|
| 109 |
+
**Operating System(s):**
|
| 110 |
+
|
| 111 |
+
- Linux (other operating systems have not been tested).
|
| 112 |
+
|
| 113 |
+
The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
|
| 114 |
+
|
| 115 |
+
## Model Version(s)
|
| 116 |
+
|
| 117 |
+
**Cosmos3-Nano-Sim-Bimanual — V1.0 (10/20/2026)**
|
| 118 |
+
|
| 119 |
+
Cosmos3-Nano-Sim-Bimanual is a post-trained, causal-distilled variant of [Cosmos3-Nano](https://huggingface.co/nvidia/Cosmos3-Nano).
|
| 120 |
+
|
| 121 |
+
It can be integrated into a closed-loop robotic policy evaluation or training pipeline that supplies an initial image and a sequence of robot actions and camera trajectories from a policy model or human controller. The model supports this interaction by returning generated video frames as visual observations for subsequent policy or controller inputs.
|
| 122 |
+
|
| 123 |
+
## Training and Evaluation Datasets
|
| 124 |
+
|
| 125 |
+
### Post-Training Datasets
|
| 126 |
+
|
| 127 |
+
**Data Modality:** Video, Text, and paired robot-action or camera-trajectory data.
|
| 128 |
+
|
| 129 |
+
**Post-Training Data Size:**
|
| 130 |
+
- **Text:** 1 Billion to 10 Trillion Tokens
|
| 131 |
+
- **Video with paired robot-action or camera-trajectory data:** 10,000 to 1 Million Hours
|
| 132 |
+
|
| 133 |
+
**Data Collection Method:** Hybrid: Automatic/Sensors, Synthetic, Automated.
|
| 134 |
+
|
| 135 |
+
**Labeling Method:** Hybrid: Manually-Labeled, Automated.
|
| 136 |
+
|
| 137 |
+
**Properties:** The post-training datasets consist of robot manipulation and camera navigation videos, associated English captions, robot actions, and camera trajectories. The data includes real and synthetic content from vendor-provided, public, and NVIDIA-created datasets.
|
| 138 |
+
|
| 139 |
+
### Dataset Overview
|
| 140 |
+
|
| 141 |
+
- **Total Size:** 1.3B data points
|
| 142 |
+
- **Total Number of Datasets:** 393 dataset entries
|
| 143 |
+
- **Dataset partition:** Training [100%], Testing [N/A — evaluation benchmarks used separately], Validation [N/A — evaluation benchmarks used separately]
|
| 144 |
+
- **Time period for training data collection:** 2024–2026
|
| 145 |
+
- **Time period for validation data collection:** N/A (standard public benchmarks)
|
| 146 |
+
|
| 147 |
+
Raw data from internal and external sources is transformed into training-ready data through multiple stages of curation, filtering, and quality review. Data acquisition spans diverse multimodal sources — robotics, autonomous driving, industrial environments, indoor and outdoor scenes, varied lighting and weather conditions, camera viewpoints, object categories, and human activities — to broaden coverage across Physical AI operating environments. Automated filtering pipelines remove corrupted, duplicate, low-quality, and restricted content. Metadata analysis, heuristic rules, and model-assisted classifiers are applied during preprocessing to flag anomalous distributions and low-diversity subsets. Human review supplements automated filtering for selected datasets, benchmark construction, and targeted quality analysis. Datasets are balanced across modalities and task categories — visual reasoning, text-to-image, text-to-video, image-to-video, audio generation, video transfer, action-conditioned generation, and action command generation — to reduce overrepresentation of narrow domains. Synthetic and simulation-based augmentation supplements coverage of rare physical interactions and edge-case scenarios. Deduplication and provenance tracking are applied across the corpus. The resulting processed data is converted into model-ready tokenized or encoded representations through modality-specific preprocessors before training begins.
|
| 148 |
+
|
| 149 |
+
Training datasets passed through multiple layers of automated and manual safeguards designed to reduce the presence of harmful or policy-violating content across categories including weapons and weapons-related instructional content, criminal planning, child sexual abuse material (CSAM), non-consensual intimate imagery (NCII), sexual content involving minors, harassment, hate speech, profanity, threats and incitement to violence, self-harm or suicide-related content, and graphic violence. Data sources are reviewed for licensing compatibility, provenance, and alignment with internal data governance and safety policies before admission into training corpora. Automated filtering pipelines combine multiple detection strategies: hash-matching against known CSAM and NCII reference databases; classifier-based moderation models trained for explicit sexual content, hate speech, violence, weapons imagery, and other restricted categories; keyword and regex-based screening for criminal-planning, threats, and self-harm phrases in text data; metadata and provenance heuristics for source-level risk signals; and embedding-based anomaly detection to surface samples that fall outside expected distributions. Human review and targeted audits supplement automated filtering for selected datasets, benchmark construction, and safety-sensitive evaluation. For multimodal Physical AI data (robotics, autonomous driving, industrial scenes), additional filtering targets invalid action trajectories, physically implausible interactions, and unsafe control sequences. Synthetic and simulation-generated data are evaluated through internal validation before inclusion. Benchmark evaluations and red-team testing are applied post-training to surface remaining safety gaps across world generation, reasoning, audio, and action tasks. No large-scale data-filtering process can guarantee complete removal of all harmful content; residual risks may remain, particularly in rare edge cases or open-world deployment settings. Ongoing monitoring and dataset review continue post-release.
|
| 150 |
+
|
| 151 |
+
For more information about the datasets used to train this model, please see the [Public Summary of Training Content](https://docs.nvidia.com/cosmos/latest/cosmos3/index.html).
|
| 152 |
+
|
| 153 |
+
**Data Modality and Training Data Size**
|
| 154 |
+
|
| 155 |
+
| Modality | Reasoning Data Sample Count | Generation Data Sample Count |
|
| 156 |
+
| -------- | ------------------- | -------------------- |
|
| 157 |
+
| Text | 22M | Not Applicable |
|
| 158 |
+
| Image | 19M | 767M |
|
| 159 |
+
| Video | 1M | 348M |
|
| 160 |
+
| Audio | Not Applicable | 139M |
|
| 161 |
+
| Action | Not Applicable | 7M |
|
| 162 |
+
|
| 163 |
+
**Data Collection Method by dataset**
|
| 164 |
+
|
| 165 |
+
- Hybrid: Automatic/Sensors, Synthetic, Automated
|
| 166 |
+
|
| 167 |
+
**Labeling Method by dataset**
|
| 168 |
+
|
| 169 |
+
- Hybrid: Manually-Labeled, Automated
|
| 170 |
+
|
| 171 |
+
**Properties:** The training, testing, and evaluation datasets consist of diverse multimodal video, image, audio, action, synthetic, and sensor-conditioned data sourced from NVIDIA-owned data and publicly available, commercially permissive datasets. These datasets are curated to exclude known restricted content and to support building an Omni model that learns to generate and reason about dynamic physical environments across world reasoning and generation tasks.
|
| 172 |
+
|
| 173 |
+
### Public Datasets
|
| 174 |
+
|
| 175 |
+
| Dataset | Samples |
|
| 176 |
+
|---------------------------|-------------|
|
| 177 |
+
| OpenImage | 1.2M |
|
| 178 |
+
| Coyo700M | 100M |
|
| 179 |
+
| YouTube Video | 340M |
|
| 180 |
+
| UMI | 4.5M |
|
| 181 |
+
|
| 182 |
+
### Private Datasets
|
| 183 |
+
|
| 184 |
+
| Dataset | Samples |
|
| 185 |
+
|---------------------------|-------------|
|
| 186 |
+
| Egocentric | 7M |
|
| 187 |
+
| Nexar | 0.6M |
|
| 188 |
+
| AgiBot | 0.2M |
|
| 189 |
+
| HOI | 0.3M |
|
| 190 |
+
|
| 191 |
+
### Synthetic Datasets
|
| 192 |
+
|
| 193 |
+
| Dataset | Samples |
|
| 194 |
+
|-----------------------------------------|-------------|
|
| 195 |
+
| synthetic images generated using HiDream-I1 | 15M |
|
| 196 |
+
| synthetic images generated using Qwen-Image-2512 | 14M |
|
| 197 |
+
| synthetic captions generated using Qwen3-VL | 1115M |
|
| 198 |
+
|
| 199 |
+
## Evaluation Datasets
|
| 200 |
+
|
| 201 |
+
**Data Collection Method by dataset**
|
| 202 |
+
|
| 203 |
+
- Hybrid: Automatic/Sensors, Synthetic, Automated
|
| 204 |
+
|
| 205 |
+
**Labeling Method by dataset**
|
| 206 |
+
|
| 207 |
+
- Hybrid: Manually-Labeled, Automated
|
| 208 |
+
|
| 209 |
+
**Properties:** The training, testing, and evaluation datasets consist of diverse multimodal video, image, audio, action, synthetic, and sensor-conditioned data sourced from NVIDIA-owned data and publicly available, commercially permissive datasets. These datasets are curated to exclude known restricted content and to support building an Omni model that learns to generate and reason about dynamic physical environments across world reasoning and generation tasks.
|
| 210 |
+
|
| 211 |
+
## Quality Benchmark(s)
|
| 212 |
+
|
| 213 |
+
### Camera Navigation Benchmark
|
| 214 |
+
|
| 215 |
+
#### [PAI-Bench-G](https://arxiv.org/abs/2512.01989) Image-to-Video (I2V)
|
| 216 |
+
|
| 217 |
+
We evaluate **Cosmos3-Nano-Sim-Bimanual** against LingBot-World 2.0 and Sana-WM on sustained viewpoint control. Each example supplies an initial image, a scene caption, and a camera trajectory; captions exclude camera motion. The [PAI-Bench-G](https://arxiv.org/abs/2512.01989) quality score measures video fidelity and consistency. [ViPE](https://research.nvidia.com/labs/toronto-ai/vipe/) estimates camera positions and orientations from the generated video. We fit the estimated trajectory to the input trajectory by rotation, translation, and uniform scaling (Umeyama alignment), allowing comparison independent of coordinate frame and scale. Absolute translation error (ATE) measures camera-position error after alignment; relative translation error (RTE) and relative rotation error (RRE) measure errors in camera displacement and orientation change between poses. Perceptual drift evaluates changes in video quality over time.
|
| 218 |
+
|
| 219 |
+
<table>
|
| 220 |
+
<thead>
|
| 221 |
+
<tr>
|
| 222 |
+
<th scope="col" style="text-align: left;">Model</th>
|
| 223 |
+
<th scope="col" style="text-align: right;">Video Quality ↑</th>
|
| 224 |
+
<th scope="col" style="text-align: right;">ATE ↓</th>
|
| 225 |
+
<th scope="col" style="text-align: right;">RTE ↓</th>
|
| 226 |
+
<th scope="col" style="text-align: right;">RRE ↓</th>
|
| 227 |
+
<th scope="col" style="text-align: right;">Perceptual Drift ↓</th>
|
| 228 |
+
</tr>
|
| 229 |
+
</thead>
|
| 230 |
+
<tbody>
|
| 231 |
+
<tr>
|
| 232 |
+
<td style="text-align: left;">LingBot-World 2.0</td>
|
| 233 |
+
<td style="text-align: right;">65.3577</td>
|
| 234 |
+
<td style="text-align: right;"><strong>4.1698</strong></td>
|
| 235 |
+
<td style="text-align: right;">0.0596</td>
|
| 236 |
+
<td style="text-align: right;">0.1781</td>
|
| 237 |
+
<td style="text-align: right;"><strong style="color: inherit;">0.0209</strong></td>
|
| 238 |
+
</tr>
|
| 239 |
+
<tr>
|
| 240 |
+
<td style="text-align: left;">Sana-WM</td>
|
| 241 |
+
<td style="text-align: right;">65.2090</td>
|
| 242 |
+
<td style="text-align: right;">6.4823</td>
|
| 243 |
+
<td style="text-align: right;">0.1284</td>
|
| 244 |
+
<td style="text-align: right;">0.1601</td>
|
| 245 |
+
<td style="text-align: right;">0.0299</td>
|
| 246 |
+
</tr>
|
| 247 |
+
<tr>
|
| 248 |
+
<td style="text-align: left; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">Cosmos3-Nano-Sim-Bimanual</strong></td>
|
| 249 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">73.9099</strong></td>
|
| 250 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;">4.7072</td>
|
| 251 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0545</strong></td>
|
| 252 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.1002</strong></td>
|
| 253 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;">0.0210</td>
|
| 254 |
+
</tr>
|
| 255 |
+
</tbody>
|
| 256 |
+
</table>
|
| 257 |
+
|
| 258 |
+
#### Sana-WM simple
|
| 259 |
+
|
| 260 |
+
The simple split of [Sana-WM-Bench](https://arxiv.org/abs/2605.15178) reports visual quality, camera alignment, and drift. We pair benchmark runs by source inputs and trajectories, using the same metrics as the [PAI-Bench-G](https://arxiv.org/abs/2512.01989) I2V evaluation.
|
| 261 |
+
|
| 262 |
+
<table>
|
| 263 |
+
<thead>
|
| 264 |
+
<tr>
|
| 265 |
+
<th scope="col" style="text-align: left;">Model</th>
|
| 266 |
+
<th scope="col" style="text-align: right;">Video Quality ↑</th>
|
| 267 |
+
<th scope="col" style="text-align: right;">ATE ↓</th>
|
| 268 |
+
<th scope="col" style="text-align: right;">RTE ↓</th>
|
| 269 |
+
<th scope="col" style="text-align: right;">RRE ↓</th>
|
| 270 |
+
<th scope="col" style="text-align: right;">Perceptual Drift ↓</th>
|
| 271 |
+
</tr>
|
| 272 |
+
</thead>
|
| 273 |
+
<tbody>
|
| 274 |
+
<tr>
|
| 275 |
+
<td style="text-align: left;">LingBot-World 2.0</td>
|
| 276 |
+
<td style="text-align: right;">65.9748</td>
|
| 277 |
+
<td style="text-align: right;">1.4254</td>
|
| 278 |
+
<td style="text-align: right;">0.0185</td>
|
| 279 |
+
<td style="text-align: right;">0.2591</td>
|
| 280 |
+
<td style="text-align: right;">0.0371</td>
|
| 281 |
+
</tr>
|
| 282 |
+
<tr>
|
| 283 |
+
<td style="text-align: left;">Sana-WM</td>
|
| 284 |
+
<td style="text-align: right;">66.0599</td>
|
| 285 |
+
<td style="text-align: right;">1.2159</td>
|
| 286 |
+
<td style="text-align: right;">0.0192</td>
|
| 287 |
+
<td style="text-align: right;">0.2356</td>
|
| 288 |
+
<td style="text-align: right;">0.0223</td>
|
| 289 |
+
</tr>
|
| 290 |
+
<tr>
|
| 291 |
+
<td style="text-align: left; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">Cosmos3-Nano-Sim-Bimanual</strong></td>
|
| 292 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">77.2887</strong></td>
|
| 293 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">1.0040</strong></td>
|
| 294 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0124</strong></td>
|
| 295 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0551</strong></td>
|
| 296 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0118</strong></td>
|
| 297 |
+
</tr>
|
| 298 |
+
</tbody>
|
| 299 |
+
</table>
|
| 300 |
+
|
| 301 |
+
#### Sana-WM hard
|
| 302 |
+
|
| 303 |
+
The hard split of [Sana-WM-Bench](https://arxiv.org/abs/2605.15178) reports visual quality, camera alignment, and drift. We pair benchmark runs by source inputs and trajectories, using the same metrics as the simple split.
|
| 304 |
+
|
| 305 |
+
<table>
|
| 306 |
+
<thead>
|
| 307 |
+
<tr>
|
| 308 |
+
<th scope="col" style="text-align: left;">Model</th>
|
| 309 |
+
<th scope="col" style="text-align: right;">Video Quality ↑</th>
|
| 310 |
+
<th scope="col" style="text-align: right;">ATE ↓</th>
|
| 311 |
+
<th scope="col" style="text-align: right;">RTE ↓</th>
|
| 312 |
+
<th scope="col" style="text-align: right;">RRE ↓</th>
|
| 313 |
+
<th scope="col" style="text-align: right;">Perceptual Drift ↓</th>
|
| 314 |
+
</tr>
|
| 315 |
+
</thead>
|
| 316 |
+
<tbody>
|
| 317 |
+
<tr>
|
| 318 |
+
<td style="text-align: left;">LingBot-World 2.0</td>
|
| 319 |
+
<td style="text-align: right;">64.5462</td>
|
| 320 |
+
<td style="text-align: right;">1.3423</td>
|
| 321 |
+
<td style="text-align: right;"><strong>0.0161</strong></td>
|
| 322 |
+
<td style="text-align: right;">0.3264</td>
|
| 323 |
+
<td style="text-align: right;">0.0372</td>
|
| 324 |
+
</tr>
|
| 325 |
+
<tr>
|
| 326 |
+
<td style="text-align: left;">Sana-WM</td>
|
| 327 |
+
<td style="text-align: right;">65.5699</td>
|
| 328 |
+
<td style="text-align: right;">1.3165</td>
|
| 329 |
+
<td style="text-align: right;">0.0178</td>
|
| 330 |
+
<td style="text-align: right;">0.1826</td>
|
| 331 |
+
<td style="text-align: right;">0.0251</td>
|
| 332 |
+
</tr>
|
| 333 |
+
<tr>
|
| 334 |
+
<td style="text-align: left; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">Cosmos3-Nano-Sim-Bimanual</strong></td>
|
| 335 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">76.3971</strong></td>
|
| 336 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">1.3085</strong></td>
|
| 337 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0161</strong></td>
|
| 338 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.1774</strong></td>
|
| 339 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0161</strong></td>
|
| 340 |
+
</tr>
|
| 341 |
+
</tbody>
|
| 342 |
+
</table>
|
| 343 |
+
|
| 344 |
+
### Robot Manipulation Benchmark
|
| 345 |
+
|
| 346 |
+
#### AgiBot-World
|
| 347 |
+
|
| 348 |
+
We evaluate **Cosmos3-Nano-Sim-Bimanual** on observation prediction conditioned on AgiBot-World action sequences, an initial image, and a caption, comparing against the bidirectional DreamDojo 2B and 14B teacher models. The [PAI-Bench-G](https://arxiv.org/abs/2512.01989) quality score measures video fidelity and consistency. PSNR measures pixel-level agreement, SSIM measures image-structure agreement, and [LPIPS](https://arxiv.org/abs/1801.03924) measures differences in learned visual features. These metrics compare generated and reference frames and serve as proxies for action following. Perceptual drift evaluates changes in video quality over time.
|
| 349 |
+
|
| 350 |
+
<table>
|
| 351 |
+
<thead>
|
| 352 |
+
<tr>
|
| 353 |
+
<th scope="col" style="text-align: left;">Model</th>
|
| 354 |
+
<th scope="col" style="text-align: right;">Video Quality ↑</th>
|
| 355 |
+
<th scope="col" style="text-align: right;">PSNR (dB) ↑</th>
|
| 356 |
+
<th scope="col" style="text-align: right;">SSIM ↑</th>
|
| 357 |
+
<th scope="col" style="text-align: right;">LPIPS ↓</th>
|
| 358 |
+
<th scope="col" style="text-align: right;">Perceptual Drift ↓</th>
|
| 359 |
+
</tr>
|
| 360 |
+
</thead>
|
| 361 |
+
<tbody>
|
| 362 |
+
<tr>
|
| 363 |
+
<td style="text-align: left;">DreamDojo 2B</td>
|
| 364 |
+
<td style="text-align: right;">74.5327</td>
|
| 365 |
+
<td style="text-align: right;">12.1016</td>
|
| 366 |
+
<td style="text-align: right;">0.6405</td>
|
| 367 |
+
<td style="text-align: right;">0.3766</td>
|
| 368 |
+
<td style="text-align: right;">0.0149</td>
|
| 369 |
+
</tr>
|
| 370 |
+
<tr>
|
| 371 |
+
<td style="text-align: left;">DreamDojo 14B</td>
|
| 372 |
+
<td style="text-align: right;">74.9390</td>
|
| 373 |
+
<td style="text-align: right;">12.3561</td>
|
| 374 |
+
<td style="text-align: right;">0.6635</td>
|
| 375 |
+
<td style="text-align: right;">0.3586</td>
|
| 376 |
+
<td style="text-align: right;">0.0133</td>
|
| 377 |
+
</tr>
|
| 378 |
+
<tr>
|
| 379 |
+
<td style="text-align: left; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">Cosmos3-Nano-Sim-Bimanual</strong></td>
|
| 380 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">76.0884</strong></td>
|
| 381 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">14.4138</strong></td>
|
| 382 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.6839</strong></td>
|
| 383 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.3454</strong></td>
|
| 384 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0118</strong></td>
|
| 385 |
+
</tr>
|
| 386 |
+
</tbody>
|
| 387 |
+
</table>
|
| 388 |
+
|
| 389 |
+
#### AgiBot-GEAR
|
| 390 |
+
|
| 391 |
+
We evaluate **Cosmos3-Nano-Sim-Bimanual** on observation prediction conditioned on AgiBot-GEAR action sequences, an initial image, and a caption, comparing against the bidirectional DreamDojo 2B and 14B teacher models. We use the same video quality, reference-frame agreement, and drift metrics as the AgiBot-World evaluation.
|
| 392 |
+
|
| 393 |
+
<table>
|
| 394 |
+
<thead>
|
| 395 |
+
<tr>
|
| 396 |
+
<th scope="col" style="text-align: left;">Model</th>
|
| 397 |
+
<th scope="col" style="text-align: right;">Video Quality ↑</th>
|
| 398 |
+
<th scope="col" style="text-align: right;">PSNR (dB) ↑</th>
|
| 399 |
+
<th scope="col" style="text-align: right;">SSIM ↑</th>
|
| 400 |
+
<th scope="col" style="text-align: right;">LPIPS ↓</th>
|
| 401 |
+
<th scope="col" style="text-align: right;">Perceptual Drift ↓</th>
|
| 402 |
+
</tr>
|
| 403 |
+
</thead>
|
| 404 |
+
<tbody>
|
| 405 |
+
<tr>
|
| 406 |
+
<td style="text-align: left;">DreamDojo 2B</td>
|
| 407 |
+
<td style="text-align: right;">72.4586</td>
|
| 408 |
+
<td style="text-align: right;">11.5709</td>
|
| 409 |
+
<td style="text-align: right;">0.5437</td>
|
| 410 |
+
<td style="text-align: right;">0.4638</td>
|
| 411 |
+
<td style="text-align: right;">0.0232</td>
|
| 412 |
+
</tr>
|
| 413 |
+
<tr>
|
| 414 |
+
<td style="text-align: left;">DreamDojo 14B</td>
|
| 415 |
+
<td style="text-align: right;">73.0033</td>
|
| 416 |
+
<td style="text-align: right;">12.1113</td>
|
| 417 |
+
<td style="text-align: right;">0.5818</td>
|
| 418 |
+
<td style="text-align: right;">0.4326</td>
|
| 419 |
+
<td style="text-align: right;">0.0218</td>
|
| 420 |
+
</tr>
|
| 421 |
+
<tr>
|
| 422 |
+
<td style="text-align: left; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">Cosmos3-Nano-Sim-Bimanual</strong></td>
|
| 423 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">74.4912</strong></td>
|
| 424 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">14.0158</strong></td>
|
| 425 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.6172</strong></td>
|
| 426 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.4057</strong></td>
|
| 427 |
+
<td style="text-align: right; background-color: #e8f4df; color: #18202b;"><strong style="color: inherit;">0.0188</strong></td>
|
| 428 |
+
</tr>
|
| 429 |
+
</tbody>
|
| 430 |
+
</table>
|
| 431 |
+
|
| 432 |
+
## Inference
|
| 433 |
+
|
| 434 |
+
**Acceleration Engine:** [PyTorch](https://pytorch.org/), [vLLM-Omni](https://github.com/vllm-project/vllm-omni)
|
| 435 |
+
|
| 436 |
+
**Test Hardware:** NVIDIA Blackwell GB200, NVIDIA Hopper H100, H200, H20, RTX PRO 6000
|
| 437 |
+
|
| 438 |
+
### Inference Performance
|
| 439 |
+
|
| 440 |
+
Numbers are based on vLLM-Omni, using a single GPU. Empty table cells indicate that measurements are not reported.
|
| 441 |
+
|
| 442 |
+
| | 480p inference throughput (frames/s) |
|
| 443 |
+
| --- | --- |
|
| 444 |
+
| GB200 | 54.1 |
|
| 445 |
+
| H100 | 32.6 |
|
| 446 |
+
| H200 | 32.4 |
|
| 447 |
+
| H20 | 10.7 |
|
| 448 |
+
| RTX Pro 6000 | 20.6 |
|
| 449 |
+
|
| 450 |
+
## Usage Example
|
| 451 |
+
|
| 452 |
+
### Quick Start
|
| 453 |
+
|
| 454 |
+
This example serves the model with
|
| 455 |
+
[vLLM-Omni](https://github.com/vllm-project/vllm-omni) and generates the mixed-action AgiBot example: 901 frames at 30 FPS from one initial
|
| 456 |
+
image and 900 normalized 59D actions combining robot actions and camera motion.
|
| 457 |
+
The checkpoint uses a fixed four-step sampler.
|
| 458 |
+
|
| 459 |
+
#### 1. Start the server
|
| 460 |
+
|
| 461 |
+
On a Linux host with an NVIDIA GPU (tested on one GB200), [Docker](https://docs.docker.com/engine/install/), and the [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html):
|
| 462 |
+
|
| 463 |
+
```bash
|
| 464 |
+
docker run --runtime nvidia --gpus all --ipc=host \
|
| 465 |
+
-v ~/.cache/huggingface:/root/.cache/huggingface \
|
| 466 |
+
--env "HF_TOKEN=$HF_TOKEN" \
|
| 467 |
+
--env VLLM_OMNI_VIDEO_SYNC_TIMEOUT=1800 \
|
| 468 |
+
-p 127.0.0.1:8000:8000 \
|
| 469 |
+
vllm/vllm-omni:cosmos3 \
|
| 470 |
+
vllm serve nvidia/Cosmos3-Nano-Sim-Bimanual \
|
| 471 |
+
--omni --model-class-name Cosmos3NanoSimBimanualPipeline \
|
| 472 |
+
--deploy-config /app/vllm-omni/vllm_omni/deploy/cosmos3_nano_sim_bimanual_windowed.yaml \
|
| 473 |
+
--no-guardrails \
|
| 474 |
+
--host 0.0.0.0 --port 8000 --init-timeout 1800
|
| 475 |
+
```
|
| 476 |
+
|
| 477 |
+
The container publishes the server on `127.0.0.1:8000` of the host.
|
| 478 |
+
`--deploy-config` selects the Bimanual deployment: one request at a time, eager
|
| 479 |
+
execution, and a 64-latent-frame rolling key/value cache with eight attention
|
| 480 |
+
sinks. `--no-guardrails` disables the Cosmos3 text and video safety checks. The
|
| 481 |
+
first start downloads the checkpoint. For optimized server settings and
|
| 482 |
+
additional examples, see the
|
| 483 |
+
[Bimanual vLLM-Omni notebook](https://github.com/NVIDIA/cosmos-framework/blob/main/.agents/skills/cookbooks/cosmos3/simulation/bimanual/run_bimanual_with_vllm_omni.ipynb).
|
| 484 |
+
|
| 485 |
+
#### 2. Generate the mixed-action video
|
| 486 |
+
|
| 487 |
+
Install the client dependencies in your Python environment:
|
| 488 |
+
|
| 489 |
+
```bash
|
| 490 |
+
pip install requests huggingface_hub
|
| 491 |
+
```
|
| 492 |
+
|
| 493 |
+
Run the client on the same machine as the server. The example metadata supplies
|
| 494 |
+
the prompt, frame count, frame rate, action domain, resolution, and seed 2026:
|
| 495 |
+
|
| 496 |
+
```python
|
| 497 |
+
import json
|
| 498 |
+
from pathlib import Path
|
| 499 |
+
from uuid import uuid4
|
| 500 |
+
|
| 501 |
+
import requests
|
| 502 |
+
from huggingface_hub import snapshot_download
|
| 503 |
+
|
| 504 |
+
MODEL = "nvidia/Cosmos3-Nano-Sim-Bimanual"
|
| 505 |
+
repo_dir = Path(snapshot_download(MODEL, allow_patterns=[
|
| 506 |
+
"examples/inference/sim_bimanual_mixed_action_hf.jsonl",
|
| 507 |
+
"assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png",
|
| 508 |
+
"assets/sim_bimanual/agibot/actions/agibot_mixed_028_normalized_mixed_action.json",
|
| 509 |
+
"cosmos3_nano_sim_bimanual_action_sources/action_schema.json",
|
| 510 |
+
]))
|
| 511 |
+
manifest_path = repo_dir / "examples/inference/sim_bimanual_mixed_action_hf.jsonl"
|
| 512 |
+
image_path = repo_dir / "assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png"
|
| 513 |
+
action_path = repo_dir / "assets/sim_bimanual/agibot/actions/agibot_mixed_028_normalized_mixed_action.json"
|
| 514 |
+
schema_path = repo_dir / "cosmos3_nano_sim_bimanual_action_sources/action_schema.json"
|
| 515 |
+
|
| 516 |
+
examples = [json.loads(line) for line in manifest_path.read_text().splitlines()]
|
| 517 |
+
example = next(item for item in examples if item["action_path"].endswith("/" + action_path.name))
|
| 518 |
+
action = json.loads(action_path.read_text())
|
| 519 |
+
schema = json.loads(schema_path.read_text())
|
| 520 |
+
domain_ids = {
|
| 521 |
+
name: values["domain_id"]
|
| 522 |
+
for name, values in schema["embodiments"].items()
|
| 523 |
+
}
|
| 524 |
+
|
| 525 |
+
extra_params = {
|
| 526 |
+
"session_id": uuid4().hex,
|
| 527 |
+
"reset": True,
|
| 528 |
+
"close_session": True,
|
| 529 |
+
"action_mode": "forward_dynamics",
|
| 530 |
+
"action_space": action["action_space"],
|
| 531 |
+
"action": action["action"],
|
| 532 |
+
"domain_name": example["domain_name"],
|
| 533 |
+
"domain_id": [domain_ids[name] for name in action["domain_names"]],
|
| 534 |
+
"resolution": int(example["resolution"]),
|
| 535 |
+
}
|
| 536 |
+
|
| 537 |
+
with image_path.open("rb") as image_file:
|
| 538 |
+
response = requests.post(
|
| 539 |
+
"http://127.0.0.1:8000/v1/videos/sync",
|
| 540 |
+
data={
|
| 541 |
+
"model": MODEL,
|
| 542 |
+
"prompt": example["prompt"],
|
| 543 |
+
"num_frames": str(example["num_frames"]),
|
| 544 |
+
"fps": str(example["fps"]),
|
| 545 |
+
"seed": str(example["seed"]),
|
| 546 |
+
"extra_params": json.dumps(extra_params),
|
| 547 |
+
},
|
| 548 |
+
files={"input_reference": (image_path.name, image_file, "image/png")},
|
| 549 |
+
headers={"Accept": "video/mp4"},
|
| 550 |
+
timeout=1860,
|
| 551 |
+
)
|
| 552 |
+
response.raise_for_status()
|
| 553 |
+
Path("cosmos3-nano-sim-bimanual-mixed-028.mp4").write_bytes(response.content)
|
| 554 |
+
```
|
| 555 |
+
|
| 556 |
+
Generated output:
|
| 557 |
+
|
| 558 |
+
<video src="https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/output/cosmos3-nano-sim-bimanual-quickstart.mp4" controls autoplay loop muted playsinline width="100%"></video>
|
| 559 |
+
|
| 560 |
+
The initial image is the only visual observation sent to the server. The action
|
| 561 |
+
values are already normalized for this checkpoint.
|
| 562 |
+
|
| 563 |
+
## Ethical Considerations
|
| 564 |
+
|
| 565 |
+
NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. Developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
|
| 566 |
+
|
| 567 |
+
Please make sure you have proper rights and permissions for all input image and video content; if image or video includes people, personal health information, or intellectual property, the image or video generated will not blur or maintain proportions of image subjects included.
|
| 568 |
+
|
| 569 |
+
Users are responsible for model inputs and outputs. Users are responsible for ensuring safe integration of this model, including implementing guardrails as well as other safety mechanisms, prior to deployment.
|
| 570 |
+
|
| 571 |
+
For more detailed information on ethical considerations for this model, please see the Model Card++ [Explainability](explainability-Cosmos3-Nano-Sim-Bimanual.md), [Bias](bias.md), [Safety & Security](safety.md), and [Privacy](privacy.md) subcards. Please report model quality, risk, security vulnerabilities or NVIDIA AI Concerns [here](https://www.nvidia.com/en-us/support/submit-security-vulnerability/).
|
assets/output/cosmos3-nano-sim-bimanual-quickstart.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c04c1c883d135e3a3d6d76ccff406aae1e36bff7b8ee637dc5ec274c23cb217f
|
| 3 |
+
size 9417890
|
assets/sim_bimanual/agibot/actions/agibot_001_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/agibot/actions/agibot_034_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/agibot/actions/agibot_mixed_007_normalized_mixed_action.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/agibot/actions/agibot_mixed_028_normalized_mixed_action.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/agibot/actions/agibot_original_long60_019_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/agibot/images/agibot_001_initial.png
ADDED
|
Git LFS Details
|
assets/sim_bimanual/agibot/images/agibot_034_initial.png
ADDED
|
Git LFS Details
|
assets/sim_bimanual/agibot/images/agibot_mixed_007_initial.png
ADDED
|
Git LFS Details
|
assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png
ADDED
|
Git LFS Details
|
assets/sim_bimanual/agibot/images/agibot_original_long60_019_initial.png
ADDED
|
Git LFS Details
|
assets/sim_bimanual/yam/actions/yam_019_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/yam/actions/yam_023_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/sim_bimanual/yam/images/yam_019_initial.png
ADDED
|
Git LFS Details
|
assets/sim_bimanual/yam/images/yam_023_initial.png
ADDED
|
Git LFS Details
|
assets/training/agibot/actions/agibot_001_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/training/agibot/actions/agibot_006_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/training/agibot/actions/agibot_034_normalized_unifiedv1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
assets/training/agibot/gt/agibot_001_gt.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9b520df89e6f881bf3a5a57116032d62cf11dbec0d0ae3a86d869f26b5a022aa
|
| 3 |
+
size 2544736
|
assets/training/agibot/gt/agibot_006_gt.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f905b8e4963d7a705b16d3b0c5e5adfac40fa4ce37c7b34f960df4caf9be5089
|
| 3 |
+
size 1847124
|
assets/training/agibot/gt/agibot_034_gt.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:047b191903bfbfe8f79c6dbeeb518e0497ee772cad8010284147dac8c6dc7bad
|
| 3 |
+
size 2301040
|
assets/training/agibot/images/agibot_001_initial.png
ADDED
|
Git LFS Details
|
assets/training/agibot/images/agibot_006_initial.png
ADDED
|
Git LFS Details
|
assets/training/agibot/images/agibot_034_initial.png
ADDED
|
Git LFS Details
|
bias.md
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
## Bias
|
| 2 |
+
|
| 3 |
+
| Field | Response |
|
| 4 |
+
| :---- | :---- |
|
| 5 |
+
| Participation considerations from adversely impacted groups [protected classes](https://www.senate.ca.gov/content/protected-classes) in model design and testing | None. |
|
| 6 |
+
| Measures taken to mitigate against unwanted bias | Training, evaluation, and testing data are curated before release to filter restricted content, including content relating to protected classes. Downstream evaluation should assess coverage and model behavior across the environments and operating conditions relevant to the intended application. |
|
| 7 |
+
| Which characteristic (feature) show(s) the greatest difference in performance?: | Performance may vary with scene complexity, rollout length, and the input conditions used for generation. Long-horizon temporal consistency and fine-grained physical interactions are known challenges described for the Cosmos3-Nano base model and are relevant considerations for these simulation models. |
|
| 8 |
+
| Which feature(s) have the worst performance overall? | Challenging scenarios include long rollouts and complex physical interactions, where temporal consistency and fidelity to the supplied controls may degrade. Users should evaluate these aspects under their intended simulation conditions. |
|
| 9 |
+
| If using internal data, description of methods implemented in data acquisition or processing, if any, to address the prevalence of identifiable biases in the training, testing, and validation data: | Bias-specific methods applied during data processing include person-presence screening, demographic-taxonomy classification (age, gender, ethnicity), embedding-based diversity analysis, and dataset balancing across sources. Internal analysis of the Cosmos3-Nano base model's training data surfaced: non-person scenes are more prevalent than person-centric content; demographic-taxonomy outputs on person-present samples are most frequently "uncertain" across age, gender, and ethnicity dimensions; and source-type variation, with people-centric image and video datasets showing higher demographic signal than document-, object-, robotics-, or scene-focused datasets. *(Quantitative details are provided below.)* Downstream deployments should add bias audits, fairness evaluation, red-teaming, demographically balanced fine-tuning, or counterfactual augmentation as mitigations. |
|
| 10 |
+
| Tools used to assess statistical imbalances and highlight patterns that may introduce bias into AI models: | Dataset analytics pipelines, metadata distribution analysis, heuristic quality checks, embedding-based clustering, model-assisted filtering systems, and benchmark evaluation suites are used to assess statistical imbalances and identify patterns that may introduce bias into model behavior. |
|
| 11 |
+
| Tools used to assess statistical imbalances and highlight patterns that may introduce bias into AI models: | The following analysis covers training data used for the Cosmos3-Nano base model. These datasets, such as OpenImages-derived detection-to-NLP datasets, visual grounding and VQA datasets, document/image understanding datasets, video/action understanding datasets, and NVIDIA-created or curated visual datasets, do not collectively or exhaustively represent all demographic groups (and proportionally therein). For instance, automated person-presence screening did not identify a person in approximately 58% of visual samples analyzed across approximately 400 datasets, while person-present signals were identified in approximately 42% of analyzed samples. In the subset where person-present signals were identified, these datasets contain uneven representation splits across the measured visual taxonomies: age outputs were most frequently uncertain, followed by child and adult; gender outputs were most frequently uncertain, followed by male and female; and ethnicity outputs were most frequently uncertain, followed by Hispanic and White as the most frequent identified categories. Dataset-level results vary by source type, with people-centric image and video datasets containing higher person-present and demographic-taxonomy signals than document-, object-, robotics-, or scene-focused datasets. To mitigate these imbalances, we recommend considering evaluation techniques such as bias audits, task-specific fairness evaluation, and red-teaming, along with fine-tuning with demographically balanced datasets and counterfactual data augmentation to align with the desired model behavior. This evaluation used a baseline of 200 samples across all datasets, with larger subsets of up to 3,000 samples utilized for certain in-depth analyses, identified as optimal thresholds for maximizing embedder accuracy. |
|
chat_template.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0].role == 'system' %}\n {%- if messages[0].content is string %}\n {{- messages[0].content }}\n {%- else %}\n {%- for content in messages[0].content %}\n {%- if 'text' in content %}\n {{- content.text }}\n {%- endif %}\n {%- endfor %}\n {%- endif %}\n {{- '\\n\\n' }}\n {%- endif %}\n {{- \"# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n {%- if messages[0].role == 'system' %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0].content is string %}\n {{- messages[0].content }}\n {%- else %}\n {%- for content in messages[0].content %}\n {%- if 'text' in content %}\n {{- content.text }}\n {%- endif %}\n {%- endfor %}\n {%- endif %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- set image_count = namespace(value=0) %}\n{%- set video_count = namespace(value=0) %}\n{%- for message in messages %}\n {%- if message.role == \"user\" %}\n {{- '<|im_start|>' + message.role + '\\n' }}\n {%- if message.content is string %}\n {{- message.content }}\n {%- else %}\n {%- for content in message.content %}\n {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}\n {%- set image_count.value = image_count.value + 1 %}\n {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}\n <|vision_start|><|image_pad|><|vision_end|>\n {%- elif content.type == 'video' or 'video' in content %}\n {%- set video_count.value = video_count.value + 1 %}\n {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}\n <|vision_start|><|video_pad|><|vision_end|>\n {%- elif 'text' in content %}\n {{- content.text }}\n {%- endif %}\n {%- endfor %}\n {%- endif %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role + '\\n' }}\n {%- if message.content is string %}\n {{- message.content }}\n {%- else %}\n {%- for content_item in message.content %}\n {%- if 'text' in content_item %}\n {{- content_item.text }}\n {%- endif %}\n {%- endfor %}\n {%- endif %}\n {%- if message.tool_calls %}\n {%- for tool_call in message.tool_calls %}\n {%- if (loop.first and message.content) or (not loop.first) %}\n {{- '\\n' }}\n {%- endif %}\n {%- if tool_call.function %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {%- if tool_call.arguments is string %}\n {{- tool_call.arguments }}\n {%- else %}\n {{- tool_call.arguments | tojson }}\n {%- endif %}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {%- endif %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if loop.first or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {%- if message.content is string %}\n {{- message.content }}\n {%- else %}\n {%- for content in message.content %}\n {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}\n {%- set image_count.value = image_count.value + 1 %}\n {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}\n <|vision_start|><|image_pad|><|vision_end|>\n {%- elif content.type == 'video' or 'video' in content %}\n {%- set video_count.value = video_count.value + 1 %}\n {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}\n <|vision_start|><|video_pad|><|vision_end|>\n {%- elif 'text' in content %}\n {{- content.text }}\n {%- endif %}\n {%- endfor %}\n {%- endif %}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n"
|
| 3 |
+
}
|
checkpoint.json
ADDED
|
@@ -0,0 +1,249 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"checkpoint_type": "hf",
|
| 3 |
+
"cosmos3_nano_sim_bimanual": {
|
| 4 |
+
"attention_mode": "three_way",
|
| 5 |
+
"checkpoint_hash": "cc3075bfc7ffbf1ce361d0da01ee1348cc11d4b94aab980e648435306a6be53d",
|
| 6 |
+
"checkpoint_id": "simulation_sfdmd_bimanual_prod_64N_chunk2_v002_step4@iter_000001380",
|
| 7 |
+
"checkpoint_iteration": 1380,
|
| 8 |
+
"chunk_partition": [
|
| 9 |
+
1,
|
| 10 |
+
2
|
| 11 |
+
],
|
| 12 |
+
"chunk_size": 2,
|
| 13 |
+
"conditioning": {
|
| 14 |
+
"action_tokens_per_frame": 4,
|
| 15 |
+
"contract_sha256": "bb69c3678ea0a17e0685933606b44d5b9fe946d1f657fa945d7b4dce321723df",
|
| 16 |
+
"default_embodiment": "agibotworld",
|
| 17 |
+
"embodiments": {
|
| 18 |
+
"abc_yam": {
|
| 19 |
+
"domain_id": 16,
|
| 20 |
+
"input_action_dim": 59
|
| 21 |
+
},
|
| 22 |
+
"agibot_gear_gripper": {
|
| 23 |
+
"domain_id": 15,
|
| 24 |
+
"input_action_dim": 59
|
| 25 |
+
},
|
| 26 |
+
"agibot_gear_gripper_ext": {
|
| 27 |
+
"domain_id": 15,
|
| 28 |
+
"input_action_dim": 59
|
| 29 |
+
},
|
| 30 |
+
"agibotworld": {
|
| 31 |
+
"domain_id": 15,
|
| 32 |
+
"input_action_dim": 59
|
| 33 |
+
},
|
| 34 |
+
"behavior1k_lerobot": {
|
| 35 |
+
"domain_id": 22,
|
| 36 |
+
"input_action_dim": 59
|
| 37 |
+
},
|
| 38 |
+
"camera_pose": {
|
| 39 |
+
"domain_id": 2,
|
| 40 |
+
"input_action_dim": 59
|
| 41 |
+
},
|
| 42 |
+
"hand_pose": {
|
| 43 |
+
"domain_id": 3,
|
| 44 |
+
"input_action_dim": 59
|
| 45 |
+
},
|
| 46 |
+
"molmoact2_yam": {
|
| 47 |
+
"domain_id": 16,
|
| 48 |
+
"input_action_dim": 59
|
| 49 |
+
},
|
| 50 |
+
"xdof_yam": {
|
| 51 |
+
"domain_id": 16,
|
| 52 |
+
"input_action_dim": 59
|
| 53 |
+
}
|
| 54 |
+
},
|
| 55 |
+
"input_contract": {
|
| 56 |
+
"action_dim": 59,
|
| 57 |
+
"action_space": "normalized_unified_v1",
|
| 58 |
+
"domain_routing": "scalar_or_per_action_row",
|
| 59 |
+
"model_mode": "forward_dynamics",
|
| 60 |
+
"runtime_normalization": false,
|
| 61 |
+
"validity": "masked_to_zero_after_normalization_by_source"
|
| 62 |
+
},
|
| 63 |
+
"layout": {
|
| 64 |
+
"fields": [
|
| 65 |
+
{
|
| 66 |
+
"name": "ego_pose",
|
| 67 |
+
"offset": 0,
|
| 68 |
+
"size": 9
|
| 69 |
+
},
|
| 70 |
+
{
|
| 71 |
+
"name": "right_wrist_pose",
|
| 72 |
+
"offset": 9,
|
| 73 |
+
"size": 9
|
| 74 |
+
},
|
| 75 |
+
{
|
| 76 |
+
"name": "right_fingertips",
|
| 77 |
+
"offset": 18,
|
| 78 |
+
"size": 15
|
| 79 |
+
},
|
| 80 |
+
{
|
| 81 |
+
"name": "right_gripper_width_m",
|
| 82 |
+
"offset": 33,
|
| 83 |
+
"size": 1
|
| 84 |
+
},
|
| 85 |
+
{
|
| 86 |
+
"name": "left_wrist_pose",
|
| 87 |
+
"offset": 34,
|
| 88 |
+
"size": 9
|
| 89 |
+
},
|
| 90 |
+
{
|
| 91 |
+
"name": "left_fingertips",
|
| 92 |
+
"offset": 43,
|
| 93 |
+
"size": 15
|
| 94 |
+
},
|
| 95 |
+
{
|
| 96 |
+
"name": "left_gripper_width_m",
|
| 97 |
+
"offset": 58,
|
| 98 |
+
"size": 1
|
| 99 |
+
}
|
| 100 |
+
],
|
| 101 |
+
"id": "unified_v1",
|
| 102 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 103 |
+
"rotation_representation": "rot6d_columns"
|
| 104 |
+
},
|
| 105 |
+
"mode": "action",
|
| 106 |
+
"model_action_dim": 64,
|
| 107 |
+
"normalizer": {
|
| 108 |
+
"method": "global_asinh_unified_v1",
|
| 109 |
+
"runtime_application": false,
|
| 110 |
+
"source": {
|
| 111 |
+
"artifact_path": "cosmos3_nano_sim_bimanual_action_sources/ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0.json",
|
| 112 |
+
"path": "projects/cosmos3/cosmos3/datasets/action/normalizers/global_asinh_v1_backward_anchored_gripper_metric.json",
|
| 113 |
+
"pose_convention_authority": "resolved_canonical_lance_training_config",
|
| 114 |
+
"repository_revision": "970970f3a3024563973d4114e7c60cfda62c3af0",
|
| 115 |
+
"sha256": "ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0"
|
| 116 |
+
}
|
| 117 |
+
},
|
| 118 |
+
"num_embodiment_domains": 32,
|
| 119 |
+
"padding": {
|
| 120 |
+
"stage": "after_normalization_and_validity_mask",
|
| 121 |
+
"value": 0.0
|
| 122 |
+
},
|
| 123 |
+
"schema_version": 5,
|
| 124 |
+
"training_config_excerpt": {
|
| 125 |
+
"datasets": [
|
| 126 |
+
{
|
| 127 |
+
"camera_mask": false,
|
| 128 |
+
"embodiment": "camera_pose",
|
| 129 |
+
"name": "camera_lance_480_chunk_anchored",
|
| 130 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 131 |
+
"rotation_format": "rot6d",
|
| 132 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.lance.lance_camera_dataset.LanceCameraDataset",
|
| 133 |
+
"translation_scale": 1.0
|
| 134 |
+
},
|
| 135 |
+
{
|
| 136 |
+
"camera_mask": true,
|
| 137 |
+
"embodiment": "agibotworld",
|
| 138 |
+
"name": "agibotworld_beta",
|
| 139 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 140 |
+
"rotation_format": "rot6d",
|
| 141 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 142 |
+
"translation_scale": 1.0
|
| 143 |
+
},
|
| 144 |
+
{
|
| 145 |
+
"camera_mask": true,
|
| 146 |
+
"embodiment": "agibot_gear_gripper",
|
| 147 |
+
"name": "agibotgeargripper_offshelf",
|
| 148 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 149 |
+
"rotation_format": "rot6d",
|
| 150 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 151 |
+
"translation_scale": 1.0
|
| 152 |
+
},
|
| 153 |
+
{
|
| 154 |
+
"camera_mask": true,
|
| 155 |
+
"embodiment": "agibot_gear_gripper",
|
| 156 |
+
"name": "agibotgeargripper_custom",
|
| 157 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 158 |
+
"rotation_format": "rot6d",
|
| 159 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 160 |
+
"translation_scale": 1.0
|
| 161 |
+
},
|
| 162 |
+
{
|
| 163 |
+
"camera_mask": true,
|
| 164 |
+
"embodiment": "agibot_gear_gripper_ext",
|
| 165 |
+
"name": "agibotgeargripper_customext",
|
| 166 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 167 |
+
"rotation_format": "rot6d",
|
| 168 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 169 |
+
"translation_scale": 1.0
|
| 170 |
+
},
|
| 171 |
+
{
|
| 172 |
+
"camera_mask": true,
|
| 173 |
+
"embodiment": "molmoact2_yam",
|
| 174 |
+
"name": "molmoact2_yam",
|
| 175 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 176 |
+
"rotation_format": "rot6d",
|
| 177 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 178 |
+
"translation_scale": 1.0
|
| 179 |
+
},
|
| 180 |
+
{
|
| 181 |
+
"camera_mask": true,
|
| 182 |
+
"embodiment": "xdof_yam",
|
| 183 |
+
"name": "xdofyam",
|
| 184 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 185 |
+
"rotation_format": "rot6d",
|
| 186 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 187 |
+
"translation_scale": 1.0
|
| 188 |
+
},
|
| 189 |
+
{
|
| 190 |
+
"camera_mask": true,
|
| 191 |
+
"embodiment": "abc_yam",
|
| 192 |
+
"name": "abcyam",
|
| 193 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 194 |
+
"rotation_format": "rot6d",
|
| 195 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 196 |
+
"translation_scale": 1.0
|
| 197 |
+
},
|
| 198 |
+
{
|
| 199 |
+
"camera_mask": true,
|
| 200 |
+
"embodiment": "behavior1k_lerobot",
|
| 201 |
+
"name": "behavior1k",
|
| 202 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 203 |
+
"rotation_format": "rot6d",
|
| 204 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceBehavior1KDataset",
|
| 205 |
+
"translation_scale": 1.0
|
| 206 |
+
}
|
| 207 |
+
],
|
| 208 |
+
"experiment": "simulation_sfdmd_bimanual_prod_64N_chunk2_v002_step4",
|
| 209 |
+
"loader": {
|
| 210 |
+
"action_schema": "unified_v1",
|
| 211 |
+
"append_action_caption_semantics": false,
|
| 212 |
+
"append_duration_fps_timestamps": false,
|
| 213 |
+
"append_resolution_info": false,
|
| 214 |
+
"append_viewpoint_info": false,
|
| 215 |
+
"enable_mode_specific_prompt": false,
|
| 216 |
+
"format_prompt_as_json": false,
|
| 217 |
+
"global_asinh": true,
|
| 218 |
+
"metric_gripper": true
|
| 219 |
+
}
|
| 220 |
+
}
|
| 221 |
+
},
|
| 222 |
+
"fixed_step_sampler_config": {
|
| 223 |
+
"num_train_timesteps": 1000,
|
| 224 |
+
"sample_type": "sde",
|
| 225 |
+
"t_list": [
|
| 226 |
+
1.0,
|
| 227 |
+
0.9375,
|
| 228 |
+
0.8333333333333334,
|
| 229 |
+
0.625
|
| 230 |
+
]
|
| 231 |
+
},
|
| 232 |
+
"latent_patch_size": 2,
|
| 233 |
+
"mrope": {
|
| 234 |
+
"base_fps": 24,
|
| 235 |
+
"enable_fps_modulation": true,
|
| 236 |
+
"reset_spatial_ids": true,
|
| 237 |
+
"temporal_modality_margin": 15000
|
| 238 |
+
},
|
| 239 |
+
"schema_version": 1,
|
| 240 |
+
"sink_frames": 0,
|
| 241 |
+
"temporal_compression_factor": 4,
|
| 242 |
+
"text_cache_max_len": 512,
|
| 243 |
+
"vae_spatial_compression_factor": 16,
|
| 244 |
+
"video_temporal_causal": true,
|
| 245 |
+
"window_frames": null
|
| 246 |
+
},
|
| 247 |
+
"source_weights": "ema",
|
| 248 |
+
"student_only": true
|
| 249 |
+
}
|
config.json
ADDED
|
@@ -0,0 +1,424 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"allow_patterns_overrides": [
|
| 3 |
+
"*/*.safetensors"
|
| 4 |
+
],
|
| 5 |
+
"architectures": [
|
| 6 |
+
"Cosmos3ForConditionalGeneration"
|
| 7 |
+
],
|
| 8 |
+
"cosmos3_nano_sim_bimanual": {
|
| 9 |
+
"attention_mode": "three_way",
|
| 10 |
+
"base_fps": 24,
|
| 11 |
+
"checkpoint_hash": "cc3075bfc7ffbf1ce361d0da01ee1348cc11d4b94aab980e648435306a6be53d",
|
| 12 |
+
"checkpoint_id": "simulation_sfdmd_bimanual_prod_64N_chunk2_v002_step4@iter_000001380",
|
| 13 |
+
"checkpoint_iteration": 1380,
|
| 14 |
+
"chunk_size": 2,
|
| 15 |
+
"enable_fps_modulation": true,
|
| 16 |
+
"fixed_step_sampler_config": {
|
| 17 |
+
"num_train_timesteps": 1000,
|
| 18 |
+
"sample_type": "sde",
|
| 19 |
+
"t_list": [
|
| 20 |
+
1.0,
|
| 21 |
+
0.9375,
|
| 22 |
+
0.8333333333333334,
|
| 23 |
+
0.625
|
| 24 |
+
]
|
| 25 |
+
},
|
| 26 |
+
"latent_patch_size": 2,
|
| 27 |
+
"schema_version": 1,
|
| 28 |
+
"sink_frames": 0,
|
| 29 |
+
"temporal_compression_factor": 4,
|
| 30 |
+
"temporal_modality_margin": 15000,
|
| 31 |
+
"text_cache_max_len": 1024,
|
| 32 |
+
"unified_3d_mrope_reset_spatial_ids": true,
|
| 33 |
+
"vae_spatial_compression_factor": 16,
|
| 34 |
+
"video_temporal_causal": true,
|
| 35 |
+
"window_frames": null
|
| 36 |
+
},
|
| 37 |
+
"image_token_id": 151655,
|
| 38 |
+
"model": {
|
| 39 |
+
"_recursive_": false,
|
| 40 |
+
"_target": "omni_mot_causal_model",
|
| 41 |
+
"config": {
|
| 42 |
+
"_type": "omni_mot_causal_model_config",
|
| 43 |
+
"action_gen": true,
|
| 44 |
+
"action_global_asinh": false,
|
| 45 |
+
"action_io_projector_type": "domain_aware",
|
| 46 |
+
"action_metric_gripper": false,
|
| 47 |
+
"action_schema": "native",
|
| 48 |
+
"activation_checkpointing": {
|
| 49 |
+
"_type": "activation_checkpointing_config",
|
| 50 |
+
"determinism_check": "default",
|
| 51 |
+
"min_cgroup_memory_free_fraction": 0.1,
|
| 52 |
+
"mode": "selective",
|
| 53 |
+
"offload_to_cpu": false,
|
| 54 |
+
"preserve_rng_state": true,
|
| 55 |
+
"save_only_marked_ops": false,
|
| 56 |
+
"save_ops_regex": [
|
| 57 |
+
"fmha_forward",
|
| 58 |
+
"_flash_attn_varlen_forward",
|
| 59 |
+
"_flash_attn_forward",
|
| 60 |
+
"fmha_fwd"
|
| 61 |
+
]
|
| 62 |
+
},
|
| 63 |
+
"ar_distilled_sample_type": null,
|
| 64 |
+
"ar_static_und_cache_max_len": 512,
|
| 65 |
+
"attention_io_layout": "sequence_sharded",
|
| 66 |
+
"attention_sink_size": 0,
|
| 67 |
+
"causal_training_strategy": "none",
|
| 68 |
+
"clamp_empty_varlen_kv": true,
|
| 69 |
+
"compile": {
|
| 70 |
+
"_type": "compile_config",
|
| 71 |
+
"ar_post_saturation_mode": "default",
|
| 72 |
+
"compile_dynamic": true,
|
| 73 |
+
"compiled_region": "language",
|
| 74 |
+
"coordinate_descent_tuning": false,
|
| 75 |
+
"cuda_graph_scope": "block",
|
| 76 |
+
"enabled": false,
|
| 77 |
+
"mark_unbacked": false,
|
| 78 |
+
"max_autotune_pointwise": false,
|
| 79 |
+
"use_cuda_graphs": false
|
| 80 |
+
},
|
| 81 |
+
"correct_cp_gradients": false,
|
| 82 |
+
"diffusion_expert_config": {
|
| 83 |
+
"_type": "diffusion_expert_config",
|
| 84 |
+
"base_fps": 24,
|
| 85 |
+
"enable_action_modality_embedding": true,
|
| 86 |
+
"enable_fps_modulation": true,
|
| 87 |
+
"enable_media_modality_embedding": false,
|
| 88 |
+
"enable_sound_modality_embedding": true,
|
| 89 |
+
"enable_vision_modality_embeddings": false,
|
| 90 |
+
"lidar_patch_spatial_hw": null,
|
| 91 |
+
"load_weights_from_pretrained": false,
|
| 92 |
+
"max_vae_latent_side_after_patchify": 52,
|
| 93 |
+
"num_view_embeddings": 0,
|
| 94 |
+
"patch_spatial": 2,
|
| 95 |
+
"radar_patch_spatial_hw": null,
|
| 96 |
+
"sound_base_temporal_compression_factor": null,
|
| 97 |
+
"timestep_range": 1.0,
|
| 98 |
+
"unified_3d_mrope_reset_spatial_ids": true,
|
| 99 |
+
"unified_3d_mrope_temporal_modality_margin": 15000,
|
| 100 |
+
"vision_temporal_position_mode": "latent_index"
|
| 101 |
+
},
|
| 102 |
+
"ema": {
|
| 103 |
+
"_type": "ema_config",
|
| 104 |
+
"enabled": false,
|
| 105 |
+
"iteration_shift": 0,
|
| 106 |
+
"rate": 0.1
|
| 107 |
+
},
|
| 108 |
+
"enable_input_bias": true,
|
| 109 |
+
"enable_moba": false,
|
| 110 |
+
"exclude_reasoner_weights_from_checkpoint": false,
|
| 111 |
+
"fixed_step_sampler_config": {
|
| 112 |
+
"_type": "fixed_step_sampler_config",
|
| 113 |
+
"sample_type": "sde",
|
| 114 |
+
"t_list": [
|
| 115 |
+
1.0,
|
| 116 |
+
0.9375,
|
| 117 |
+
0.8333333333333334,
|
| 118 |
+
0.625
|
| 119 |
+
]
|
| 120 |
+
},
|
| 121 |
+
"flex_attention": {
|
| 122 |
+
"_type": "causal_flex_attention_config",
|
| 123 |
+
"backend": "auto",
|
| 124 |
+
"deduplicate_cross_view": false,
|
| 125 |
+
"geometry_position_encoding": "baseline",
|
| 126 |
+
"mask": {
|
| 127 |
+
"_type": "causal_flex_attention_mask_config",
|
| 128 |
+
"attention_scope": "all_views",
|
| 129 |
+
"control_attends_sensor": false,
|
| 130 |
+
"control_to_control_window": null,
|
| 131 |
+
"control_to_sensor_window": null,
|
| 132 |
+
"decomposed_temporal_window_includes_first_frame": false,
|
| 133 |
+
"decomposed_temporal_window_seconds": null,
|
| 134 |
+
"lidar_attends_captions": true,
|
| 135 |
+
"radar_attends_captions": true,
|
| 136 |
+
"sensor_to_control_window": null,
|
| 137 |
+
"sensor_to_sensor_window": null
|
| 138 |
+
},
|
| 139 |
+
"maskless_cross_view_band_keys": "all_views",
|
| 140 |
+
"maskless_cross_view_band_radius": 0,
|
| 141 |
+
"maskless_cross_view_include_frame_zero": false,
|
| 142 |
+
"prope_intrinsics": false,
|
| 143 |
+
"rigrope_canonical_frame": false,
|
| 144 |
+
"rigrope_channels": "all",
|
| 145 |
+
"rigrope_head_fraction": 1.0,
|
| 146 |
+
"rigrope_include_time": false,
|
| 147 |
+
"rigrope_max_freq_exponent": 3.0,
|
| 148 |
+
"rigrope_min_freq_exponent": 0.0,
|
| 149 |
+
"rigrope_missing_geometry": "error",
|
| 150 |
+
"rigrope_moment_normalization": "fixed",
|
| 151 |
+
"rigrope_moment_scale_floor_m": 0.05,
|
| 152 |
+
"rigrope_moment_scale_m": 25.0,
|
| 153 |
+
"rigrope_pose_world_frame": "scene",
|
| 154 |
+
"rigrope_time_scale_s": 1.0
|
| 155 |
+
},
|
| 156 |
+
"grad_clip": false,
|
| 157 |
+
"input_caption_key": "ai_caption",
|
| 158 |
+
"input_image_key": "images",
|
| 159 |
+
"input_video_key": "video",
|
| 160 |
+
"joint_attn_implementation": "three_way",
|
| 161 |
+
"kv_cache_dtype": null,
|
| 162 |
+
"kv_cache_inference_size": null,
|
| 163 |
+
"kv_cache_kernel_impl": "triton",
|
| 164 |
+
"latent_downsample_factor": 16,
|
| 165 |
+
"lbl": {
|
| 166 |
+
"_type": "lbl_config",
|
| 167 |
+
"coeff_gen": null,
|
| 168 |
+
"coeff_und": null,
|
| 169 |
+
"method": "local"
|
| 170 |
+
},
|
| 171 |
+
"lidar_fps": null,
|
| 172 |
+
"lidar_state_ch": null,
|
| 173 |
+
"lidar_tokenizer": null,
|
| 174 |
+
"load_vision_tokenizer": true,
|
| 175 |
+
"lora_alpha": 32,
|
| 176 |
+
"lora_enabled": false,
|
| 177 |
+
"lora_rank": 16,
|
| 178 |
+
"lora_target_modules": "q_proj_moe_gen,k_proj_moe_gen,v_proj_moe_gen,o_proj_moe_gen",
|
| 179 |
+
"max_action_dim": 64,
|
| 180 |
+
"max_num_tokens_after_packing": 500000,
|
| 181 |
+
"moba_bidirectional_steps": 1,
|
| 182 |
+
"moba_causal_steps": 1,
|
| 183 |
+
"multiview_action_conditioning": null,
|
| 184 |
+
"multiview_attention": {
|
| 185 |
+
"_type": "flex_attention_config",
|
| 186 |
+
"backend": "auto",
|
| 187 |
+
"deduplicate_cross_view": false,
|
| 188 |
+
"geometry_position_encoding": "baseline",
|
| 189 |
+
"mask": {
|
| 190 |
+
"_type": "flex_attention_mask_config",
|
| 191 |
+
"attention_scope": "all_views",
|
| 192 |
+
"control_attends_sensor": false,
|
| 193 |
+
"control_to_control_window": null,
|
| 194 |
+
"control_to_sensor_window": null,
|
| 195 |
+
"decomposed_temporal_window_includes_first_frame": false,
|
| 196 |
+
"decomposed_temporal_window_seconds": null,
|
| 197 |
+
"lidar_attends_captions": true,
|
| 198 |
+
"radar_attends_captions": true,
|
| 199 |
+
"sensor_to_control_window": null,
|
| 200 |
+
"sensor_to_sensor_window": null
|
| 201 |
+
},
|
| 202 |
+
"maskless_cross_view_band_keys": "all_views",
|
| 203 |
+
"maskless_cross_view_band_radius": 0,
|
| 204 |
+
"maskless_cross_view_include_frame_zero": false,
|
| 205 |
+
"prope_intrinsics": false,
|
| 206 |
+
"rigrope_canonical_frame": false,
|
| 207 |
+
"rigrope_channels": "all",
|
| 208 |
+
"rigrope_head_fraction": 1.0,
|
| 209 |
+
"rigrope_include_time": false,
|
| 210 |
+
"rigrope_max_freq_exponent": 3.0,
|
| 211 |
+
"rigrope_min_freq_exponent": 0.0,
|
| 212 |
+
"rigrope_missing_geometry": "error",
|
| 213 |
+
"rigrope_moment_normalization": "fixed",
|
| 214 |
+
"rigrope_moment_scale_floor_m": 0.05,
|
| 215 |
+
"rigrope_moment_scale_m": 25.0,
|
| 216 |
+
"rigrope_pose_world_frame": "scene",
|
| 217 |
+
"rigrope_time_scale_s": 1.0
|
| 218 |
+
},
|
| 219 |
+
"natten_parameter_list": null,
|
| 220 |
+
"net": null,
|
| 221 |
+
"num_embodiment_domains": 32,
|
| 222 |
+
"nvfp4_linears": null,
|
| 223 |
+
"parallelism": {
|
| 224 |
+
"_type": "parallelism_config",
|
| 225 |
+
"cfg_parallel_shard_degree": 1,
|
| 226 |
+
"context_parallel_shard_degree": 1,
|
| 227 |
+
"data_parallel_replicate_degree": 1,
|
| 228 |
+
"data_parallel_shard_degree": 1,
|
| 229 |
+
"enable_inference_mode": false,
|
| 230 |
+
"fsdp_cpu_offload": false,
|
| 231 |
+
"fsdp_master_dtype": "float32",
|
| 232 |
+
"fsdp_reduce_dtype": null,
|
| 233 |
+
"vae_load_balance_group_size": 1
|
| 234 |
+
},
|
| 235 |
+
"precision": "bfloat16",
|
| 236 |
+
"radar_fps": null,
|
| 237 |
+
"radar_state_ch": null,
|
| 238 |
+
"radar_tokenizer": null,
|
| 239 |
+
"rectified_flow_inference_config": {
|
| 240 |
+
"_type": "rectified_flow_inference_config",
|
| 241 |
+
"num_train_timesteps": 1000,
|
| 242 |
+
"scheduler_type": "unipc",
|
| 243 |
+
"shift": 1,
|
| 244 |
+
"use_dynamic_shifting": false
|
| 245 |
+
},
|
| 246 |
+
"rectified_flow_training_config": {
|
| 247 |
+
"_type": "rectified_flow_training_config",
|
| 248 |
+
"action_loss_weight": 10.0,
|
| 249 |
+
"exclude_fully_conditioned_items": null,
|
| 250 |
+
"image_loss_scale": null,
|
| 251 |
+
"independent_action_schedule": false,
|
| 252 |
+
"independent_sound_schedule": false,
|
| 253 |
+
"lidar_loss_scale": null,
|
| 254 |
+
"loss_reduction_mode": "local_item_mean",
|
| 255 |
+
"loss_scale": 10.0,
|
| 256 |
+
"normalize_loss_by_active": false,
|
| 257 |
+
"radar_loss_scale": null,
|
| 258 |
+
"shift": {
|
| 259 |
+
"256": 3,
|
| 260 |
+
"480": 5
|
| 261 |
+
},
|
| 262 |
+
"shift_action": null,
|
| 263 |
+
"shift_image": null,
|
| 264 |
+
"shift_lidar": null,
|
| 265 |
+
"shift_sound": null,
|
| 266 |
+
"sound_loss_scale": null,
|
| 267 |
+
"train_time_action_distribution": "logitnormal",
|
| 268 |
+
"train_time_image_distribution": "logitnormal",
|
| 269 |
+
"train_time_sound_distribution": "logitnormal",
|
| 270 |
+
"train_time_video_distribution": "uniform",
|
| 271 |
+
"train_time_weight": "uniform",
|
| 272 |
+
"use_discrete_rf": false,
|
| 273 |
+
"use_dynamic_shift": false
|
| 274 |
+
},
|
| 275 |
+
"release_cp_raw_batch_after_preprocessing": false,
|
| 276 |
+
"resolution": "480",
|
| 277 |
+
"rolling_kv_cache_chunks": null,
|
| 278 |
+
"rolling_kv_sink_chunks": 1,
|
| 279 |
+
"rolling_media_rope_scale": 1.0,
|
| 280 |
+
"rolling_media_sink_age_cap_seconds": null,
|
| 281 |
+
"rolling_media_text_extent": null,
|
| 282 |
+
"rolling_prompt_context_after": 0,
|
| 283 |
+
"rolling_prompt_context_before": 0,
|
| 284 |
+
"rolling_prompt_policy": null,
|
| 285 |
+
"rolling_static_caption_max_tokens": 576,
|
| 286 |
+
"rolling_text_position_mode": "fixed",
|
| 287 |
+
"rolling_text_sink_tokens": 4,
|
| 288 |
+
"sigma_diffusion_forcing": 0.02,
|
| 289 |
+
"sound_dim": null,
|
| 290 |
+
"sound_gen": false,
|
| 291 |
+
"sound_latent_fps": 25,
|
| 292 |
+
"sound_tokenizer": null,
|
| 293 |
+
"sr_latent_condition_noise": null,
|
| 294 |
+
"state_ch": 48,
|
| 295 |
+
"state_t": 300,
|
| 296 |
+
"teacher_forcing_detach_clean_kv": false,
|
| 297 |
+
"teacher_forcing_frames_per_chunk": 2,
|
| 298 |
+
"teacher_forcing_kv_implementation": "singleview_threeway_kv",
|
| 299 |
+
"teacher_forcing_replay_policy": {
|
| 300 |
+
"_type": "teacher_forcing_replay_policy_config",
|
| 301 |
+
"clean_pass_causality": "frame",
|
| 302 |
+
"control_visibility": "global",
|
| 303 |
+
"controls_read_strict_past_clean_rgb": false,
|
| 304 |
+
"decomposed_temporal_window_seconds": null,
|
| 305 |
+
"deduplicate_cross_view": false,
|
| 306 |
+
"multiview_attention_scope": "all_views"
|
| 307 |
+
},
|
| 308 |
+
"teacher_forcing_target_only_no_text_pass2": false,
|
| 309 |
+
"teacher_forcing_transfer_control_dropout_rate": 0.0,
|
| 310 |
+
"timestep_embedding_chunk_size": 128000,
|
| 311 |
+
"tokenizer": {
|
| 312 |
+
"_target": "wan2pt2_vae_interface",
|
| 313 |
+
"bucket_name": "bucket",
|
| 314 |
+
"causal": true,
|
| 315 |
+
"chunk_duration": 93,
|
| 316 |
+
"encode_bucket_multiple": null,
|
| 317 |
+
"encode_chunk_frames": {
|
| 318 |
+
"256": 68,
|
| 319 |
+
"480": 24
|
| 320 |
+
},
|
| 321 |
+
"encode_exact_durations": null,
|
| 322 |
+
"keep_decoder_cache": false,
|
| 323 |
+
"object_store_credential_path_pretrained": null,
|
| 324 |
+
"spatial_compression_factor": 16,
|
| 325 |
+
"temporal_compression_factor": 4,
|
| 326 |
+
"temporal_window": null,
|
| 327 |
+
"use_streaming_encode": false,
|
| 328 |
+
"vae_path": "pretrained/tokenizers/video/wan2pt2/Wan2.2_VAE.pth"
|
| 329 |
+
},
|
| 330 |
+
"video_temporal_causal": true,
|
| 331 |
+
"vision_gen": true,
|
| 332 |
+
"vlm_config": {
|
| 333 |
+
"_type": "vlm_config",
|
| 334 |
+
"layer_module": null,
|
| 335 |
+
"model_instance": {
|
| 336 |
+
"_target": "qwen3_vl_text_for_causal_lm",
|
| 337 |
+
"config": {
|
| 338 |
+
"_target": "create_vlm_config",
|
| 339 |
+
"base_config": {
|
| 340 |
+
"_target": "qwen3_vl_mot_config_from_json_file",
|
| 341 |
+
"json_file": "cosmos3://vfm/models/vlm/qwen3_vl/configs/Qwen3-VL-8B-Instruct.json"
|
| 342 |
+
},
|
| 343 |
+
"qk_norm_for_text": true
|
| 344 |
+
}
|
| 345 |
+
},
|
| 346 |
+
"model_name": "nvidia/Cosmos3-Nano-Reasoner",
|
| 347 |
+
"pretrained_weights": {
|
| 348 |
+
"_type": "pretrained_weights_config",
|
| 349 |
+
"backbone_path": "",
|
| 350 |
+
"checkpoint_format": null,
|
| 351 |
+
"credentials_path": "",
|
| 352 |
+
"enable_gcs_patch_in_boto3": false,
|
| 353 |
+
"enabled": false
|
| 354 |
+
},
|
| 355 |
+
"qk_norm": false,
|
| 356 |
+
"safetensors_path": "",
|
| 357 |
+
"tie_word_embeddings": false,
|
| 358 |
+
"tokenizer": {
|
| 359 |
+
"_target": "create_qwen2_tokenizer_with_download",
|
| 360 |
+
"config_variant": "hf",
|
| 361 |
+
"pretrained_model_name": "Qwen/Qwen3-VL-8B-Instruct"
|
| 362 |
+
},
|
| 363 |
+
"use_system_prompt": false
|
| 364 |
+
}
|
| 365 |
+
}
|
| 366 |
+
},
|
| 367 |
+
"model_type": "cosmos3_omni",
|
| 368 |
+
"per_view_captions": false,
|
| 369 |
+
"text_config": {
|
| 370 |
+
"attention_bias": false,
|
| 371 |
+
"attention_dropout": 0.0,
|
| 372 |
+
"bos_token_id": 151643,
|
| 373 |
+
"dtype": "bfloat16",
|
| 374 |
+
"eos_token_id": 151645,
|
| 375 |
+
"head_dim": 128,
|
| 376 |
+
"hidden_act": "silu",
|
| 377 |
+
"hidden_size": 4096,
|
| 378 |
+
"initializer_range": 0.02,
|
| 379 |
+
"intermediate_size": 12288,
|
| 380 |
+
"max_position_embeddings": 262144,
|
| 381 |
+
"model_type": "qwen3_vl_text",
|
| 382 |
+
"num_attention_heads": 32,
|
| 383 |
+
"num_hidden_layers": 36,
|
| 384 |
+
"num_key_value_heads": 8,
|
| 385 |
+
"rms_norm_eps": 1e-06,
|
| 386 |
+
"rope_scaling": {
|
| 387 |
+
"mrope_interleaved": true,
|
| 388 |
+
"mrope_section": [
|
| 389 |
+
24,
|
| 390 |
+
20,
|
| 391 |
+
20
|
| 392 |
+
],
|
| 393 |
+
"rope_type": "default"
|
| 394 |
+
},
|
| 395 |
+
"rope_theta": 5000000,
|
| 396 |
+
"use_cache": true,
|
| 397 |
+
"vocab_size": 151936
|
| 398 |
+
},
|
| 399 |
+
"tie_word_embeddings": false,
|
| 400 |
+
"transformers_version": "4.57.0.dev0",
|
| 401 |
+
"video_token_id": 151656,
|
| 402 |
+
"vision_config": {
|
| 403 |
+
"deepstack_visual_indexes": [
|
| 404 |
+
8,
|
| 405 |
+
16,
|
| 406 |
+
24
|
| 407 |
+
],
|
| 408 |
+
"depth": 27,
|
| 409 |
+
"hidden_act": "gelu_pytorch_tanh",
|
| 410 |
+
"hidden_size": 1152,
|
| 411 |
+
"in_channels": 3,
|
| 412 |
+
"initializer_range": 0.02,
|
| 413 |
+
"intermediate_size": 4304,
|
| 414 |
+
"model_type": "qwen3_vl",
|
| 415 |
+
"num_heads": 16,
|
| 416 |
+
"num_position_embeddings": 2304,
|
| 417 |
+
"out_hidden_size": 4096,
|
| 418 |
+
"patch_size": 16,
|
| 419 |
+
"spatial_merge_size": 2,
|
| 420 |
+
"temporal_patch_size": 2
|
| 421 |
+
},
|
| 422 |
+
"vision_end_token_id": 151653,
|
| 423 |
+
"vision_start_token_id": 151652
|
| 424 |
+
}
|
cosmos3_nano_sim_bimanual_action_sources/4d38b6321a92741c93e8cf1e96cba18979399ddb5c037708d3aaeffd666faa10.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"embodiment_type": "agibot",
|
| 4 |
+
"pose_convention": "backward_framewise",
|
| 5 |
+
"rotation_format": "rot6d",
|
| 6 |
+
"action_dim": 29,
|
| 7 |
+
"skip_rotation_dims": [3, 4, 5, 6, 7, 8, 12, 13, 14, 15, 16, 17, 22, 23, 24, 25, 26, 27],
|
| 8 |
+
"chunk_length": 16,
|
| 9 |
+
"sample_stride": 16,
|
| 10 |
+
"dataset_name": "agibot_shared_gripper_beta",
|
| 11 |
+
"dataset_class": "AgibotGEARGripperDataset+AgibotGEARGripperExtDataset+AgiBotWorldBetaDataset",
|
| 12 |
+
"dataset_root": "AgiBotWorld-GEAR + AgiBotWorld-Beta",
|
| 13 |
+
"split": "train",
|
| 14 |
+
"num_samples_stats": 200000,
|
| 15 |
+
"reservoir_size": 200000,
|
| 16 |
+
"max_samples": 200000,
|
| 17 |
+
"sampling_seed": 42,
|
| 18 |
+
"dataset_lengths": {
|
| 19 |
+
"agibot_gear_gripper": 10723406,
|
| 20 |
+
"agibot_gear_gripper_ext": 209875,
|
| 21 |
+
"agibotworld_beta": 16688991
|
| 22 |
+
},
|
| 23 |
+
"normalizer_note": "Shared AgiBot 29D FK-pose normalizer for GEAR gripper/ext and AgiBotWorld-Beta. Intended for action_normalization=asinh_rot to compress valid heavy-tailed base motion without clamping."
|
| 24 |
+
},
|
| 25 |
+
"global": {
|
| 26 |
+
"mean": [ 0.000007, -0.000028, 0.000017, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000054, 0.000055, -0.000004, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.461416, -0.000065, 0.000066, 0.000026, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.511525],
|
| 27 |
+
"std": [ 0.001146, 0.003775, 0.006236, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.005170, 0.005957, 0.005775, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.365715, 0.004577, 0.006320, 0.004809, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.368419],
|
| 28 |
+
"min": [-0.332076, -0.461419, -1.287811, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.181742, -1.008569, -0.964748, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000, -1.086141, -1.194795, -0.704462, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000],
|
| 29 |
+
"max": [ 0.332075, 0.461429, 1.287816, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.181742, 1.008569, 0.968047, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.077497, 1.194157, 0.696745, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000],
|
| 30 |
+
"q01": [-0.000474, -0.004285, -0.006603, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -0.011330, -0.012649, -0.014966, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000, -0.010683, -0.011185, -0.012754, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000],
|
| 31 |
+
"q99": [ 0.000472, 0.003180, 0.007315, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.011657, 0.012629, 0.013074, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.010122, 0.011385, 0.011604, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000]
|
| 32 |
+
},
|
| 33 |
+
"global_raw": {
|
| 34 |
+
"mean": [ 0.000007, -0.000028, 0.000017, 0.999997, 0.000012, -0.000011, -0.000013, 0.999992, 0.000013, 0.000054, 0.000055, -0.000004, 0.999755, 0.000162, -0.000206, -0.000163, 0.999770, -0.000068, 0.461416, -0.000065, 0.000066, 0.000026, 0.999821, -0.000156, 0.000218, 0.000156, 0.999840, -0.000065, 0.511525],
|
| 35 |
+
"std": [ 0.001146, 0.003775, 0.006236, 0.000050, 0.001322, 0.001990, 0.001321, 0.000091, 0.003722, 0.005170, 0.005957, 0.005775, 0.001184, 0.016934, 0.014178, 0.016919, 0.001112, 0.013136, 0.365715, 0.004577, 0.006320, 0.004809, 0.001140, 0.014271, 0.012406, 0.014263, 0.000987, 0.010745, 0.368419],
|
| 36 |
+
"min": [-0.332076, -0.461419, -1.287811, 0.992475, -0.047897, -0.115986, -0.048200, 0.986519, -0.163639, -1.181742, -1.008569, -0.964748, 0.834839, -0.535909, -0.508249, -0.417918, 0.835656, -0.360748, 0.000000, -1.086141, -1.194795, -0.704462, 0.885655, -0.422226, -0.392999, -0.404670, 0.847586, -0.267267, 0.000000],
|
| 37 |
+
"max": [ 0.332075, 0.461429, 1.287816, 1.000000, 0.048245, 0.111197, 0.047794, 1.000000, 0.130982, 1.181742, 1.008569, 0.968047, 1.000000, 0.413419, 0.506131, 0.545759, 1.000000, 0.405109, 1.000000, 1.077497, 1.194157, 0.696745, 1.000000, 0.405554, 0.460652, 0.428101, 1.000000, 0.423156, 1.000000],
|
| 38 |
+
"q01": [-0.000474, -0.004285, -0.006603, 0.999969, -0.000524, -0.000606, -0.000540, 0.999801, -0.000404, -0.011330, -0.012649, -0.014966, 0.995746, -0.055268, -0.044579, -0.057030, 0.995879, -0.041778, 0.000000, -0.010683, -0.011185, -0.012754, 0.996474, -0.048846, -0.038892, -0.044784, 0.996874, -0.034269, 0.000000],
|
| 39 |
+
"q99": [ 0.000472, 0.003180, 0.007315, 1.000000, 0.000532, 0.000603, 0.000514, 1.000000, 0.000454, 0.011657, 0.012629, 0.013074, 1.000000, 0.057221, 0.045662, 0.055355, 1.000000, 0.042508, 1.000000, 0.010122, 0.011385, 0.011604, 1.000000, 0.044671, 0.039433, 0.048886, 1.000000, 0.036261, 1.000000]
|
| 40 |
+
}
|
| 41 |
+
}
|
cosmos3_nano_sim_bimanual_action_sources/action_schema.json
ADDED
|
@@ -0,0 +1,208 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"action_tokens_per_frame": 4,
|
| 3 |
+
"contract_sha256": "bb69c3678ea0a17e0685933606b44d5b9fe946d1f657fa945d7b4dce321723df",
|
| 4 |
+
"default_embodiment": "agibotworld",
|
| 5 |
+
"embodiments": {
|
| 6 |
+
"abc_yam": {
|
| 7 |
+
"domain_id": 16,
|
| 8 |
+
"input_action_dim": 59
|
| 9 |
+
},
|
| 10 |
+
"agibot_gear_gripper": {
|
| 11 |
+
"domain_id": 15,
|
| 12 |
+
"input_action_dim": 59
|
| 13 |
+
},
|
| 14 |
+
"agibot_gear_gripper_ext": {
|
| 15 |
+
"domain_id": 15,
|
| 16 |
+
"input_action_dim": 59
|
| 17 |
+
},
|
| 18 |
+
"agibotworld": {
|
| 19 |
+
"domain_id": 15,
|
| 20 |
+
"input_action_dim": 59
|
| 21 |
+
},
|
| 22 |
+
"behavior1k_lerobot": {
|
| 23 |
+
"domain_id": 22,
|
| 24 |
+
"input_action_dim": 59
|
| 25 |
+
},
|
| 26 |
+
"camera_pose": {
|
| 27 |
+
"domain_id": 2,
|
| 28 |
+
"input_action_dim": 59
|
| 29 |
+
},
|
| 30 |
+
"hand_pose": {
|
| 31 |
+
"domain_id": 3,
|
| 32 |
+
"input_action_dim": 59
|
| 33 |
+
},
|
| 34 |
+
"molmoact2_yam": {
|
| 35 |
+
"domain_id": 16,
|
| 36 |
+
"input_action_dim": 59
|
| 37 |
+
},
|
| 38 |
+
"xdof_yam": {
|
| 39 |
+
"domain_id": 16,
|
| 40 |
+
"input_action_dim": 59
|
| 41 |
+
}
|
| 42 |
+
},
|
| 43 |
+
"input_contract": {
|
| 44 |
+
"action_dim": 59,
|
| 45 |
+
"action_space": "normalized_unified_v1",
|
| 46 |
+
"domain_routing": "scalar_or_per_action_row",
|
| 47 |
+
"model_mode": "forward_dynamics",
|
| 48 |
+
"runtime_normalization": false,
|
| 49 |
+
"validity": "masked_to_zero_after_normalization_by_source"
|
| 50 |
+
},
|
| 51 |
+
"layout": {
|
| 52 |
+
"fields": [
|
| 53 |
+
{
|
| 54 |
+
"name": "ego_pose",
|
| 55 |
+
"offset": 0,
|
| 56 |
+
"size": 9
|
| 57 |
+
},
|
| 58 |
+
{
|
| 59 |
+
"name": "right_wrist_pose",
|
| 60 |
+
"offset": 9,
|
| 61 |
+
"size": 9
|
| 62 |
+
},
|
| 63 |
+
{
|
| 64 |
+
"name": "right_fingertips",
|
| 65 |
+
"offset": 18,
|
| 66 |
+
"size": 15
|
| 67 |
+
},
|
| 68 |
+
{
|
| 69 |
+
"name": "right_gripper_width_m",
|
| 70 |
+
"offset": 33,
|
| 71 |
+
"size": 1
|
| 72 |
+
},
|
| 73 |
+
{
|
| 74 |
+
"name": "left_wrist_pose",
|
| 75 |
+
"offset": 34,
|
| 76 |
+
"size": 9
|
| 77 |
+
},
|
| 78 |
+
{
|
| 79 |
+
"name": "left_fingertips",
|
| 80 |
+
"offset": 43,
|
| 81 |
+
"size": 15
|
| 82 |
+
},
|
| 83 |
+
{
|
| 84 |
+
"name": "left_gripper_width_m",
|
| 85 |
+
"offset": 58,
|
| 86 |
+
"size": 1
|
| 87 |
+
}
|
| 88 |
+
],
|
| 89 |
+
"id": "unified_v1",
|
| 90 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 91 |
+
"rotation_representation": "rot6d_columns"
|
| 92 |
+
},
|
| 93 |
+
"model_action_dim": 64,
|
| 94 |
+
"normalizer": {
|
| 95 |
+
"method": "global_asinh_unified_v1",
|
| 96 |
+
"runtime_application": false,
|
| 97 |
+
"source": {
|
| 98 |
+
"artifact_path": "cosmos3_nano_sim_bimanual_action_sources/ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0.json",
|
| 99 |
+
"path": "projects/cosmos3/cosmos3/datasets/action/normalizers/global_asinh_v1_backward_anchored_gripper_metric.json",
|
| 100 |
+
"pose_convention_authority": "resolved_canonical_lance_training_config",
|
| 101 |
+
"repository_revision": "970970f3a3024563973d4114e7c60cfda62c3af0",
|
| 102 |
+
"sha256": "ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0"
|
| 103 |
+
}
|
| 104 |
+
},
|
| 105 |
+
"num_embodiment_domains": 32,
|
| 106 |
+
"padding": {
|
| 107 |
+
"stage": "after_normalization_and_validity_mask",
|
| 108 |
+
"value": 0.0
|
| 109 |
+
},
|
| 110 |
+
"schema_version": 5,
|
| 111 |
+
"training_config_excerpt": {
|
| 112 |
+
"datasets": [
|
| 113 |
+
{
|
| 114 |
+
"camera_mask": false,
|
| 115 |
+
"embodiment": "camera_pose",
|
| 116 |
+
"name": "camera_lance_480_chunk_anchored",
|
| 117 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 118 |
+
"rotation_format": "rot6d",
|
| 119 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.lance.lance_camera_dataset.LanceCameraDataset",
|
| 120 |
+
"translation_scale": 1.0
|
| 121 |
+
},
|
| 122 |
+
{
|
| 123 |
+
"camera_mask": true,
|
| 124 |
+
"embodiment": "agibotworld",
|
| 125 |
+
"name": "agibotworld_beta",
|
| 126 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 127 |
+
"rotation_format": "rot6d",
|
| 128 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 129 |
+
"translation_scale": 1.0
|
| 130 |
+
},
|
| 131 |
+
{
|
| 132 |
+
"camera_mask": true,
|
| 133 |
+
"embodiment": "agibot_gear_gripper",
|
| 134 |
+
"name": "agibotgeargripper_offshelf",
|
| 135 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 136 |
+
"rotation_format": "rot6d",
|
| 137 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 138 |
+
"translation_scale": 1.0
|
| 139 |
+
},
|
| 140 |
+
{
|
| 141 |
+
"camera_mask": true,
|
| 142 |
+
"embodiment": "agibot_gear_gripper",
|
| 143 |
+
"name": "agibotgeargripper_custom",
|
| 144 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 145 |
+
"rotation_format": "rot6d",
|
| 146 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 147 |
+
"translation_scale": 1.0
|
| 148 |
+
},
|
| 149 |
+
{
|
| 150 |
+
"camera_mask": true,
|
| 151 |
+
"embodiment": "agibot_gear_gripper_ext",
|
| 152 |
+
"name": "agibotgeargripper_customext",
|
| 153 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 154 |
+
"rotation_format": "rot6d",
|
| 155 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 156 |
+
"translation_scale": 1.0
|
| 157 |
+
},
|
| 158 |
+
{
|
| 159 |
+
"camera_mask": true,
|
| 160 |
+
"embodiment": "molmoact2_yam",
|
| 161 |
+
"name": "molmoact2_yam",
|
| 162 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 163 |
+
"rotation_format": "rot6d",
|
| 164 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 165 |
+
"translation_scale": 1.0
|
| 166 |
+
},
|
| 167 |
+
{
|
| 168 |
+
"camera_mask": true,
|
| 169 |
+
"embodiment": "xdof_yam",
|
| 170 |
+
"name": "xdofyam",
|
| 171 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 172 |
+
"rotation_format": "rot6d",
|
| 173 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 174 |
+
"translation_scale": 1.0
|
| 175 |
+
},
|
| 176 |
+
{
|
| 177 |
+
"camera_mask": true,
|
| 178 |
+
"embodiment": "abc_yam",
|
| 179 |
+
"name": "abcyam",
|
| 180 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 181 |
+
"rotation_format": "rot6d",
|
| 182 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 183 |
+
"translation_scale": 1.0
|
| 184 |
+
},
|
| 185 |
+
{
|
| 186 |
+
"camera_mask": true,
|
| 187 |
+
"embodiment": "behavior1k_lerobot",
|
| 188 |
+
"name": "behavior1k",
|
| 189 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 190 |
+
"rotation_format": "rot6d",
|
| 191 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceBehavior1KDataset",
|
| 192 |
+
"translation_scale": 1.0
|
| 193 |
+
}
|
| 194 |
+
],
|
| 195 |
+
"experiment": "simulation_sfdmd_bimanual_prod_64N_chunk2_v002_step4",
|
| 196 |
+
"loader": {
|
| 197 |
+
"action_schema": "unified_v1",
|
| 198 |
+
"append_action_caption_semantics": false,
|
| 199 |
+
"append_duration_fps_timestamps": false,
|
| 200 |
+
"append_resolution_info": false,
|
| 201 |
+
"append_viewpoint_info": false,
|
| 202 |
+
"enable_mode_specific_prompt": false,
|
| 203 |
+
"format_prompt_as_json": false,
|
| 204 |
+
"global_asinh": true,
|
| 205 |
+
"metric_gripper": true
|
| 206 |
+
}
|
| 207 |
+
}
|
| 208 |
+
}
|
cosmos3_nano_sim_bimanual_action_sources/ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0.json
ADDED
|
@@ -0,0 +1,162 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"name": "global_asinh_v1_backward_anchored_gripper_metric",
|
| 4 |
+
"method": "global_asinh",
|
| 5 |
+
"version": 3,
|
| 6 |
+
"action_schema": "unified_v1",
|
| 7 |
+
"dim": 59,
|
| 8 |
+
"allowed_pose_conventions": [
|
| 9 |
+
"backward_anchored"
|
| 10 |
+
],
|
| 11 |
+
"rotation_format": "rot6d",
|
| 12 |
+
"gripper_representation": "metric_width_m",
|
| 13 |
+
"metric_gripper_reference_width_m": 0.082,
|
| 14 |
+
"description": "q01 and q99 encode offset=(q01+q99)/2 and scale=(q99-q01)/2 for the canonical 59D action.",
|
| 15 |
+
"provenance": {
|
| 16 |
+
"ego_translation_override": "Set zero offset and the same 0.2822-meter scale for x, y, and z.",
|
| 17 |
+
"ego_translation_rationale": "AV motion is primarily along z while general camera motion is directionally unconstrained, so there is no robust cross-domain rule for axis-specific centers or scales.",
|
| 18 |
+
"ego_rotation": "hand_pose_backward_anchored_rot6d.json global_raw[3:9]",
|
| 19 |
+
"right_wrist_pose_and_openness": "yam_ee_backward_anchored_rot6d.json global_raw[10:20]",
|
| 20 |
+
"right_fingertips": "hand_pose_backward_anchored_rot6d.json global_raw[18:33]",
|
| 21 |
+
"left_wrist_pose_and_openness": "yam_ee_backward_anchored_rot6d.json global_raw[0:10]",
|
| 22 |
+
"left_fingertips": "hand_pose_backward_anchored_rot6d.json global_raw[42:57]",
|
| 23 |
+
"metric_gripper_override": "Set both gripper slots to YAM's zero-closed, 0.082-meter fully-open reference range."
|
| 24 |
+
},
|
| 25 |
+
"layout": {
|
| 26 |
+
"0:3": "ego translation xyz in meters; zero-centered with scale 0.2822 meters per axis",
|
| 27 |
+
"3:9": "ego rotation as rot6d from HandPose",
|
| 28 |
+
"9:12": "right wrist translation xyz in meters from YAM",
|
| 29 |
+
"12:18": "right wrist rotation as rot6d from YAM",
|
| 30 |
+
"18:33": "right fingertips: thumb, index, middle, ring, pinky xyz in the wrist frame from HandPose",
|
| 31 |
+
"33:34": "right gripper width in meters, where 0 is closed and 0.082 is the YAM fully-open reference width",
|
| 32 |
+
"34:37": "left wrist translation xyz in meters from YAM",
|
| 33 |
+
"37:43": "left wrist rotation as rot6d from YAM",
|
| 34 |
+
"43:58": "left fingertips: thumb, index, middle, ring, pinky xyz in the wrist frame from HandPose",
|
| 35 |
+
"58:59": "left gripper width in meters, where 0 is closed and 0.082 is the YAM fully-open reference width"
|
| 36 |
+
}
|
| 37 |
+
},
|
| 38 |
+
"global": {
|
| 39 |
+
"q01": [
|
| 40 |
+
-0.2822,
|
| 41 |
+
-0.2822,
|
| 42 |
+
-0.2822,
|
| 43 |
+
0.709793,
|
| 44 |
+
-0.528049,
|
| 45 |
+
-0.211989,
|
| 46 |
+
-0.526309,
|
| 47 |
+
0.717806,
|
| 48 |
+
-0.2239,
|
| 49 |
+
-0.121196,
|
| 50 |
+
-0.106861,
|
| 51 |
+
-0.161145,
|
| 52 |
+
0.604712,
|
| 53 |
+
-0.500095,
|
| 54 |
+
-0.529701,
|
| 55 |
+
-0.484055,
|
| 56 |
+
0.689496,
|
| 57 |
+
-0.384133,
|
| 58 |
+
-0.078661,
|
| 59 |
+
0.030485,
|
| 60 |
+
0.058749,
|
| 61 |
+
-0.037953,
|
| 62 |
+
0.011371,
|
| 63 |
+
0.063185,
|
| 64 |
+
-0.007476,
|
| 65 |
+
0.01086,
|
| 66 |
+
0.047803,
|
| 67 |
+
0.00622,
|
| 68 |
+
0.00845,
|
| 69 |
+
0.037389,
|
| 70 |
+
0.019229,
|
| 71 |
+
0.003273,
|
| 72 |
+
0.03427,
|
| 73 |
+
0,
|
| 74 |
+
-0.104546,
|
| 75 |
+
-0.095492,
|
| 76 |
+
-0.149185,
|
| 77 |
+
0.666787,
|
| 78 |
+
-0.419017,
|
| 79 |
+
-0.473956,
|
| 80 |
+
-0.444951,
|
| 81 |
+
0.744433,
|
| 82 |
+
-0.35486,
|
| 83 |
+
-0.0134,
|
| 84 |
+
0.024862,
|
| 85 |
+
0.053364,
|
| 86 |
+
-0.011772,
|
| 87 |
+
0.002479,
|
| 88 |
+
0.066106,
|
| 89 |
+
-0.037275,
|
| 90 |
+
0.000915,
|
| 91 |
+
0.053206,
|
| 92 |
+
-0.058168,
|
| 93 |
+
-0.002249,
|
| 94 |
+
0.043841,
|
| 95 |
+
-0.076916,
|
| 96 |
+
-0.005779,
|
| 97 |
+
0.039363,
|
| 98 |
+
0
|
| 99 |
+
],
|
| 100 |
+
"q99": [
|
| 101 |
+
0.2822,
|
| 102 |
+
0.2822,
|
| 103 |
+
0.2822,
|
| 104 |
+
1,
|
| 105 |
+
0.521286,
|
| 106 |
+
0.203338,
|
| 107 |
+
0.529919,
|
| 108 |
+
0.999999,
|
| 109 |
+
0.259182,
|
| 110 |
+
0.124457,
|
| 111 |
+
0.12985,
|
| 112 |
+
0.123286,
|
| 113 |
+
1,
|
| 114 |
+
0.485177,
|
| 115 |
+
0.515304,
|
| 116 |
+
0.504032,
|
| 117 |
+
1,
|
| 118 |
+
0.440837,
|
| 119 |
+
0.016207,
|
| 120 |
+
0.105381,
|
| 121 |
+
0.129947,
|
| 122 |
+
0.014021,
|
| 123 |
+
0.090293,
|
| 124 |
+
0.187483,
|
| 125 |
+
0.037809,
|
| 126 |
+
0.088395,
|
| 127 |
+
0.189717,
|
| 128 |
+
0.058805,
|
| 129 |
+
0.084619,
|
| 130 |
+
0.175622,
|
| 131 |
+
0.073382,
|
| 132 |
+
0.069707,
|
| 133 |
+
0.143841,
|
| 134 |
+
0.082,
|
| 135 |
+
0.103266,
|
| 136 |
+
0.11501,
|
| 137 |
+
0.114157,
|
| 138 |
+
1,
|
| 139 |
+
0.440519,
|
| 140 |
+
0.501329,
|
| 141 |
+
0.416844,
|
| 142 |
+
1,
|
| 143 |
+
0.408429,
|
| 144 |
+
0.086612,
|
| 145 |
+
0.10718,
|
| 146 |
+
0.12828,
|
| 147 |
+
0.038691,
|
| 148 |
+
0.089021,
|
| 149 |
+
0.190503,
|
| 150 |
+
0.008158,
|
| 151 |
+
0.087324,
|
| 152 |
+
0.194924,
|
| 153 |
+
-0.006426,
|
| 154 |
+
0.083202,
|
| 155 |
+
0.181171,
|
| 156 |
+
-0.020051,
|
| 157 |
+
0.068746,
|
| 158 |
+
0.149127,
|
| 159 |
+
0.082
|
| 160 |
+
]
|
| 161 |
+
}
|
| 162 |
+
}
|
cosmos3_nano_sim_bimanual_action_sources/cosmos3_nano_sim_bimanual_action_schema.schema.json
ADDED
|
@@ -0,0 +1,300 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"type": "object",
|
| 3 |
+
"additionalProperties": false,
|
| 4 |
+
"required": [
|
| 5 |
+
"schema_version",
|
| 6 |
+
"action_tokens_per_frame",
|
| 7 |
+
"model_action_dim",
|
| 8 |
+
"num_embodiment_domains",
|
| 9 |
+
"default_embodiment",
|
| 10 |
+
"embodiments",
|
| 11 |
+
"padding",
|
| 12 |
+
"input_contract",
|
| 13 |
+
"layout",
|
| 14 |
+
"normalizer",
|
| 15 |
+
"training_config_excerpt",
|
| 16 |
+
"contract_sha256"
|
| 17 |
+
],
|
| 18 |
+
"properties": {
|
| 19 |
+
"schema_version": {
|
| 20 |
+
"const": 5
|
| 21 |
+
},
|
| 22 |
+
"action_tokens_per_frame": {
|
| 23 |
+
"const": 4
|
| 24 |
+
},
|
| 25 |
+
"model_action_dim": {
|
| 26 |
+
"const": 64
|
| 27 |
+
},
|
| 28 |
+
"num_embodiment_domains": {
|
| 29 |
+
"const": 32
|
| 30 |
+
},
|
| 31 |
+
"default_embodiment": {
|
| 32 |
+
"enum": [
|
| 33 |
+
"agibotworld",
|
| 34 |
+
"agibot_gear_gripper",
|
| 35 |
+
"agibot_gear_gripper_ext",
|
| 36 |
+
"molmoact2_yam",
|
| 37 |
+
"xdof_yam",
|
| 38 |
+
"abc_yam",
|
| 39 |
+
"hand_pose",
|
| 40 |
+
"behavior1k_lerobot",
|
| 41 |
+
"camera_pose"
|
| 42 |
+
]
|
| 43 |
+
},
|
| 44 |
+
"embodiments": {
|
| 45 |
+
"type": "object",
|
| 46 |
+
"minProperties": 1,
|
| 47 |
+
"additionalProperties": false,
|
| 48 |
+
"properties": {
|
| 49 |
+
"agibotworld": {
|
| 50 |
+
"const": {
|
| 51 |
+
"domain_id": 15,
|
| 52 |
+
"input_action_dim": 59
|
| 53 |
+
}
|
| 54 |
+
},
|
| 55 |
+
"agibot_gear_gripper": {
|
| 56 |
+
"const": {
|
| 57 |
+
"domain_id": 15,
|
| 58 |
+
"input_action_dim": 59
|
| 59 |
+
}
|
| 60 |
+
},
|
| 61 |
+
"agibot_gear_gripper_ext": {
|
| 62 |
+
"const": {
|
| 63 |
+
"domain_id": 15,
|
| 64 |
+
"input_action_dim": 59
|
| 65 |
+
}
|
| 66 |
+
},
|
| 67 |
+
"molmoact2_yam": {
|
| 68 |
+
"const": {
|
| 69 |
+
"domain_id": 16,
|
| 70 |
+
"input_action_dim": 59
|
| 71 |
+
}
|
| 72 |
+
},
|
| 73 |
+
"xdof_yam": {
|
| 74 |
+
"const": {
|
| 75 |
+
"domain_id": 16,
|
| 76 |
+
"input_action_dim": 59
|
| 77 |
+
}
|
| 78 |
+
},
|
| 79 |
+
"abc_yam": {
|
| 80 |
+
"const": {
|
| 81 |
+
"domain_id": 16,
|
| 82 |
+
"input_action_dim": 59
|
| 83 |
+
}
|
| 84 |
+
},
|
| 85 |
+
"hand_pose": {
|
| 86 |
+
"const": {
|
| 87 |
+
"domain_id": 3,
|
| 88 |
+
"input_action_dim": 59
|
| 89 |
+
}
|
| 90 |
+
},
|
| 91 |
+
"behavior1k_lerobot": {
|
| 92 |
+
"const": {
|
| 93 |
+
"domain_id": 22,
|
| 94 |
+
"input_action_dim": 59
|
| 95 |
+
}
|
| 96 |
+
},
|
| 97 |
+
"camera_pose": {
|
| 98 |
+
"const": {
|
| 99 |
+
"domain_id": 2,
|
| 100 |
+
"input_action_dim": 59
|
| 101 |
+
}
|
| 102 |
+
}
|
| 103 |
+
}
|
| 104 |
+
},
|
| 105 |
+
"padding": {
|
| 106 |
+
"const": {
|
| 107 |
+
"stage": "after_normalization_and_validity_mask",
|
| 108 |
+
"value": 0.0
|
| 109 |
+
}
|
| 110 |
+
},
|
| 111 |
+
"input_contract": {
|
| 112 |
+
"const": {
|
| 113 |
+
"action_space": "normalized_unified_v1",
|
| 114 |
+
"action_dim": 59,
|
| 115 |
+
"domain_routing": "scalar_or_per_action_row",
|
| 116 |
+
"validity": "masked_to_zero_after_normalization_by_source",
|
| 117 |
+
"runtime_normalization": false,
|
| 118 |
+
"model_mode": "forward_dynamics"
|
| 119 |
+
}
|
| 120 |
+
},
|
| 121 |
+
"layout": {
|
| 122 |
+
"const": {
|
| 123 |
+
"id": "unified_v1",
|
| 124 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 125 |
+
"rotation_representation": "rot6d_columns",
|
| 126 |
+
"fields": [
|
| 127 |
+
{
|
| 128 |
+
"name": "ego_pose",
|
| 129 |
+
"offset": 0,
|
| 130 |
+
"size": 9
|
| 131 |
+
},
|
| 132 |
+
{
|
| 133 |
+
"name": "right_wrist_pose",
|
| 134 |
+
"offset": 9,
|
| 135 |
+
"size": 9
|
| 136 |
+
},
|
| 137 |
+
{
|
| 138 |
+
"name": "right_fingertips",
|
| 139 |
+
"offset": 18,
|
| 140 |
+
"size": 15
|
| 141 |
+
},
|
| 142 |
+
{
|
| 143 |
+
"name": "right_gripper_width_m",
|
| 144 |
+
"offset": 33,
|
| 145 |
+
"size": 1
|
| 146 |
+
},
|
| 147 |
+
{
|
| 148 |
+
"name": "left_wrist_pose",
|
| 149 |
+
"offset": 34,
|
| 150 |
+
"size": 9
|
| 151 |
+
},
|
| 152 |
+
{
|
| 153 |
+
"name": "left_fingertips",
|
| 154 |
+
"offset": 43,
|
| 155 |
+
"size": 15
|
| 156 |
+
},
|
| 157 |
+
{
|
| 158 |
+
"name": "left_gripper_width_m",
|
| 159 |
+
"offset": 58,
|
| 160 |
+
"size": 1
|
| 161 |
+
}
|
| 162 |
+
]
|
| 163 |
+
}
|
| 164 |
+
},
|
| 165 |
+
"normalizer": {
|
| 166 |
+
"type": "object",
|
| 167 |
+
"additionalProperties": false,
|
| 168 |
+
"required": [
|
| 169 |
+
"method",
|
| 170 |
+
"runtime_application",
|
| 171 |
+
"source"
|
| 172 |
+
],
|
| 173 |
+
"properties": {
|
| 174 |
+
"method": {
|
| 175 |
+
"const": "global_asinh_unified_v1"
|
| 176 |
+
},
|
| 177 |
+
"runtime_application": {
|
| 178 |
+
"const": false
|
| 179 |
+
},
|
| 180 |
+
"source": {
|
| 181 |
+
"type": "object",
|
| 182 |
+
"additionalProperties": false,
|
| 183 |
+
"required": [
|
| 184 |
+
"path",
|
| 185 |
+
"artifact_path",
|
| 186 |
+
"sha256",
|
| 187 |
+
"repository_revision",
|
| 188 |
+
"pose_convention_authority"
|
| 189 |
+
],
|
| 190 |
+
"properties": {
|
| 191 |
+
"path": {
|
| 192 |
+
"const": "projects/cosmos3/cosmos3/datasets/action/normalizers/global_asinh_v1_backward_anchored_gripper_metric.json"
|
| 193 |
+
},
|
| 194 |
+
"artifact_path": {
|
| 195 |
+
"type": "string",
|
| 196 |
+
"pattern": "^cosmos3_nano_sim_bimanual_action_sources/[0-9a-f]{64}\\.json$"
|
| 197 |
+
},
|
| 198 |
+
"sha256": {
|
| 199 |
+
"type": "string",
|
| 200 |
+
"pattern": "^[0-9a-f]{64}$"
|
| 201 |
+
},
|
| 202 |
+
"repository_revision": {
|
| 203 |
+
"type": "string",
|
| 204 |
+
"pattern": "^[0-9a-f]{40}$"
|
| 205 |
+
},
|
| 206 |
+
"pose_convention_authority": {
|
| 207 |
+
"const": "resolved_canonical_lance_training_config"
|
| 208 |
+
}
|
| 209 |
+
}
|
| 210 |
+
}
|
| 211 |
+
}
|
| 212 |
+
},
|
| 213 |
+
"training_config_excerpt": {
|
| 214 |
+
"type": "object",
|
| 215 |
+
"additionalProperties": false,
|
| 216 |
+
"required": [
|
| 217 |
+
"experiment",
|
| 218 |
+
"loader",
|
| 219 |
+
"datasets"
|
| 220 |
+
],
|
| 221 |
+
"properties": {
|
| 222 |
+
"experiment": {
|
| 223 |
+
"type": "string",
|
| 224 |
+
"minLength": 1
|
| 225 |
+
},
|
| 226 |
+
"loader": {
|
| 227 |
+
"const": {
|
| 228 |
+
"action_schema": "unified_v1",
|
| 229 |
+
"global_asinh": true,
|
| 230 |
+
"metric_gripper": true,
|
| 231 |
+
"format_prompt_as_json": false,
|
| 232 |
+
"append_viewpoint_info": false,
|
| 233 |
+
"append_action_caption_semantics": false,
|
| 234 |
+
"append_duration_fps_timestamps": false,
|
| 235 |
+
"append_resolution_info": false,
|
| 236 |
+
"enable_mode_specific_prompt": false
|
| 237 |
+
}
|
| 238 |
+
},
|
| 239 |
+
"datasets": {
|
| 240 |
+
"type": "array",
|
| 241 |
+
"minItems": 1,
|
| 242 |
+
"items": {
|
| 243 |
+
"type": "object",
|
| 244 |
+
"additionalProperties": false,
|
| 245 |
+
"required": [
|
| 246 |
+
"name",
|
| 247 |
+
"target",
|
| 248 |
+
"embodiment",
|
| 249 |
+
"pose_convention",
|
| 250 |
+
"rotation_format",
|
| 251 |
+
"translation_scale",
|
| 252 |
+
"camera_mask"
|
| 253 |
+
],
|
| 254 |
+
"properties": {
|
| 255 |
+
"name": {
|
| 256 |
+
"type": "string",
|
| 257 |
+
"minLength": 1
|
| 258 |
+
},
|
| 259 |
+
"target": {
|
| 260 |
+
"type": "string",
|
| 261 |
+
"minLength": 1
|
| 262 |
+
},
|
| 263 |
+
"embodiment": {
|
| 264 |
+
"enum": [
|
| 265 |
+
"agibotworld",
|
| 266 |
+
"agibot_gear_gripper",
|
| 267 |
+
"agibot_gear_gripper_ext",
|
| 268 |
+
"molmoact2_yam",
|
| 269 |
+
"xdof_yam",
|
| 270 |
+
"abc_yam",
|
| 271 |
+
"hand_pose",
|
| 272 |
+
"behavior1k_lerobot",
|
| 273 |
+
"camera_pose"
|
| 274 |
+
]
|
| 275 |
+
},
|
| 276 |
+
"pose_convention": {
|
| 277 |
+
"const": "backward_chunk_anchored_16f"
|
| 278 |
+
},
|
| 279 |
+
"rotation_format": {
|
| 280 |
+
"const": "rot6d"
|
| 281 |
+
},
|
| 282 |
+
"translation_scale": {
|
| 283 |
+
"const": 1.0
|
| 284 |
+
},
|
| 285 |
+
"camera_mask": {
|
| 286 |
+
"type": "boolean"
|
| 287 |
+
}
|
| 288 |
+
}
|
| 289 |
+
}
|
| 290 |
+
}
|
| 291 |
+
}
|
| 292 |
+
},
|
| 293 |
+
"contract_sha256": {
|
| 294 |
+
"type": "string",
|
| 295 |
+
"pattern": "^[0-9a-f]{64}$"
|
| 296 |
+
}
|
| 297 |
+
},
|
| 298 |
+
"$schema": "https://json-schema.org/draft/2020-12/schema",
|
| 299 |
+
"title": "Cosmos3 Sim Bimanual model-ready unified-v1 contract (schema 5)"
|
| 300 |
+
}
|
cosmos3_nano_sim_bimanual_action_sources/resolved_action_config.json
ADDED
|
@@ -0,0 +1,97 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"datasets": [
|
| 3 |
+
{
|
| 4 |
+
"camera_mask": false,
|
| 5 |
+
"embodiment": "camera_pose",
|
| 6 |
+
"name": "camera_lance_480_chunk_anchored",
|
| 7 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 8 |
+
"rotation_format": "rot6d",
|
| 9 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.lance.lance_camera_dataset.LanceCameraDataset",
|
| 10 |
+
"translation_scale": 1.0
|
| 11 |
+
},
|
| 12 |
+
{
|
| 13 |
+
"camera_mask": true,
|
| 14 |
+
"embodiment": "agibotworld",
|
| 15 |
+
"name": "agibotworld_beta",
|
| 16 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 17 |
+
"rotation_format": "rot6d",
|
| 18 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 19 |
+
"translation_scale": 1.0
|
| 20 |
+
},
|
| 21 |
+
{
|
| 22 |
+
"camera_mask": true,
|
| 23 |
+
"embodiment": "agibot_gear_gripper",
|
| 24 |
+
"name": "agibotgeargripper_offshelf",
|
| 25 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 26 |
+
"rotation_format": "rot6d",
|
| 27 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 28 |
+
"translation_scale": 1.0
|
| 29 |
+
},
|
| 30 |
+
{
|
| 31 |
+
"camera_mask": true,
|
| 32 |
+
"embodiment": "agibot_gear_gripper",
|
| 33 |
+
"name": "agibotgeargripper_custom",
|
| 34 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 35 |
+
"rotation_format": "rot6d",
|
| 36 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 37 |
+
"translation_scale": 1.0
|
| 38 |
+
},
|
| 39 |
+
{
|
| 40 |
+
"camera_mask": true,
|
| 41 |
+
"embodiment": "agibot_gear_gripper_ext",
|
| 42 |
+
"name": "agibotgeargripper_customext",
|
| 43 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 44 |
+
"rotation_format": "rot6d",
|
| 45 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceAgiBotDataset",
|
| 46 |
+
"translation_scale": 1.0
|
| 47 |
+
},
|
| 48 |
+
{
|
| 49 |
+
"camera_mask": true,
|
| 50 |
+
"embodiment": "molmoact2_yam",
|
| 51 |
+
"name": "molmoact2_yam",
|
| 52 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 53 |
+
"rotation_format": "rot6d",
|
| 54 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 55 |
+
"translation_scale": 1.0
|
| 56 |
+
},
|
| 57 |
+
{
|
| 58 |
+
"camera_mask": true,
|
| 59 |
+
"embodiment": "xdof_yam",
|
| 60 |
+
"name": "xdofyam",
|
| 61 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 62 |
+
"rotation_format": "rot6d",
|
| 63 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 64 |
+
"translation_scale": 1.0
|
| 65 |
+
},
|
| 66 |
+
{
|
| 67 |
+
"camera_mask": true,
|
| 68 |
+
"embodiment": "abc_yam",
|
| 69 |
+
"name": "abcyam",
|
| 70 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 71 |
+
"rotation_format": "rot6d",
|
| 72 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceYAMDataset",
|
| 73 |
+
"translation_scale": 1.0
|
| 74 |
+
},
|
| 75 |
+
{
|
| 76 |
+
"camera_mask": true,
|
| 77 |
+
"embodiment": "behavior1k_lerobot",
|
| 78 |
+
"name": "behavior1k",
|
| 79 |
+
"pose_convention": "backward_chunk_anchored_16f",
|
| 80 |
+
"rotation_format": "rot6d",
|
| 81 |
+
"target": "projects.cosmos3.cosmos3.datasets.action.canonical_lance_action_datasets.CanonicalLanceBehavior1KDataset",
|
| 82 |
+
"translation_scale": 1.0
|
| 83 |
+
}
|
| 84 |
+
],
|
| 85 |
+
"experiment": "simulation_sfdmd_bimanual_prod_64N_chunk2_v002_step4",
|
| 86 |
+
"loader": {
|
| 87 |
+
"action_schema": "unified_v1",
|
| 88 |
+
"append_action_caption_semantics": false,
|
| 89 |
+
"append_duration_fps_timestamps": false,
|
| 90 |
+
"append_resolution_info": false,
|
| 91 |
+
"append_viewpoint_info": false,
|
| 92 |
+
"enable_mode_specific_prompt": false,
|
| 93 |
+
"format_prompt_as_json": false,
|
| 94 |
+
"global_asinh": true,
|
| 95 |
+
"metric_gripper": true
|
| 96 |
+
}
|
| 97 |
+
}
|
cosmos3_nano_sim_bimanual_artifact.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cosmos3_nano_sim_bimanual_expected_diffusers_weights.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cosmos3_nano_sim_bimanual_expected_weights.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cosmos_dreams_action_sources/4d38b6321a92741c93e8cf1e96cba18979399ddb5c037708d3aaeffd666faa10.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"embodiment_type": "agibot",
|
| 4 |
+
"pose_convention": "backward_framewise",
|
| 5 |
+
"rotation_format": "rot6d",
|
| 6 |
+
"action_dim": 29,
|
| 7 |
+
"skip_rotation_dims": [3, 4, 5, 6, 7, 8, 12, 13, 14, 15, 16, 17, 22, 23, 24, 25, 26, 27],
|
| 8 |
+
"chunk_length": 16,
|
| 9 |
+
"sample_stride": 16,
|
| 10 |
+
"dataset_name": "agibot_shared_gripper_beta",
|
| 11 |
+
"dataset_class": "AgibotGEARGripperDataset+AgibotGEARGripperExtDataset+AgiBotWorldBetaDataset",
|
| 12 |
+
"dataset_root": "AgiBotWorld-GEAR + AgiBotWorld-Beta",
|
| 13 |
+
"split": "train",
|
| 14 |
+
"num_samples_stats": 200000,
|
| 15 |
+
"reservoir_size": 200000,
|
| 16 |
+
"max_samples": 200000,
|
| 17 |
+
"sampling_seed": 42,
|
| 18 |
+
"dataset_lengths": {
|
| 19 |
+
"agibot_gear_gripper": 10723406,
|
| 20 |
+
"agibot_gear_gripper_ext": 209875,
|
| 21 |
+
"agibotworld_beta": 16688991
|
| 22 |
+
},
|
| 23 |
+
"normalizer_note": "Shared AgiBot 29D FK-pose normalizer for GEAR gripper/ext and AgiBotWorld-Beta. Intended for action_normalization=asinh_rot to compress valid heavy-tailed base motion without clamping."
|
| 24 |
+
},
|
| 25 |
+
"global": {
|
| 26 |
+
"mean": [ 0.000007, -0.000028, 0.000017, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000054, 0.000055, -0.000004, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.461416, -0.000065, 0.000066, 0.000026, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.511525],
|
| 27 |
+
"std": [ 0.001146, 0.003775, 0.006236, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.005170, 0.005957, 0.005775, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.365715, 0.004577, 0.006320, 0.004809, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.368419],
|
| 28 |
+
"min": [-0.332076, -0.461419, -1.287811, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.181742, -1.008569, -0.964748, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000, -1.086141, -1.194795, -0.704462, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000],
|
| 29 |
+
"max": [ 0.332075, 0.461429, 1.287816, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.181742, 1.008569, 0.968047, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.077497, 1.194157, 0.696745, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000],
|
| 30 |
+
"q01": [-0.000474, -0.004285, -0.006603, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -0.011330, -0.012649, -0.014966, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000, -0.010683, -0.011185, -0.012754, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, -1.000000, 0.000000],
|
| 31 |
+
"q99": [ 0.000472, 0.003180, 0.007315, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.011657, 0.012629, 0.013074, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 0.010122, 0.011385, 0.011604, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000, 1.000000]
|
| 32 |
+
},
|
| 33 |
+
"global_raw": {
|
| 34 |
+
"mean": [ 0.000007, -0.000028, 0.000017, 0.999997, 0.000012, -0.000011, -0.000013, 0.999992, 0.000013, 0.000054, 0.000055, -0.000004, 0.999755, 0.000162, -0.000206, -0.000163, 0.999770, -0.000068, 0.461416, -0.000065, 0.000066, 0.000026, 0.999821, -0.000156, 0.000218, 0.000156, 0.999840, -0.000065, 0.511525],
|
| 35 |
+
"std": [ 0.001146, 0.003775, 0.006236, 0.000050, 0.001322, 0.001990, 0.001321, 0.000091, 0.003722, 0.005170, 0.005957, 0.005775, 0.001184, 0.016934, 0.014178, 0.016919, 0.001112, 0.013136, 0.365715, 0.004577, 0.006320, 0.004809, 0.001140, 0.014271, 0.012406, 0.014263, 0.000987, 0.010745, 0.368419],
|
| 36 |
+
"min": [-0.332076, -0.461419, -1.287811, 0.992475, -0.047897, -0.115986, -0.048200, 0.986519, -0.163639, -1.181742, -1.008569, -0.964748, 0.834839, -0.535909, -0.508249, -0.417918, 0.835656, -0.360748, 0.000000, -1.086141, -1.194795, -0.704462, 0.885655, -0.422226, -0.392999, -0.404670, 0.847586, -0.267267, 0.000000],
|
| 37 |
+
"max": [ 0.332075, 0.461429, 1.287816, 1.000000, 0.048245, 0.111197, 0.047794, 1.000000, 0.130982, 1.181742, 1.008569, 0.968047, 1.000000, 0.413419, 0.506131, 0.545759, 1.000000, 0.405109, 1.000000, 1.077497, 1.194157, 0.696745, 1.000000, 0.405554, 0.460652, 0.428101, 1.000000, 0.423156, 1.000000],
|
| 38 |
+
"q01": [-0.000474, -0.004285, -0.006603, 0.999969, -0.000524, -0.000606, -0.000540, 0.999801, -0.000404, -0.011330, -0.012649, -0.014966, 0.995746, -0.055268, -0.044579, -0.057030, 0.995879, -0.041778, 0.000000, -0.010683, -0.011185, -0.012754, 0.996474, -0.048846, -0.038892, -0.044784, 0.996874, -0.034269, 0.000000],
|
| 39 |
+
"q99": [ 0.000472, 0.003180, 0.007315, 1.000000, 0.000532, 0.000603, 0.000514, 1.000000, 0.000454, 0.011657, 0.012629, 0.013074, 1.000000, 0.057221, 0.045662, 0.055355, 1.000000, 0.042508, 1.000000, 0.010122, 0.011385, 0.011604, 1.000000, 0.044671, 0.039433, 0.048886, 1.000000, 0.036261, 1.000000]
|
| 40 |
+
}
|
| 41 |
+
}
|
cosmos_dreams_action_sources/action_schema.json
ADDED
|
@@ -0,0 +1,95 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"action_tokens_per_frame": 4,
|
| 3 |
+
"contract_sha256": "2c75f43813f507ab5d382d5d64c41955b3871ee164c093fa098c8429cb8547db",
|
| 4 |
+
"default_embodiment": "camera_pose",
|
| 5 |
+
"embodiments": {
|
| 6 |
+
"camera_pose": {
|
| 7 |
+
"domain_id": 2,
|
| 8 |
+
"layout": {
|
| 9 |
+
"delta_equation": "T_i^-1 @ T_{i+1}",
|
| 10 |
+
"fields": [
|
| 11 |
+
{
|
| 12 |
+
"name": "camera_translation",
|
| 13 |
+
"offset": 0,
|
| 14 |
+
"size": 3,
|
| 15 |
+
"unit": "meter"
|
| 16 |
+
},
|
| 17 |
+
{
|
| 18 |
+
"name": "camera_rotation",
|
| 19 |
+
"offset": 3,
|
| 20 |
+
"representation": "rot6d_columns",
|
| 21 |
+
"size": 6,
|
| 22 |
+
"unit": "dimensionless"
|
| 23 |
+
}
|
| 24 |
+
],
|
| 25 |
+
"id": "camera_pose_backward_framewise_rot6d_v1",
|
| 26 |
+
"pose_convention": "backward_framewise",
|
| 27 |
+
"rotation_representation": "rot6d_columns"
|
| 28 |
+
},
|
| 29 |
+
"normalizer": {
|
| 30 |
+
"derivation": {
|
| 31 |
+
"rotation_scale": 1.0,
|
| 32 |
+
"translation_scale": 10.0
|
| 33 |
+
},
|
| 34 |
+
"method": "pose_scale",
|
| 35 |
+
"schema_version": 1,
|
| 36 |
+
"training_config": {
|
| 37 |
+
"experiment": "causal_8b_sf_dmd_max_4step_cam_chunk4_480p_961f",
|
| 38 |
+
"repository_revision": "483375f79279832b52d9e5c3598345a86f7137d5",
|
| 39 |
+
"resolved_sha256": "0e7b2be45c82b13c6f1454cdafd7e75d1941a6565e3b72b15a03a99d65177f29"
|
| 40 |
+
},
|
| 41 |
+
"transform": {
|
| 42 |
+
"forward_clamp": false,
|
| 43 |
+
"offset": [
|
| 44 |
+
0.0,
|
| 45 |
+
0.0,
|
| 46 |
+
0.0,
|
| 47 |
+
0.0,
|
| 48 |
+
0.0,
|
| 49 |
+
0.0,
|
| 50 |
+
0.0,
|
| 51 |
+
0.0,
|
| 52 |
+
0.0
|
| 53 |
+
],
|
| 54 |
+
"scale": [
|
| 55 |
+
0.10000000149011612,
|
| 56 |
+
0.10000000149011612,
|
| 57 |
+
0.10000000149011612,
|
| 58 |
+
1.0,
|
| 59 |
+
1.0,
|
| 60 |
+
1.0,
|
| 61 |
+
1.0,
|
| 62 |
+
1.0,
|
| 63 |
+
1.0
|
| 64 |
+
],
|
| 65 |
+
"type": "affine"
|
| 66 |
+
},
|
| 67 |
+
"transform_sha256": "bb8102c686b4813c1cdcc82b78cb174225fa35f8bfafd07d36b81a767bb6e698"
|
| 68 |
+
},
|
| 69 |
+
"raw_action_dim": 9
|
| 70 |
+
}
|
| 71 |
+
},
|
| 72 |
+
"model_action_dim": 64,
|
| 73 |
+
"num_embodiment_domains": 32,
|
| 74 |
+
"padding": {
|
| 75 |
+
"stage": "after_normalization",
|
| 76 |
+
"value": 0.0
|
| 77 |
+
},
|
| 78 |
+
"schema_version": 3,
|
| 79 |
+
"training_config_excerpt": {
|
| 80 |
+
"datasets": [
|
| 81 |
+
{
|
| 82 |
+
"apply_forward_clamp": false,
|
| 83 |
+
"dataset_class": "CameraDatasetSharded",
|
| 84 |
+
"embodiment": "camera_pose",
|
| 85 |
+
"method": "pose_scale",
|
| 86 |
+
"mode": "forward_dynamics",
|
| 87 |
+
"pose_convention": "backward_framewise",
|
| 88 |
+
"rotation_format": "rot6d",
|
| 89 |
+
"rotation_scale": 1.0,
|
| 90 |
+
"translation_scale": 10.0
|
| 91 |
+
}
|
| 92 |
+
],
|
| 93 |
+
"experiment": "causal_8b_sf_dmd_max_4step_cam_chunk4_480p_961f"
|
| 94 |
+
}
|
| 95 |
+
}
|
cosmos_dreams_action_sources/cosmos_dreams_action_schema.schema.json
ADDED
|
@@ -0,0 +1,201 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"$schema": "https://json-schema.org/draft/2020-12/schema",
|
| 3 |
+
"$id": "https://developer.nvidia.com/cosmos-dreams/action-schema-v3.schema.json",
|
| 4 |
+
"title": "Cosmos-Dreams Per-Embodiment Action Schema v3",
|
| 5 |
+
"type": "object",
|
| 6 |
+
"additionalProperties": false,
|
| 7 |
+
"required": ["schema_version", "action_tokens_per_frame", "model_action_dim", "num_embodiment_domains", "default_embodiment", "embodiments", "padding", "training_config_excerpt", "contract_sha256"],
|
| 8 |
+
"properties": {
|
| 9 |
+
"schema_version": {"const": 3},
|
| 10 |
+
"action_tokens_per_frame": {"const": 4},
|
| 11 |
+
"model_action_dim": {"const": 64},
|
| 12 |
+
"num_embodiment_domains": {"const": 32},
|
| 13 |
+
"default_embodiment": {"enum": ["abc_yam", "agibotworld", "agibot_gear_gripper", "agibot_gear_gripper_ext", "camera_pose", "molmoact2_yam", "xdof_yam"]},
|
| 14 |
+
"embodiments": {
|
| 15 |
+
"type": "object",
|
| 16 |
+
"minProperties": 1,
|
| 17 |
+
"additionalProperties": false,
|
| 18 |
+
"properties": {
|
| 19 |
+
"abc_yam": {"$ref": "#/$defs/yam_embodiment"},
|
| 20 |
+
"agibotworld": {"$ref": "#/$defs/agibot_embodiment"},
|
| 21 |
+
"agibot_gear_gripper": {"$ref": "#/$defs/agibot_embodiment"},
|
| 22 |
+
"agibot_gear_gripper_ext": {"$ref": "#/$defs/agibot_embodiment"},
|
| 23 |
+
"camera_pose": {"$ref": "#/$defs/camera_embodiment"},
|
| 24 |
+
"molmoact2_yam": {"$ref": "#/$defs/yam_embodiment"},
|
| 25 |
+
"xdof_yam": {"$ref": "#/$defs/yam_embodiment"}
|
| 26 |
+
}
|
| 27 |
+
},
|
| 28 |
+
"padding": {"const": {"stage": "after_normalization", "value": 0.0}},
|
| 29 |
+
"training_config_excerpt": {
|
| 30 |
+
"type": "object",
|
| 31 |
+
"additionalProperties": false,
|
| 32 |
+
"required": ["datasets", "experiment"],
|
| 33 |
+
"properties": {
|
| 34 |
+
"datasets": {"type": "array", "minItems": 1, "items": {"oneOf": [{"$ref": "#/$defs/agibot_dataset_descriptor"}, {"$ref": "#/$defs/yam_dataset_descriptor"}, {"$ref": "#/$defs/camera_dataset_descriptor"}]}},
|
| 35 |
+
"experiment": {"type": "string", "minLength": 1}
|
| 36 |
+
}
|
| 37 |
+
},
|
| 38 |
+
"contract_sha256": {"$ref": "#/$defs/sha256"}
|
| 39 |
+
},
|
| 40 |
+
"$defs": {
|
| 41 |
+
"sha256": {"type": "string", "pattern": "^[0-9a-f]{64}$"},
|
| 42 |
+
"repository_revision": {"type": "string", "pattern": "^[0-9a-f]{40}$"},
|
| 43 |
+
"training_config": {
|
| 44 |
+
"type": "object",
|
| 45 |
+
"additionalProperties": false,
|
| 46 |
+
"required": ["experiment", "resolved_sha256", "repository_revision"],
|
| 47 |
+
"properties": {
|
| 48 |
+
"experiment": {"type": "string", "minLength": 1},
|
| 49 |
+
"resolved_sha256": {"$ref": "#/$defs/sha256"},
|
| 50 |
+
"repository_revision": {"$ref": "#/$defs/repository_revision"}
|
| 51 |
+
}
|
| 52 |
+
},
|
| 53 |
+
"agibot_layout": {
|
| 54 |
+
"const": {"id": "agibot_backward_framewise_rot6d_v1", "pose_convention": "backward_framewise", "delta_equation": "T_i^-1 @ T_{i+1}", "rotation_representation": "rot6d_columns", "fields": [{"name": "head_translation", "offset": 0, "size": 3, "unit": "meter"}, {"name": "head_rotation", "offset": 3, "size": 6, "unit": "dimensionless", "representation": "rot6d_columns"}, {"name": "right_translation", "offset": 9, "size": 3, "unit": "meter"}, {"name": "right_rotation", "offset": 12, "size": 6, "unit": "dimensionless", "representation": "rot6d_columns"}, {"name": "right_gripper", "offset": 18, "size": 1, "unit": "open_fraction", "closed_value": 0.0, "open_value": 1.0}, {"name": "left_translation", "offset": 19, "size": 3, "unit": "meter"}, {"name": "left_rotation", "offset": 22, "size": 6, "unit": "dimensionless", "representation": "rot6d_columns"}, {"name": "left_gripper", "offset": 28, "size": 1, "unit": "open_fraction", "closed_value": 0.0, "open_value": 1.0}]}
|
| 55 |
+
},
|
| 56 |
+
"yam_layout": {
|
| 57 |
+
"const": {"id": "legacy_yam_fk_backward_framewise_rot6d_v1", "pose_convention": "backward_framewise", "delta_equation": "T_i^-1 @ T_{i+1}", "rotation_representation": "rot6d_columns", "fields": [{"name": "left_translation", "offset": 0, "size": 3, "unit": "meter"}, {"name": "left_rotation", "offset": 3, "size": 6, "unit": "dimensionless", "representation": "rot6d_columns"}, {"name": "left_gripper", "offset": 9, "size": 1, "unit": "open_fraction", "closed_value": 0.0, "open_value": 1.0}, {"name": "right_translation", "offset": 10, "size": 3, "unit": "meter"}, {"name": "right_rotation", "offset": 13, "size": 6, "unit": "dimensionless", "representation": "rot6d_columns"}, {"name": "right_gripper", "offset": 19, "size": 1, "unit": "open_fraction", "closed_value": 0.0, "open_value": 1.0}]}
|
| 58 |
+
},
|
| 59 |
+
"camera_layout": {
|
| 60 |
+
"const": {"id": "camera_pose_backward_framewise_rot6d_v1", "pose_convention": "backward_framewise", "delta_equation": "T_i^-1 @ T_{i+1}", "rotation_representation": "rot6d_columns", "fields": [{"name": "camera_translation", "offset": 0, "size": 3, "unit": "meter"}, {"name": "camera_rotation", "offset": 3, "size": 6, "unit": "dimensionless", "representation": "rot6d_columns"}]}
|
| 61 |
+
},
|
| 62 |
+
"agibot_embodiment": {
|
| 63 |
+
"type": "object",
|
| 64 |
+
"additionalProperties": false,
|
| 65 |
+
"required": ["domain_id", "raw_action_dim", "layout", "normalizer"],
|
| 66 |
+
"properties": {"domain_id": {"const": 15}, "raw_action_dim": {"const": 29}, "layout": {"$ref": "#/$defs/agibot_layout"}, "normalizer": {"$ref": "#/$defs/quantile_rot"}}
|
| 67 |
+
},
|
| 68 |
+
"yam_embodiment": {
|
| 69 |
+
"type": "object",
|
| 70 |
+
"additionalProperties": false,
|
| 71 |
+
"required": ["domain_id", "raw_action_dim", "layout", "normalizer"],
|
| 72 |
+
"properties": {"domain_id": {"const": 16}, "raw_action_dim": {"const": 20}, "layout": {"$ref": "#/$defs/yam_layout"}, "normalizer": {"$ref": "#/$defs/yam_quantile_rot"}}
|
| 73 |
+
},
|
| 74 |
+
"camera_embodiment": {
|
| 75 |
+
"type": "object",
|
| 76 |
+
"additionalProperties": false,
|
| 77 |
+
"required": ["domain_id", "raw_action_dim", "layout", "normalizer"],
|
| 78 |
+
"properties": {"domain_id": {"const": 2}, "raw_action_dim": {"const": 9}, "layout": {"$ref": "#/$defs/camera_layout"}, "normalizer": {"$ref": "#/$defs/pose_scale"}}
|
| 79 |
+
},
|
| 80 |
+
"agibot_dataset_descriptor": {
|
| 81 |
+
"type": "object",
|
| 82 |
+
"additionalProperties": false,
|
| 83 |
+
"required": ["dataset_class", "embodiment", "method", "apply_forward_clamp", "pose_convention", "rotation_format", "stats_filename"],
|
| 84 |
+
"properties": {
|
| 85 |
+
"dataset_class": {"enum": ["AgiBotWorldBetaDataset", "AgibotGEARGripperDataset", "AgibotGEARGripperExtDataset"]},
|
| 86 |
+
"embodiment": {"enum": ["agibotworld", "agibot_gear_gripper", "agibot_gear_gripper_ext"]},
|
| 87 |
+
"method": {"const": "quantile_rot"},
|
| 88 |
+
"apply_forward_clamp": {"const": false},
|
| 89 |
+
"pose_convention": {"const": "backward_framewise"},
|
| 90 |
+
"rotation_format": {"const": "rot6d"},
|
| 91 |
+
"stats_filename": {"enum": ["agibot_backward_framewise_rot6d.json", "agibot_gear_gripper_backward_framewise_rot6d.json"]}
|
| 92 |
+
},
|
| 93 |
+
"allOf": [
|
| 94 |
+
{"if": {"properties": {"dataset_class": {"const": "AgiBotWorldBetaDataset"}}}, "then": {"properties": {"embodiment": {"const": "agibotworld"}, "stats_filename": {"const": "agibot_backward_framewise_rot6d.json"}}}},
|
| 95 |
+
{"if": {"properties": {"dataset_class": {"const": "AgibotGEARGripperDataset"}}}, "then": {"properties": {"embodiment": {"const": "agibot_gear_gripper"}}}},
|
| 96 |
+
{"if": {"properties": {"dataset_class": {"const": "AgibotGEARGripperExtDataset"}}}, "then": {"properties": {"embodiment": {"const": "agibot_gear_gripper_ext"}}}}
|
| 97 |
+
]
|
| 98 |
+
},
|
| 99 |
+
"yam_dataset_descriptor": {
|
| 100 |
+
"type": "object",
|
| 101 |
+
"additionalProperties": false,
|
| 102 |
+
"required": ["dataset_class", "embodiment", "method", "apply_forward_clamp", "pose_convention", "rotation_format", "stats_filename"],
|
| 103 |
+
"properties": {
|
| 104 |
+
"dataset_class": {"enum": ["ABCYAMLeRobotDataset", "MolmoAct2YAMDataset", "XDOFYAMDataset"]},
|
| 105 |
+
"embodiment": {"enum": ["abc_yam", "molmoact2_yam", "xdof_yam"]},
|
| 106 |
+
"method": {"const": "quantile_rot"},
|
| 107 |
+
"apply_forward_clamp": {"const": false},
|
| 108 |
+
"pose_convention": {"const": "backward_framewise"},
|
| 109 |
+
"rotation_format": {"const": "rot6d"},
|
| 110 |
+
"stats_filename": {"enum": ["abc_yam_backward_framewise_rot6d.json", "molmoact2_yam_backward_framewise_rot6d.json", "xdof_yam_backward_framewise_rot6d.json"]}
|
| 111 |
+
},
|
| 112 |
+
"allOf": [
|
| 113 |
+
{"if": {"properties": {"dataset_class": {"const": "ABCYAMLeRobotDataset"}}}, "then": {"properties": {"embodiment": {"const": "abc_yam"}, "stats_filename": {"const": "abc_yam_backward_framewise_rot6d.json"}}}},
|
| 114 |
+
{"if": {"properties": {"dataset_class": {"const": "MolmoAct2YAMDataset"}}}, "then": {"properties": {"embodiment": {"const": "molmoact2_yam"}, "stats_filename": {"const": "molmoact2_yam_backward_framewise_rot6d.json"}}}},
|
| 115 |
+
{"if": {"properties": {"dataset_class": {"const": "XDOFYAMDataset"}}}, "then": {"properties": {"embodiment": {"const": "xdof_yam"}, "stats_filename": {"const": "xdof_yam_backward_framewise_rot6d.json"}}}}
|
| 116 |
+
]
|
| 117 |
+
},
|
| 118 |
+
"camera_dataset_descriptor": {
|
| 119 |
+
"type": "object",
|
| 120 |
+
"additionalProperties": false,
|
| 121 |
+
"required": ["dataset_class", "embodiment", "method", "apply_forward_clamp", "mode", "pose_convention", "rotation_format", "translation_scale", "rotation_scale"],
|
| 122 |
+
"properties": {
|
| 123 |
+
"dataset_class": {"const": "CameraDatasetSharded"},
|
| 124 |
+
"embodiment": {"const": "camera_pose"},
|
| 125 |
+
"method": {"const": "pose_scale"},
|
| 126 |
+
"apply_forward_clamp": {"const": false},
|
| 127 |
+
"mode": {"const": "forward_dynamics"},
|
| 128 |
+
"pose_convention": {"const": "backward_framewise"},
|
| 129 |
+
"rotation_format": {"const": "rot6d"},
|
| 130 |
+
"translation_scale": {"type": "number", "exclusiveMinimum": 0},
|
| 131 |
+
"rotation_scale": {"type": "number", "exclusiveMinimum": 0}
|
| 132 |
+
}
|
| 133 |
+
},
|
| 134 |
+
"quantile_rot": {
|
| 135 |
+
"type": "object",
|
| 136 |
+
"additionalProperties": false,
|
| 137 |
+
"required": ["schema_version", "method", "transform", "derivation", "source", "training_config", "transform_sha256"],
|
| 138 |
+
"properties": {
|
| 139 |
+
"schema_version": {"const": 1},
|
| 140 |
+
"method": {"const": "quantile_rot"},
|
| 141 |
+
"transform": {
|
| 142 |
+
"type": "object",
|
| 143 |
+
"additionalProperties": false,
|
| 144 |
+
"required": ["type", "offset", "scale", "forward_clamp"],
|
| 145 |
+
"properties": {"type": {"const": "affine"}, "offset": {"type": "array", "minItems": 29, "maxItems": 29, "items": {"type": "number"}}, "scale": {"type": "array", "minItems": 29, "maxItems": 29, "items": {"type": "number", "exclusiveMinimum": 0}}, "forward_clamp": {"const": false}}
|
| 146 |
+
},
|
| 147 |
+
"derivation": {"type": "object", "additionalProperties": false, "required": ["statistics_block", "low_key", "high_key", "range_floor"], "properties": {"statistics_block": {"const": "global_raw"}, "low_key": {"const": "q01"}, "high_key": {"const": "q99"}, "range_floor": {"const": 9.99999993922529e-9}}},
|
| 148 |
+
"source": {
|
| 149 |
+
"type": "object",
|
| 150 |
+
"additionalProperties": false,
|
| 151 |
+
"required": ["path", "artifact_path", "sha256", "repository_revision"],
|
| 152 |
+
"properties": {"path": {"enum": ["projects/cosmos3/cosmos3/datasets/action/normalizers/agibot_backward_framewise_rot6d.json", "projects/cosmos3/cosmos3/datasets/action/normalizers/agibot_gear_gripper_backward_framewise_rot6d.json"]}, "artifact_path": {"type": "string", "pattern": "^cosmos_dreams_action_sources/[0-9a-f]{64}\\.json$"}, "sha256": {"$ref": "#/$defs/sha256"}, "repository_revision": {"$ref": "#/$defs/repository_revision"}}
|
| 153 |
+
},
|
| 154 |
+
"training_config": {"$ref": "#/$defs/training_config"},
|
| 155 |
+
"transform_sha256": {"$ref": "#/$defs/sha256"}
|
| 156 |
+
}
|
| 157 |
+
},
|
| 158 |
+
"yam_quantile_rot": {
|
| 159 |
+
"type": "object",
|
| 160 |
+
"additionalProperties": false,
|
| 161 |
+
"required": ["schema_version", "method", "transform", "derivation", "source", "training_config", "transform_sha256"],
|
| 162 |
+
"properties": {
|
| 163 |
+
"schema_version": {"const": 1},
|
| 164 |
+
"method": {"const": "quantile_rot"},
|
| 165 |
+
"transform": {
|
| 166 |
+
"type": "object",
|
| 167 |
+
"additionalProperties": false,
|
| 168 |
+
"required": ["type", "offset", "scale", "forward_clamp"],
|
| 169 |
+
"properties": {"type": {"const": "affine"}, "offset": {"type": "array", "minItems": 20, "maxItems": 20, "items": {"type": "number"}}, "scale": {"type": "array", "minItems": 20, "maxItems": 20, "items": {"type": "number", "exclusiveMinimum": 0}}, "forward_clamp": {"const": false}}
|
| 170 |
+
},
|
| 171 |
+
"derivation": {"type": "object", "additionalProperties": false, "required": ["statistics_block", "low_key", "high_key", "range_floor"], "properties": {"statistics_block": {"const": "global_raw"}, "low_key": {"const": "q01"}, "high_key": {"const": "q99"}, "range_floor": {"const": 9.99999993922529e-9}}},
|
| 172 |
+
"source": {
|
| 173 |
+
"type": "object",
|
| 174 |
+
"additionalProperties": false,
|
| 175 |
+
"required": ["path", "artifact_path", "sha256", "repository_revision"],
|
| 176 |
+
"properties": {"path": {"enum": ["projects/cosmos3/cosmos3/datasets/action/legacy/normalizers/abc_yam_backward_framewise_rot6d.json", "projects/cosmos3/cosmos3/datasets/action/legacy/normalizers/molmoact2_yam_backward_framewise_rot6d.json", "projects/cosmos3/cosmos3/datasets/action/legacy/normalizers/xdof_yam_backward_framewise_rot6d.json"]}, "artifact_path": {"type": "string", "pattern": "^cosmos_dreams_action_sources/[0-9a-f]{64}\\.json$"}, "sha256": {"$ref": "#/$defs/sha256"}, "repository_revision": {"$ref": "#/$defs/repository_revision"}}
|
| 177 |
+
},
|
| 178 |
+
"training_config": {"$ref": "#/$defs/training_config"},
|
| 179 |
+
"transform_sha256": {"$ref": "#/$defs/sha256"}
|
| 180 |
+
}
|
| 181 |
+
},
|
| 182 |
+
"pose_scale": {
|
| 183 |
+
"type": "object",
|
| 184 |
+
"additionalProperties": false,
|
| 185 |
+
"required": ["schema_version", "method", "transform", "derivation", "training_config", "transform_sha256"],
|
| 186 |
+
"properties": {
|
| 187 |
+
"schema_version": {"const": 1},
|
| 188 |
+
"method": {"const": "pose_scale"},
|
| 189 |
+
"transform": {
|
| 190 |
+
"type": "object",
|
| 191 |
+
"additionalProperties": false,
|
| 192 |
+
"required": ["type", "offset", "scale", "forward_clamp"],
|
| 193 |
+
"properties": {"type": {"const": "affine"}, "offset": {"type": "array", "minItems": 9, "maxItems": 9, "items": {"const": 0.0}}, "scale": {"type": "array", "minItems": 9, "maxItems": 9, "items": {"type": "number", "exclusiveMinimum": 0}}, "forward_clamp": {"const": false}}
|
| 194 |
+
},
|
| 195 |
+
"derivation": {"type": "object", "additionalProperties": false, "required": ["translation_scale", "rotation_scale"], "properties": {"translation_scale": {"type": "number", "exclusiveMinimum": 0}, "rotation_scale": {"type": "number", "exclusiveMinimum": 0}}},
|
| 196 |
+
"training_config": {"$ref": "#/$defs/training_config"},
|
| 197 |
+
"transform_sha256": {"$ref": "#/$defs/sha256"}
|
| 198 |
+
}
|
| 199 |
+
}
|
| 200 |
+
}
|
| 201 |
+
}
|
cosmos_dreams_action_sources/resolved_action_config.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"datasets": [
|
| 3 |
+
{
|
| 4 |
+
"apply_forward_clamp": false,
|
| 5 |
+
"dataset_class": "CameraDatasetSharded",
|
| 6 |
+
"embodiment": "camera_pose",
|
| 7 |
+
"method": "pose_scale",
|
| 8 |
+
"mode": "forward_dynamics",
|
| 9 |
+
"pose_convention": "backward_framewise",
|
| 10 |
+
"rotation_format": "rot6d",
|
| 11 |
+
"rotation_scale": 1.0,
|
| 12 |
+
"translation_scale": 10.0
|
| 13 |
+
}
|
| 14 |
+
],
|
| 15 |
+
"experiment": "causal_8b_sf_dmd_max_4step_cam_chunk4_480p_961f"
|
| 16 |
+
}
|
cosmos_dreams_artifact.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cosmos_dreams_expected_diffusers_weights.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cosmos_dreams_expected_weights.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cosmos_dreams_manifest.json
ADDED
|
@@ -0,0 +1,131 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"attention_mode": "three_way",
|
| 3 |
+
"checkpoint_hash": "9540afba2ac43d36b213a847f227b8011f5419feb670be722b518debdc22f63e",
|
| 4 |
+
"checkpoint_id": "causal_8b_sf_dmd_max_4step_cam_chunk4_480p_961f@iter_000002000",
|
| 5 |
+
"checkpoint_iteration": 2000,
|
| 6 |
+
"chunk_partition": [
|
| 7 |
+
1,
|
| 8 |
+
4
|
| 9 |
+
],
|
| 10 |
+
"chunk_size": 4,
|
| 11 |
+
"conditioning": {
|
| 12 |
+
"action_tokens_per_frame": 4,
|
| 13 |
+
"contract_sha256": "2c75f43813f507ab5d382d5d64c41955b3871ee164c093fa098c8429cb8547db",
|
| 14 |
+
"default_embodiment": "camera_pose",
|
| 15 |
+
"embodiments": {
|
| 16 |
+
"camera_pose": {
|
| 17 |
+
"domain_id": 2,
|
| 18 |
+
"layout": {
|
| 19 |
+
"delta_equation": "T_i^-1 @ T_{i+1}",
|
| 20 |
+
"fields": [
|
| 21 |
+
{
|
| 22 |
+
"name": "camera_translation",
|
| 23 |
+
"offset": 0,
|
| 24 |
+
"size": 3,
|
| 25 |
+
"unit": "meter"
|
| 26 |
+
},
|
| 27 |
+
{
|
| 28 |
+
"name": "camera_rotation",
|
| 29 |
+
"offset": 3,
|
| 30 |
+
"representation": "rot6d_columns",
|
| 31 |
+
"size": 6,
|
| 32 |
+
"unit": "dimensionless"
|
| 33 |
+
}
|
| 34 |
+
],
|
| 35 |
+
"id": "camera_pose_backward_framewise_rot6d_v1",
|
| 36 |
+
"pose_convention": "backward_framewise",
|
| 37 |
+
"rotation_representation": "rot6d_columns"
|
| 38 |
+
},
|
| 39 |
+
"normalizer": {
|
| 40 |
+
"derivation": {
|
| 41 |
+
"rotation_scale": 1.0,
|
| 42 |
+
"translation_scale": 10.0
|
| 43 |
+
},
|
| 44 |
+
"method": "pose_scale",
|
| 45 |
+
"schema_version": 1,
|
| 46 |
+
"training_config": {
|
| 47 |
+
"experiment": "causal_8b_sf_dmd_max_4step_cam_chunk4_480p_961f",
|
| 48 |
+
"repository_revision": "483375f79279832b52d9e5c3598345a86f7137d5",
|
| 49 |
+
"resolved_sha256": "0e7b2be45c82b13c6f1454cdafd7e75d1941a6565e3b72b15a03a99d65177f29"
|
| 50 |
+
},
|
| 51 |
+
"transform": {
|
| 52 |
+
"forward_clamp": false,
|
| 53 |
+
"offset": [
|
| 54 |
+
0.0,
|
| 55 |
+
0.0,
|
| 56 |
+
0.0,
|
| 57 |
+
0.0,
|
| 58 |
+
0.0,
|
| 59 |
+
0.0,
|
| 60 |
+
0.0,
|
| 61 |
+
0.0,
|
| 62 |
+
0.0
|
| 63 |
+
],
|
| 64 |
+
"scale": [
|
| 65 |
+
0.10000000149011612,
|
| 66 |
+
0.10000000149011612,
|
| 67 |
+
0.10000000149011612,
|
| 68 |
+
1.0,
|
| 69 |
+
1.0,
|
| 70 |
+
1.0,
|
| 71 |
+
1.0,
|
| 72 |
+
1.0,
|
| 73 |
+
1.0
|
| 74 |
+
],
|
| 75 |
+
"type": "affine"
|
| 76 |
+
},
|
| 77 |
+
"transform_sha256": "bb8102c686b4813c1cdcc82b78cb174225fa35f8bfafd07d36b81a767bb6e698"
|
| 78 |
+
},
|
| 79 |
+
"raw_action_dim": 9
|
| 80 |
+
}
|
| 81 |
+
},
|
| 82 |
+
"mode": "action",
|
| 83 |
+
"model_action_dim": 64,
|
| 84 |
+
"num_embodiment_domains": 32,
|
| 85 |
+
"padding": {
|
| 86 |
+
"stage": "after_normalization",
|
| 87 |
+
"value": 0.0
|
| 88 |
+
},
|
| 89 |
+
"schema_version": 3,
|
| 90 |
+
"training_config_excerpt": {
|
| 91 |
+
"datasets": [
|
| 92 |
+
{
|
| 93 |
+
"apply_forward_clamp": false,
|
| 94 |
+
"dataset_class": "CameraDatasetSharded",
|
| 95 |
+
"embodiment": "camera_pose",
|
| 96 |
+
"method": "pose_scale",
|
| 97 |
+
"mode": "forward_dynamics",
|
| 98 |
+
"pose_convention": "backward_framewise",
|
| 99 |
+
"rotation_format": "rot6d",
|
| 100 |
+
"rotation_scale": 1.0,
|
| 101 |
+
"translation_scale": 10.0
|
| 102 |
+
}
|
| 103 |
+
],
|
| 104 |
+
"experiment": "causal_8b_sf_dmd_max_4step_cam_chunk4_480p_961f"
|
| 105 |
+
}
|
| 106 |
+
},
|
| 107 |
+
"fixed_step_sampler_config": {
|
| 108 |
+
"num_train_timesteps": 1000,
|
| 109 |
+
"sample_type": "sde",
|
| 110 |
+
"t_list": [
|
| 111 |
+
1.0,
|
| 112 |
+
0.9375,
|
| 113 |
+
0.8333333333333334,
|
| 114 |
+
0.625
|
| 115 |
+
]
|
| 116 |
+
},
|
| 117 |
+
"latent_patch_size": 2,
|
| 118 |
+
"mrope": {
|
| 119 |
+
"base_fps": 24.0,
|
| 120 |
+
"enable_fps_modulation": true,
|
| 121 |
+
"reset_spatial_ids": true,
|
| 122 |
+
"temporal_modality_margin": 15000
|
| 123 |
+
},
|
| 124 |
+
"schema_version": 1,
|
| 125 |
+
"sink_frames": 0,
|
| 126 |
+
"temporal_compression_factor": 4,
|
| 127 |
+
"text_cache_max_len": 512,
|
| 128 |
+
"vae_spatial_compression_factor": 16,
|
| 129 |
+
"video_temporal_causal": true,
|
| 130 |
+
"window_frames": 96
|
| 131 |
+
}
|
examples/inference/README.md
ADDED
|
@@ -0,0 +1,96 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Sim-Bimanual inference examples
|
| 2 |
+
|
| 3 |
+
These manifests provide 30 FPS examples for the `normalized_unified_v1`
|
| 4 |
+
action contract used by Cosmos3 Nano Sim Bimanual. Each row contains an
|
| 5 |
+
initial image, a plain-text prompt, and normalized 59D action rows. The
|
| 6 |
+
30-second examples use 900 actions for 901 frames; the 60-second AgiBot
|
| 7 |
+
example uses 1800 actions for 1801 frames. The initial image is frame 0 of
|
| 8 |
+
its source sequence; target future frames are not included.
|
| 9 |
+
|
| 10 |
+
Image and action paths use Hugging Face `/resolve/main/` URLs. Keep `HF_TOKEN`
|
| 11 |
+
available to the Framework inference process when the gated repository
|
| 12 |
+
requires authentication. Store the token outside the checkout and do not put
|
| 13 |
+
its value in manifests, command arguments, or logs.
|
| 14 |
+
|
| 15 |
+
| Manifest | Samples | Action routing |
|
| 16 |
+
| --- | --- | --- |
|
| 17 |
+
| `sim_bimanual_unifiedv1_hf.jsonl` | Five AgiBot and two YAM trajectories | Scalar or per-row unified-action domains |
|
| 18 |
+
| `sim_bimanual_mixed_action_hf.jsonl` | Two AgiBot examples | Slow camera loop at rows 300-599 plus source camera masking in the surrounding robot spans |
|
| 19 |
+
| `sim_bimanual_agibot_long60_hf.jsonl` | One 60-second AgiBot-scene example | All 1800 rows use `agibotworld` (domain 15) with nonzero, varying robot actions; the source 900-row program is repeated twice |
|
| 20 |
+
|
| 21 |
+
The mixed-action rows are `agibot_mixed_007` and `agibot_mixed_028`. Their
|
| 22 |
+
first and last 300-row spans preserve the source production camera masking:
|
| 23 |
+
stationary robot rows use `camera_pose` (domain 2), retain the first nine
|
| 24 |
+
camera values, and zero robot-only slots 9-58. The middle 300 rows retain the
|
| 25 |
+
slow camera loop. The complete sidecars contain 385 camera rows for
|
| 26 |
+
`agibot_mixed_007` and 402 for `agibot_mixed_028`; all other rows retain their
|
| 27 |
+
original AgiBot actions.
|
| 28 |
+
|
| 29 |
+
Every PNG stores the unpadded content region. Framework reconstructs the model
|
| 30 |
+
canvas and spatial metadata during preprocessing.
|
| 31 |
+
|
| 32 |
+
## Run inference
|
| 33 |
+
|
| 34 |
+
From a current [Cosmos Framework](https://github.com/nvidia/cosmos-framework)
|
| 35 |
+
checkout with the inference environment activated:
|
| 36 |
+
|
| 37 |
+
```bash
|
| 38 |
+
export COSMOS_TRAINING=0
|
| 39 |
+
CHECKPOINT_DIR=/path/to/Cosmos3-Nano-Sim-Bimanual
|
| 40 |
+
INPUT_MANIFEST="$CHECKPOINT_DIR/examples/inference/sim_bimanual_unifiedv1_hf.jsonl"
|
| 41 |
+
OUTPUT_DIR=outputs/sim_bimanual_examples
|
| 42 |
+
|
| 43 |
+
torchrun --standalone --nproc-per-node=1 \
|
| 44 |
+
-m cosmos_framework.scripts.inference \
|
| 45 |
+
-i "$INPUT_MANIFEST" -o "$OUTPUT_DIR" \
|
| 46 |
+
--checkpoint-path "$CHECKPOINT_DIR" \
|
| 47 |
+
--model-mode=FORWARD_DYNAMICS --resolution=480 --image-size=480 \
|
| 48 |
+
--num-steps=4 --seed=2026 \
|
| 49 |
+
--autoregressive=True --kv-cache-inference-size=64 --attention-sink-size=8 \
|
| 50 |
+
--parallelism-preset=latency --max-num-seqs=1 \
|
| 51 |
+
--no-use-torch-compile --no-diffusion-cache --no-guardrails \
|
| 52 |
+
--video-save-quality=4
|
| 53 |
+
```
|
| 54 |
+
|
| 55 |
+
Select another manifest by changing `INPUT_MANIFEST` and using a new output
|
| 56 |
+
directory. Run one active sequence at a time with `--max-num-seqs=1`.
|
| 57 |
+
The checkpoint stores the four-step SDE schedule
|
| 58 |
+
`[1.0, 0.9375, 0.8333333333333334, 0.625]`; keep `--num-steps=4`.
|
| 59 |
+
|
| 60 |
+
## 60-second AgiBot example
|
| 61 |
+
|
| 62 |
+
`agibot_original_long60_019_bulk_seed2026` is included in the combined manifest
|
| 63 |
+
and in `sim_bimanual_agibot_long60_hf.jsonl` for a focused run. It uses the exact
|
| 64 |
+
initial observation and production v3 prompt from the selected website case.
|
| 65 |
+
Its 1800 normalized action rows repeat the audited 900-row source program
|
| 66 |
+
twice. All rows use `agibotworld` (domain 15), with nonzero, varying wrist and
|
| 67 |
+
gripper actions preserved from the frozen production input. No camera-only
|
| 68 |
+
substitution, renormalization or prompt rewriting is applied during export.
|
| 69 |
+
This is an open-loop action-conditioned rollout, not a new 60-second source
|
| 70 |
+
episode or a closed-loop policy evaluation.
|
| 71 |
+
|
| 72 |
+
To run this example alone, select:
|
| 73 |
+
|
| 74 |
+
```bash
|
| 75 |
+
INPUT_MANIFEST="$CHECKPOINT_DIR/examples/inference/sim_bimanual_agibot_long60_hf.jsonl"
|
| 76 |
+
OUTPUT_DIR=outputs/sim_bimanual_agibot_long60
|
| 77 |
+
```
|
| 78 |
+
|
| 79 |
+
Then run the command above. It reads `action_chunk_size` and `num_frames`
|
| 80 |
+
from each manifest row, allowing 901- and 1801-frame examples in the same
|
| 81 |
+
manifest. Do not override these with `--action-chunk-size=900 --num-frames=901`
|
| 82 |
+
when running the 60-second example. If setting them explicitly for the focused
|
| 83 |
+
manifest, use `--action-chunk-size=1800 --num-frames=1801`.
|
| 84 |
+
|
| 85 |
+
## Optional frustum history retrieval
|
| 86 |
+
|
| 87 |
+
`bimanual_frustum.json` is a reusable KV-history configuration. Copy a
|
| 88 |
+
manifest row and add a `kv_cache_frustum_path` field pointing to that file.
|
| 89 |
+
Replace its estimated 90-degree horizontal FOV with measured camera
|
| 90 |
+
intrinsics when available. Run frustum retrieval with:
|
| 91 |
+
|
| 92 |
+
```bash
|
| 93 |
+
--no-use-torch-compile --kv-cache-inference-size=0 --attention-sink-size=0
|
| 94 |
+
```
|
| 95 |
+
|
| 96 |
+
This mode uses an unbounded physical KV archive.
|
examples/inference/bimanual_frustum.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"pose_source": "bimanual_unified_v1_chunk16",
|
| 3 |
+
"image_size": [
|
| 4 |
+
832,
|
| 5 |
+
480
|
| 6 |
+
],
|
| 7 |
+
"fps": 30,
|
| 8 |
+
"horizontal_fov_degrees": 90.0,
|
| 9 |
+
"near": 0.1,
|
| 10 |
+
"far": 10.0,
|
| 11 |
+
"sink_size": 8,
|
| 12 |
+
"retrieved_size": 32,
|
| 13 |
+
"recent_size": 32
|
| 14 |
+
}
|
examples/inference/sim_bimanual_agibot_long60_hf.jsonl
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"name":"agibot_original_long60_019_bulk_seed2026","model_mode":"forward_dynamics","prompt":"A landscape, high oblique overhead medium-wide shot of an indoor tabletop robotics station, sharply focused. A white rectangular table with rounded corners runs diagonally through most of the frame over a pale beige tiled floor; a pale green wall or partition stands behind it, a tan perforated pegboard with evenly spaced dark holes is cropped along frame-left, and a partial light gray chair back sits beyond the table in the upper-right background. A matte black vase with green leafy stems and small pale blossoms rests near the upper center of the tabletop. On the right side of the table, a crumpled pink cloth with a small white tag bearing unreadable markings lies below and slightly right of several brown liquid spots, while a small crumpled translucent paper or tissue rests nearer the middle-right area. A black open trash bin lined with a black plastic bag sits below the table’s left-front edge in the lower-left/center foreground, partly under the tabletop, with discarded items already visible inside, including a clear plastic cup with a lid and straw and another container-like item with unreadable labeling. Soft, diffuse indoor lighting, mild shadows, moderate contrast, and a muted palette of white, black, pale green, beige, pink, brown, and plant green give the setting a controlled laboratory demonstration mood. A left robotic arm, cropped by frame-left, has a white cylindrical segment, black wrist and gripper assembly, visible wiring, a small bracket, and a cyan-green illuminated ring; its gripper points toward the bin rim and remains close to it. A larger right robotic arm, cropped by frame-right, has a white arm body, black cable loops, metal fasteners, a small rectangular wrist-mounted module, a black multi-finger gripper, and a cyan-green illuminated ring. The right arm occupies the right half and appears larger and nearer than the left arm; the left arm occupies the left third. The right gripper holds a clear plastic cup with a white lid and a tan straw protruding upward, with light residue or liquid visible at the bottom.","vision_path":"https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/sim_bimanual/agibot/images/agibot_original_long60_019_initial.png","domain_name":"agibotworld","action_condition":true,"raw_action_dim":59,"resolution":"480","aspect_ratio":"16,9","fps":30,"num_frames":1801,"action_chunk_size":1800,"num_steps":4,"guidance":3.0,"shift":5.0,"seed":2026,"autoregressive":true,"action_path":"https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/sim_bimanual/agibot/actions/agibot_original_long60_019_normalized_unifiedv1.json","format_prompt_as_json":false}
|
examples/inference/sim_bimanual_mixed_action_hf.jsonl
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"name":"agibot_mixed_007_bulk_seed2026","model_mode":"forward_dynamics","prompt":"The scene opens directly into a high-overhead landscape shot centered on a round white marble-pattern table with subtle gray veining and a black circular rim, sharply focused under bright, even overhead lighting that casts soft shadows and gives the clean indoor setup a controlled demonstration-lab mood. Gray-brown wood-look flooring surrounds the table; dark brown cabinet fronts line the upper background, with an open recess beneath one cabinet showing loose cables; a tall dark vertical appliance or cabinet face stands at frame-right; a pale wall or vertical panel borders frame-left; and a small brown bag-like object rests on the floor near the left side of the table, partly hidden by the left robot. A clear glass vase at the back/top-center of the tabletop holds cream-white flowers, green leaves, and small orange-red sprigs. A dark blue ribbed cup with a light interior and dark handle sits near the lower-left/center of the table, a folded deep-blue rectangular cloth lies on the right half, and a small beige oval stain sits below the cloth in the lower-right quadrant. Two large industrial-style robotic arms flank the table: the left arm is cropped by the left edge and has a white/silver structural segment, black cylindrical wrist housing, black articulated opposing gripper fingers, exposed pivots and screws, black cables, and a small yellow connector; the right arm is cropped by the right edge and has a white/silver structural segment, black wrist and gripper assembly, black cable loops, small blue/green indicator lights, and a small yellow triangular marking. Small physical markings on the robot housings are not clearly readable. The table fills most of the central frame, the vase stands behind the cup and cloth, the cup is smaller than either gripper but larger than the beige stain, and both robots face inward across the tabletop without touching each other.","vision_path":"https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/sim_bimanual/agibot/images/agibot_mixed_007_initial.png","domain_name":"agibotworld","action_condition":true,"raw_action_dim":59,"resolution":"480","aspect_ratio":"16,9","fps":30,"num_frames":901,"action_chunk_size":900,"num_steps":4,"guidance":3.0,"shift":5.0,"seed":2026,"autoregressive":true,"action_path":"https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/sim_bimanual/agibot/actions/agibot_mixed_007_normalized_mixed_action.json","format_prompt_as_json":false}
|
| 2 |
+
{"name":"agibot_mixed_028_bulk_seed2026","model_mode":"forward_dynamics","prompt":"The scene opens in a continuous landscape shot from a high oblique overhead angle, sharply focused across a glossy white kitchen countertop with faint gray streaking, soft reflections, and diffuse cool-neutral indoor light that casts gentle moving shadows and gives the workspace a clinical, controlled demonstration quality. A stainless-steel sink basin with a rounded rectangular rim, inner metal bowl, drain area, and a cropped vertical faucet assembly occupies frame-left and bottom-left; a pink textured scrub pad rests near the back-left corner beside it. A pale gray-white marble-pattern backsplash spans the rear, with a white-framed dark reflective window in the upper-left background showing indistinct reflected or exterior shapes. Along the rear counter sit a white pump bottle with an orange pump and a red physical label reading “SODA,” a small yellow rectangular container with a green-and-white label whose smaller text is unreadable, and two tall clear bottles of amber liquid, one capped yellow and one capped brown, with colored labels containing unreadable non-Latin characters. A white appliance or cabinet face with a dark inset panel is cropped at far frame-right. Centered between the arms, a soft mint-green terry-like cloth with stitched edging and a small white tag lies folded into overlapping triangular layers, occupying roughly one fifth of the frame. A left robotic arm, partially cropped by the left edge, extends from the sink side with a white cylindrical segment, silver joint ring, glowing blue circular indicator, black looped cabling, gray wrist module, black cylindrical wrist, and black angular multi-link pincer gripper bearing small unreadable white markings. A matching right robotic arm, partially cropped by the right edge, has white and silver sections, black cabling, gray wrist module, black cylindrical wrist, black articulated pincer gripper, small yellow cable ties or connectors, and glowing blue-green indicator lights. The arms fill the lower-left and lower-right portions of the frame, balance the composition, and face inward toward each other and the cloth; the rear bottles sit behind the cloth along the backsplash, while the sink lies nearer the left foreground under the left arm.","vision_path":"https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png","domain_name":"agibotworld","action_condition":true,"raw_action_dim":59,"resolution":"480","aspect_ratio":"16,9","fps":30,"num_frames":901,"action_chunk_size":900,"num_steps":4,"guidance":3.0,"shift":5.0,"seed":2026,"autoregressive":true,"action_path":"https://huggingface.co/nvidia/Cosmos3-Nano-Sim-Bimanual/resolve/main/assets/sim_bimanual/agibot/actions/agibot_mixed_028_normalized_mixed_action.json","format_prompt_as_json":false}
|