Cosmos
Diffusers
Safetensors
cosmos3_omni
nvidia
cosmos3
omnimodel
Commit
441c3ef
·
0 Parent(s):

Initial model release

Browse files

Co-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
Files changed (50) hide show
  1. .gitattributes +65 -0
  2. README.md +571 -0
  3. assets/output/cosmos3-nano-sim-bimanual-quickstart.mp4 +3 -0
  4. assets/sim_bimanual/agibot/actions/agibot_001_normalized_unifiedv1.json +0 -0
  5. assets/sim_bimanual/agibot/actions/agibot_034_normalized_unifiedv1.json +0 -0
  6. assets/sim_bimanual/agibot/actions/agibot_mixed_007_normalized_mixed_action.json +0 -0
  7. assets/sim_bimanual/agibot/actions/agibot_mixed_028_normalized_mixed_action.json +0 -0
  8. assets/sim_bimanual/agibot/actions/agibot_original_long60_019_normalized_unifiedv1.json +0 -0
  9. assets/sim_bimanual/agibot/images/agibot_001_initial.png +3 -0
  10. assets/sim_bimanual/agibot/images/agibot_034_initial.png +3 -0
  11. assets/sim_bimanual/agibot/images/agibot_mixed_007_initial.png +3 -0
  12. assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png +3 -0
  13. assets/sim_bimanual/agibot/images/agibot_original_long60_019_initial.png +3 -0
  14. assets/sim_bimanual/yam/actions/yam_019_normalized_unifiedv1.json +0 -0
  15. assets/sim_bimanual/yam/actions/yam_023_normalized_unifiedv1.json +0 -0
  16. assets/sim_bimanual/yam/images/yam_019_initial.png +3 -0
  17. assets/sim_bimanual/yam/images/yam_023_initial.png +3 -0
  18. assets/training/agibot/actions/agibot_001_normalized_unifiedv1.json +0 -0
  19. assets/training/agibot/actions/agibot_006_normalized_unifiedv1.json +0 -0
  20. assets/training/agibot/actions/agibot_034_normalized_unifiedv1.json +0 -0
  21. assets/training/agibot/gt/agibot_001_gt.mp4 +3 -0
  22. assets/training/agibot/gt/agibot_006_gt.mp4 +3 -0
  23. assets/training/agibot/gt/agibot_034_gt.mp4 +3 -0
  24. assets/training/agibot/images/agibot_001_initial.png +3 -0
  25. assets/training/agibot/images/agibot_006_initial.png +3 -0
  26. assets/training/agibot/images/agibot_034_initial.png +3 -0
  27. bias.md +11 -0
  28. chat_template.json +3 -0
  29. checkpoint.json +249 -0
  30. config.json +424 -0
  31. cosmos3_nano_sim_bimanual_action_sources/4d38b6321a92741c93e8cf1e96cba18979399ddb5c037708d3aaeffd666faa10.json +41 -0
  32. cosmos3_nano_sim_bimanual_action_sources/action_schema.json +208 -0
  33. cosmos3_nano_sim_bimanual_action_sources/ae695e2c643a5626b3c3564f00705aa3a83ab0da7640549c7340db71172594b0.json +162 -0
  34. cosmos3_nano_sim_bimanual_action_sources/cosmos3_nano_sim_bimanual_action_schema.schema.json +300 -0
  35. cosmos3_nano_sim_bimanual_action_sources/resolved_action_config.json +97 -0
  36. cosmos3_nano_sim_bimanual_artifact.json +0 -0
  37. cosmos3_nano_sim_bimanual_expected_diffusers_weights.json +0 -0
  38. cosmos3_nano_sim_bimanual_expected_weights.json +0 -0
  39. cosmos_dreams_action_sources/4d38b6321a92741c93e8cf1e96cba18979399ddb5c037708d3aaeffd666faa10.json +41 -0
  40. cosmos_dreams_action_sources/action_schema.json +95 -0
  41. cosmos_dreams_action_sources/cosmos_dreams_action_schema.schema.json +201 -0
  42. cosmos_dreams_action_sources/resolved_action_config.json +16 -0
  43. cosmos_dreams_artifact.json +0 -0
  44. cosmos_dreams_expected_diffusers_weights.json +0 -0
  45. cosmos_dreams_expected_weights.json +0 -0
  46. cosmos_dreams_manifest.json +131 -0
  47. examples/inference/README.md +96 -0
  48. examples/inference/bimanual_frustum.json +14 -0
  49. examples/inference/sim_bimanual_agibot_long60_hf.jsonl +1 -0
  50. 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
+ ![Cosmos3 product and examples lineage highlighting Cosmos3-Nano-Sim-Bimanual](images/cosmos3-nano-sim-bimanual-graph.png)
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

  • SHA256: d3abe6f6d9934d6854bc5e62a42851f689c1ca5e60bda13b7a9756b48b92f0ed
  • Pointer size: 131 Bytes
  • Size of remote file: 293 kB
assets/sim_bimanual/agibot/images/agibot_034_initial.png ADDED

Git LFS Details

  • SHA256: 909c5b534d7943fa8d673a695f718d6a1cde4336c9f1a42354af5f2ead20e13d
  • Pointer size: 131 Bytes
  • Size of remote file: 215 kB
assets/sim_bimanual/agibot/images/agibot_mixed_007_initial.png ADDED

Git LFS Details

  • SHA256: 7ea92af0c196144c664b35a50311144307780864bb563081a9a794986015a736
  • Pointer size: 131 Bytes
  • Size of remote file: 219 kB
assets/sim_bimanual/agibot/images/agibot_mixed_028_initial.png ADDED

Git LFS Details

  • SHA256: 0f475a71835620b937db27d7ed469c5a9ff57eb403298a9caa4b108c12421d74
  • Pointer size: 131 Bytes
  • Size of remote file: 227 kB
assets/sim_bimanual/agibot/images/agibot_original_long60_019_initial.png ADDED

Git LFS Details

  • SHA256: a82b7dc28283b13dc8cf8eee8fec922f297e5f74633d300d6050166af3d36114
  • Pointer size: 131 Bytes
  • Size of remote file: 215 kB
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

  • SHA256: 7bf0a8bf8aaa04a66b7156ea5e70edb45ea1523f0d874bfa538b253041e59146
  • Pointer size: 131 Bytes
  • Size of remote file: 447 kB
assets/sim_bimanual/yam/images/yam_023_initial.png ADDED

Git LFS Details

  • SHA256: a0827dd6ac1e50678031a24c2cd0cb275ac057d8c82b7c836761ba7639113b65
  • Pointer size: 131 Bytes
  • Size of remote file: 216 kB
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

  • SHA256: d3abe6f6d9934d6854bc5e62a42851f689c1ca5e60bda13b7a9756b48b92f0ed
  • Pointer size: 131 Bytes
  • Size of remote file: 293 kB
assets/training/agibot/images/agibot_006_initial.png ADDED

Git LFS Details

  • SHA256: 98a7bf4b7655aa1e6449e49626fef9ec440ee95cd08f8276181f4ba70f9cef51
  • Pointer size: 131 Bytes
  • Size of remote file: 195 kB
assets/training/agibot/images/agibot_034_initial.png ADDED

Git LFS Details

  • SHA256: 909c5b534d7943fa8d673a695f718d6a1cde4336c9f1a42354af5f2ead20e13d
  • Pointer size: 131 Bytes
  • Size of remote file: 215 kB
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}