tangyue0820 commited on
Commit
542b2c8
Β·
verified Β·
1 Parent(s): 5ee4fdd

Integrate reviewed Cosmos3-Edge-FP8 model card with current usage and performance

Browse files

Use the reviewed MR overview, quality benchmarks, and training/evaluation datasets; retain current HF usage examples and performance measurements. Add the PNG relationship diagram, set release date to November 9, 2026, and correct precision and section links. Exclude internal-only sections. Existing shared subcards are unchanged.

Files changed (2) hide show
  1. README.md +42 -70
  2. images/cosmos3_edge_fp8.png +0 -0
README.md CHANGED
@@ -17,47 +17,25 @@ tags:
17
 
18
  # Model Overview: Cosmos3-Edge-FP8
19
 
 
 
 
 
20
  ## Description
21
 
 
 
22
  Cosmos3 is a collection of Omnimodal world models capable of generating dynamic, high-quality video, image, audio, and action commands from combinations of text, image, video, and action trajectory inputs. It serves as a foundational building block for a broad range of Physical AI applications and research spanning world understanding, world generation, simulation, and embodied policy learning.
23
 
24
  This model is ready for commercial and non-commercial use.
25
 
26
- > **Update β€” August 25, 2026:** The Cosmos3-Edge-FP8 generator checkpoint, runtime defaults, usage examples, and benchmark results have been updated. Users pulling from `main` should refresh their local snapshot. See the [update announcement](https://huggingface.co/nvidia/Cosmos3-Edge-FP8/discussions/62) for compatibility and reproducibility details.
27
-
28
  **Model Developer:** NVIDIA
29
 
30
  ### Model Versions
31
 
32
- **Released on: 07/20/2026**
33
  - Cosmos3-Edge-FP8:
34
  - Given multimodal inputs including text, images, video, and action trajectories, generate coherent text, images, video, and action outputs for multimodal understanding, world simulation, future prediction, action reasoning, and Physical AI applications.
35
 
36
- - Cosmos3-Edge-Policy-DROID:
37
- - Given language instructions and visual observations from the DROID robot platform, generate robot action trajectories for manipulation and control tasks.
38
-
39
- - Cosmos3-Super-Image2Video-4Step:
40
- - Given one or more input images and optional text instructions, generate temporally coherent video sequences that are consistent with the provided visual content. Distilled from Cosmos3-Super-Image2Video using [Improved Distribution Matching Distillation (DMD2)](https://arxiv.org/abs/2405.14867), enabling high-quality generation in 4 steps.
41
-
42
- - Cosmos3-Super-Text2Image-4Step:
43
- - Given text input, generate high-fidelity images that are consistent with the provided description. Distilled from Cosmos3-Super-Text2Image using [Improved Distribution Matching Distillation (DMD2)](https://arxiv.org/abs/2405.14867), enabling high-quality generation in 4 steps.
44
-
45
- **Released on: 05/31/2026**
46
- - Cosmos3-Nano:
47
- - Given multimodal inputs including text, images, video, audio, and action trajectories, generate coherent text, images, video, audio, and action outputs for multimodal understanding, world simulation, future prediction, action reasoning, and Physical AI applications.
48
-
49
- - Cosmos3-Super:
50
- - Given multimodal inputs including text, images, video, audio, and action trajectories, generate coherent text, images, video, audio, and action outputs for multimodal understanding, world simulation, future prediction, action reasoning, and Physical AI applications.
51
-
52
- - Cosmos3-Nano-Policy-DROID:
53
- - Given language instructions and visual observations from the DROID robot platform, generate robot action trajectories for manipulation and control tasks.
54
-
55
- - Cosmos3-Super-Image2Video:
56
- - Given one or more input images and optional text instructions, generate temporally coherent video sequences that are consistent with the provided visual content.
57
-
58
- - Cosmos3-Super-Text2Image:
59
- - Given text input, generate high-fidelity images that are consistent with the provided description.
60
-
61
  ### License
62
 
63
  This model is released under the [OpenMDW1.1](https://openmdw.ai/license/1-1/)
@@ -72,8 +50,8 @@ Physical AI: Encompassing robotics, autonomous vehicles (AV), and smart space en
72
 
73
  ### Release Date
74
 
75
- Hugging Face 07/20/2026 via [https://huggingface.co/collections/nvidia/cosmos3](https://huggingface.co/collections/nvidia/cosmos3)
76
- GitHub 07/20/2026 via [https://github.com/nvidia/cosmos](https://github.com/nvidia/cosmos)
77
 
78
  ## Model Architecture
79
 
@@ -87,18 +65,8 @@ Cosmos3 is an Omni-modal foundation model built on a Mixture-of-Transformers (Mo
87
 
88
  **Number of trainable model parameters:**
89
 
90
- **Released on: 07/20/2026**
91
  - Cosmos3-Edge-FP8: 4B
92
- - Cosmos3-Edge-Policy-DROID: 4B
93
- - Cosmos3-Super-Image2Video-4Step: 64B
94
- - Cosmos3-Super-Text2Image-4Step: 64B
95
 
96
- **Released on: 05/31/2026**
97
- - Cosmos3-Nano: 16B
98
- - Cosmos3-Super: 64B
99
- - Cosmos3-Nano-Policy-DROID: 16B
100
- - Cosmos3-Super-Image2Video: 64B
101
- - Cosmos3-Super-Text2Image: 64B
102
 
103
  ## Input/Output Specifications
104
 
@@ -188,7 +156,7 @@ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated sys
188
 
189
  - Linux (We have not tested on other operating systems.)
190
 
191
- **Note:** Only BF16 precision is tested. Other precisions like FP4, FP8, and FP16 are not officially supported.
192
 
193
  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.
194
 
@@ -203,11 +171,11 @@ The integration of foundation and fine-tuned models into AI systems requires add
203
  - **Time period for testing data collection:** N/A (standard public benchmarks)
204
  - **Time period for validation data collection:** N/A (standard public benchmarks)
205
 
206
- 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, 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.
207
 
208
- 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, 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.
209
 
210
- - 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/_downloads/e482b7114ce8dbfbb07d2d4b42cafe4e/training-content-Cosmos-3.pdf).
211
 
212
  **Data Modality and Training Data Size**
213
 
@@ -216,6 +184,7 @@ Training datasets passed through multiple layers of automated and manual safegua
216
  | Text | 22M | Not Applicable |
217
  | Image | 19M | 767M |
218
  | Video | 1M | 348M |
 
219
  | Action | Not Applicable | 7M |
220
 
221
  **Data Collection Method by dataset**
@@ -224,9 +193,9 @@ Training datasets passed through multiple layers of automated and manual safegua
224
 
225
  **Labeling Method by dataset**
226
 
227
- - Hybrid: Human, Automated
228
 
229
- **Properties:** The training, testing, and evaluation datasets consist of diverse multimodal video, image, 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.
230
 
231
  ### Public Datasets
232
 
@@ -262,39 +231,42 @@ Training datasets passed through multiple layers of automated and manual safegua
262
 
263
  **Labeling Method by dataset**
264
 
265
- - Hybrid: Human, Automated
266
-
267
- **Properties:** The training, testing, and evaluation datasets consist of diverse multimodal video, image, 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.
268
-
269
- ## Benchmarks
270
-
271
- For detailed evaluations of the base model, see our [technical paper](https://research.nvidia.com/labs/cosmos-lab/cosmos3/technical-report.pdf).
272
-
273
- ### Overall
274
-
275
- The table below summarizes Cosmos3-Edge-FP8 across reasoning and generation. Each reasoning column (General, Robotics, Smart Infrastructure, Driving) reports the average score over that capability's benchmarks. For generation, **Image2Video** is the PAIBench overall score and **Policy: Robot** is the RoboLab success rate. In each column, the best result is in **bold** and the second-best is <u>underlined</u>.
276
- \* denotes post-trained Cosmos3 variants: [Cosmos3-Nano-Policy-DROID](https://huggingface.co/nvidia/Cosmos3-Nano-Policy-DROID) and [Cosmos3-Edge-Policy-DROID](https://huggingface.co/nvidia/Cosmos3-Edge-Policy-DROID#robolab).
277
 
278
- ![Overall benchmark results](https://huggingface.co/nvidia/Cosmos3-Edge/resolve/main/assets/benchmark-overall.png)
279
 
280
- ### Reasoning Benchmarks
281
 
282
- ![Reasoning benchmarks](https://huggingface.co/nvidia/Cosmos3-Edge/resolve/main/images/benchmark-reasoning.png)
283
 
284
- ### Generation Benchmarks
285
 
286
- #### Image-to-Video Generation
 
 
 
 
 
287
 
288
- All models are evaluated on image-to-video generation at **480p, 24 fps**. Throughput is the number of frames generated per second, measured in eager mode on a single **NVIDIA H100** GPU. Cosmos3-Edge-FP8 delivers the highest generation throughput while achieving competitive quality across PAIBench, RBench, and PhysicsIQ.
 
 
 
289
 
290
- ![Image-to-Video benchmark results](https://huggingface.co/nvidia/Cosmos3-Edge/resolve/main/assets/benchmark-image2video.png)
291
 
 
292
 
293
- #### Action
 
 
 
 
 
294
 
295
- The Edge model is a strong initialization for downstream action tasks. For example, post-training it on the DROID dataset produces a policy whose RoboLab success rate is reported in the [Cosmos3-Edge-Policy-DROID model card](https://huggingface.co/nvidia/Cosmos3-Edge-Policy-DROID#robolab).
296
 
297
- ### Performance Benchmarks
298
 
299
  The table reports single-GPU inference performance for Cosmos3-Edge-FP8 using FLASH_ATTN unless otherwise noted.
300
 
@@ -991,9 +963,9 @@ Cosmos3 outputs should not be treated as physically accurate simulation, reliabl
991
 
992
  ## Inference
993
 
994
- **Acceleration Engine:** [vLLM-Omni](https://github.com/vllm-project/vllm-omni), [vLLM](https://github.com/vllm-project/vllm), [PyTorch](https://pytorch.org/)
995
 
996
- **Test Hardware:** H100, H20, RTX PRO 6000, DGX Station, DGX Spark, Jetson Thor, Jetson AGX Orin
997
 
998
  ## Ethical Considerations
999
 
 
17
 
18
  # Model Overview: Cosmos3-Edge-FP8
19
 
20
+ ![Cosmos3-Edge base model and its FP8-quantized variant](images/cosmos3_edge_fp8.png)
21
+
22
+ The highlighted checkpoint, Cosmos3-Edge-FP8, is an FP8-quantized version of the Cosmos3-Edge base model.
23
+
24
  ## Description
25
 
26
+ Cosmos3-Edge-FP8 is an FP8-quantized variant of Cosmos3-Edge. It retains the base model architecture and intended applications, with FP8 quantization applied to selected model components.
27
+
28
  Cosmos3 is a collection of Omnimodal world models capable of generating dynamic, high-quality video, image, audio, and action commands from combinations of text, image, video, and action trajectory inputs. It serves as a foundational building block for a broad range of Physical AI applications and research spanning world understanding, world generation, simulation, and embodied policy learning.
29
 
30
  This model is ready for commercial and non-commercial use.
31
 
 
 
32
  **Model Developer:** NVIDIA
33
 
34
  ### Model Versions
35
 
 
36
  - Cosmos3-Edge-FP8:
37
  - Given multimodal inputs including text, images, video, and action trajectories, generate coherent text, images, video, and action outputs for multimodal understanding, world simulation, future prediction, action reasoning, and Physical AI applications.
38
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  ### License
40
 
41
  This model is released under the [OpenMDW1.1](https://openmdw.ai/license/1-1/)
 
50
 
51
  ### Release Date
52
 
53
+ Hugging Face 11/09/2026 via [https://huggingface.co/collections/nvidia/cosmos3](https://huggingface.co/collections/nvidia/cosmos3)
54
+ GitHub 11/09/2026 via [https://github.com/nvidia/cosmos](https://github.com/nvidia/cosmos)
55
 
56
  ## Model Architecture
57
 
 
65
 
66
  **Number of trainable model parameters:**
67
 
 
68
  - Cosmos3-Edge-FP8: 4B
 
 
 
69
 
 
 
 
 
 
 
70
 
71
  ## Input/Output Specifications
72
 
 
156
 
157
  - Linux (We have not tested on other operating systems.)
158
 
159
+ **Precision:** This repository provides the FP8-quantized Cosmos3-Edge checkpoint. Components excluded from quantization retain their original precision. See [Performance Benchmark](#performance-benchmark) for backend- and hardware-specific precision details.
160
 
161
  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.
162
 
 
171
  - **Time period for testing data collection:** N/A (standard public benchmarks)
172
  - **Time period for validation data collection:** N/A (standard public benchmarks)
173
 
174
+ 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.
175
 
176
+ 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.
177
 
178
+ 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).
179
 
180
  **Data Modality and Training Data Size**
181
 
 
184
  | Text | 22M | Not Applicable |
185
  | Image | 19M | 767M |
186
  | Video | 1M | 348M |
187
+ | Audio | Not Applicable | 139M |
188
  | Action | Not Applicable | 7M |
189
 
190
  **Data Collection Method by dataset**
 
193
 
194
  **Labeling Method by dataset**
195
 
196
+ - Hybrid: Manually-Labelled, Automated
197
 
198
+ **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.
199
 
200
  ### Public Datasets
201
 
 
231
 
232
  **Labeling Method by dataset**
233
 
234
+ - Hybrid: Manually-Labelled, Automated
 
 
 
 
 
 
 
 
 
 
 
235
 
236
+ **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.
237
 
238
+ ## Quality Benchmark
239
 
240
+ ### Reasoner Benchmarks
241
 
242
+ We compare **Cosmos3-Edge-FP8** with its BF16 counterpart across general reasoning, robotics, smart infrastructure, and driving benchmarks. Each domain score is the unweighted mean of the included benchmark scores. The benchmark list below defines the evaluation scope. Benchmarks without FP8 results are excluded.
243
 
244
+ | Domain | BF16 | FP8 |
245
+ | --- | --- | --- |
246
+ | General ↑ | 56.56 | 58.10 |
247
+ | Robotics ↑ | 49.17 | 48.68 |
248
+ | Smart Infrastructure ↑ | 52.00 | 51.95 |
249
+ | Driving ↑ | 62.50 | 62.38 |
250
 
251
+ - **General** (18 benchmarks): MMBench_DEV_EN_V11, RealWorldQA, CVBench, VideoPhy2, CausalVQA_4fps, MVPBench, CountBenchQA, AI2D_TEST, DocVQA_VAL, InfoVQA_VAL, OCRBench_v2, LogicVista, MMMU_Pro_10c, MVBench_2fps64f, BlinkSpatial, BlinkDepth, HallusionBench, IFBench
252
+ - **Robotics** (10 benchmarks): CosmosReason, RefSpatial, VSI-Bench_32frame, SparBench, MMSIBench_wo_circular, MMSIVideoBench_32frame, HealthSurgiBench, CosmosERQA, RoboSpatialHome, Where2Place
253
+ - **Smart Infrastructure** (8 benchmarks): VANTAGE_2DGrounding, VANTAGE_Astro2D, VANTAGE_2DPointing, VANTAGE_DVC, VANTAGE_EventVerification, VANTAGE_SOT, VANTAGE_Temporal, VANTAGE_VQA
254
+ - **Driving** (3 benchmarks): LingoQA, AVSpecialCollisionBench, AVSpecialStopBehaviorBench
255
 
256
+ ### Generator Benchmarks
257
 
258
+ We compare **Cosmos3-Edge-FP8** with its BF16 counterpart. PAIBench-G evaluates text-to-video and image-to-video generation. RBench assesses task correctness, physical plausibility, and visual fidelity in robotics videos. PhysicsIQ compares video continuations with recorded physical outcomes.
259
 
260
+ | Benchmark | BF16 | FP8 |
261
+ | --- | --- | --- |
262
+ | PAIBench-G I2V ↑ | 81.10 | 80.75 |
263
+ | PAIBench-G T2V ↑ | 76.90 | 76.58 |
264
+ | RBench ↑ | 55.10 | 56.52 |
265
+ | PhysicsIQ ↑ | 31.28 | 31.37 |
266
 
267
+ *For quality benchmarks, BF16 results were obtained using PyTorch. FP8 results were obtained using vLLM-Omni.*
268
 
269
+ ## Performance Benchmark
270
 
271
  The table reports single-GPU inference performance for Cosmos3-Edge-FP8 using FLASH_ATTN unless otherwise noted.
272
 
 
963
 
964
  ## Inference
965
 
966
+ **Acceleration Engine:** [vLLM-Omni](https://github.com/vllm-project/vllm-omni), [vLLM](https://github.com/vllm-project/vllm), [PyTorch](https://pytorch.org/), [Hugging Face Diffusers](https://github.com/huggingface/diffusers)
967
 
968
+ **Test Hardware:** See [Performance Benchmark](#performance-benchmark) for the evaluated GPU configurations, including emulated T3000 results and Orin weight-only W8A16 compute.
969
 
970
  ## Ethical Considerations
971
 
images/cosmos3_edge_fp8.png ADDED