Cosmos
Diffusers
Safetensors
cosmos3_omni
nvidia
cosmos3
vllm
vllm-omni
sglang
sglang-diffusion
text, image, video, audio, and action generation
omnimodel
Instructions to use nvidia/Cosmos3-Nano-FP8 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Cosmos
How to use nvidia/Cosmos3-Nano-FP8 with Cosmos:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
- Diffusers
How to use nvidia/Cosmos3-Nano-FP8 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("nvidia/Cosmos3-Nano-FP8", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Add enterprise GPU benchmarks and clarify FP8 model card
#8
by fsergiy - opened
- EXPLAINABILITY.md +2 -2
- README.md +34 -14
EXPLAINABILITY.md
CHANGED
|
@@ -5,12 +5,12 @@
|
|
| 5 |
| Intended Application & Domain | World reasoning and generation for Physical AI. |
|
| 6 |
| Model Type | Mixture-of-Transformers architecture with two towers. One is an autoregressive model for Physical AI reasoning; the other is a diffusion model for Physical AI generation. |
|
| 7 |
| Intended Users | Physical AI developers, researchers, and practitioners building or evaluating autonomous vehicle, robotics, and world-generation workflows. |
|
| 8 |
-
| Output |
|
| 9 |
| Tools used to evaluate datasets to identify synthetic data and ensure data authenticity. | Dataset provenance analysis, metadata validation, watermark and artifact detection, embedding-based clustering, heuristic quality checks, and model-assisted data validation pipelines are used to identify synthetic content patterns, assess dataset authenticity, and improve data quality during dataset curation. |
|
| 10 |
| Describe how the model works | Cosmos3 is an Omni world foundation model that generates texts, images, videos, audio, and action commands from combinations of text, images, videos, and action trajectory inputs. Input tokens from multiple modalities are packed into a shared sequence and processed by our mixture-of-transformer backbone with modality-specific output heads. |
|
| 11 |
| Name the adversely impacted groups this has been tested to deliver comparable outcomes regardless of: | None. |
|
| 12 |
| Technical Limitations | The model may not follow text, image, video, audio, or action trajectory inputs accurately in challenging cases, especially where the input contains complex scene composition, unusual camera motion, multiple interacting agents, low lighting, high motion blur, or fine-grained physical interactions. Generated outputs may contain temporal inconsistency, object morphing, inaccurate 3D structure, or implausible physical dynamics. Generated audio may not accurately render intelligible speech, or maintain strict temporal and semantic alignment with the visual context. |
|
| 13 |
| Verified to have met prescribed NVIDIA quality standards | Yes. |
|
| 14 |
-
| Performance Metrics | Video generation is measured using PAIBench-G, RBench, PhysicsIQ, and
|
| 15 |
| Potential Known Risks | This model can generate synthetic media and may produce content that is offensive, unsafe, misleading, indecent, or unsuitable for a target deployment. Users should implement robust safety guardrails — including content filtering, abuse monitoring, and access controls — to reduce the risk of harmful outputs. Users are responsible for ensuring that their use of the model complies with all applicable laws and regulations, and for regularly reviewing and updating their guardrails as risks evolve. |
|
| 16 |
| Licensing | [OpenMDW1.1](https://openmdw.ai/) |
|
|
|
|
| 5 |
| Intended Application & Domain | World reasoning and generation for Physical AI. |
|
| 6 |
| Model Type | Mixture-of-Transformers architecture with two towers. One is an autoregressive model for Physical AI reasoning; the other is a diffusion model for Physical AI generation. |
|
| 7 |
| Intended Users | Physical AI developers, researchers, and practitioners building or evaluating autonomous vehicle, robotics, and world-generation workflows. |
|
| 8 |
+
| Output | Text, images, videos, audio, and action commands. |
|
| 9 |
| Tools used to evaluate datasets to identify synthetic data and ensure data authenticity. | Dataset provenance analysis, metadata validation, watermark and artifact detection, embedding-based clustering, heuristic quality checks, and model-assisted data validation pipelines are used to identify synthetic content patterns, assess dataset authenticity, and improve data quality during dataset curation. |
|
| 10 |
| Describe how the model works | Cosmos3 is an Omni world foundation model that generates texts, images, videos, audio, and action commands from combinations of text, images, videos, and action trajectory inputs. Input tokens from multiple modalities are packed into a shared sequence and processed by our mixture-of-transformer backbone with modality-specific output heads. |
|
| 11 |
| Name the adversely impacted groups this has been tested to deliver comparable outcomes regardless of: | None. |
|
| 12 |
| Technical Limitations | The model may not follow text, image, video, audio, or action trajectory inputs accurately in challenging cases, especially where the input contains complex scene composition, unusual camera motion, multiple interacting agents, low lighting, high motion blur, or fine-grained physical interactions. Generated outputs may contain temporal inconsistency, object morphing, inaccurate 3D structure, or implausible physical dynamics. Generated audio may not accurately render intelligible speech, or maintain strict temporal and semantic alignment with the visual context. |
|
| 13 |
| Verified to have met prescribed NVIDIA quality standards | Yes. |
|
| 14 |
+
| Performance Metrics | Video generation is measured using PAIBench-G, RBench, PhysicsIQ, and Artificial Analysis Image2Video benchmark. Image generation uses UniGenBench and Artificial Analysis Text2Image benchmark. For transfer evaluation, we use PAIBench-C and AVBench-C. Audio generation uses internal benchmarks. Action prediction uses metrics such as action MSE, Absolute Translation Error, Relative Translation Error, Relative Rotation Error, PSNR, and robotic task completion success rate. |
|
| 15 |
| Potential Known Risks | This model can generate synthetic media and may produce content that is offensive, unsafe, misleading, indecent, or unsuitable for a target deployment. Users should implement robust safety guardrails — including content filtering, abuse monitoring, and access controls — to reduce the risk of harmful outputs. Users are responsible for ensuring that their use of the model complies with all applicable laws and regulations, and for regularly reviewing and updating their guardrails as risks evolve. |
|
| 16 |
| Licensing | [OpenMDW1.1](https://openmdw.ai/) |
|
README.md
CHANGED
|
@@ -91,13 +91,13 @@ Cosmos3 is an Omni-modal foundation model built on a Mixture-of-Transformers (Mo
|
|
| 91 |
- Text: String
|
| 92 |
- Image: jpg, png, jpeg, webp
|
| 93 |
- Video (with or without audio): mp4
|
| 94 |
-
- Action:
|
| 95 |
- **Input Parameters**:
|
| 96 |
- Text: One-dimensional (1D)
|
| 97 |
- Image: Two-dimensional (2D)
|
| 98 |
- Video: Three-dimensional (3D)
|
| 99 |
- Audio: One-dimensional (1D)
|
| 100 |
-
- Action trajectory:
|
| 101 |
- **Other Properties Related to Input**:
|
| 102 |
- For video inputs, we accept various resolutions, including 720p, 480p, and 256p.
|
| 103 |
- When using input video with audio muxed into the video MP4 file, the audio should have 2 channels (stereo) and a 48 kHz sample rate.
|
|
@@ -116,17 +116,17 @@ Cosmos3 is an Omni-modal foundation model built on a Mixture-of-Transformers (Mo
|
|
| 116 |
- Image: JPG
|
| 117 |
- Video: MP4
|
| 118 |
- Audio: Advanced Audio Coding (AAC) stream (muxed within the MP4)
|
| 119 |
-
- Action:
|
| 120 |
- Text: string
|
| 121 |
- **Output Parameters**:
|
| 122 |
- Image: Two-dimensional (2D)
|
| 123 |
- Video: Three-dimensional (3D)
|
| 124 |
- Audio: One-dimensional (1D)
|
| 125 |
-
- Action:
|
| 126 |
- Text: One-dimensional (1D)
|
| 127 |
- **Other Properties Related to Output**:
|
| 128 |
- The generated video is an MP4 file, with the resolution, frame rate, and duration specified in the input. The generated audio is encoded in AAC format, muxed into the video MP4 file with 2 channels (stereo) and a 48 kHz sample rate.
|
| 129 |
-
-
|
| 130 |
- The generated action is only supported for compatible embodiments, including general camera motion (9D), autonomous vehicle (9D), egocentric motion (57D), single Franka Panda arm with RobotiQ gripper (10D), dual Franka Panda arm with RobotiQ gripper (20D), Agibot (29D), UR (10D), Google robot (10D), WidowX 250 (10D), UMI (9D).
|
| 131 |
- Audio: 48 kHz stereo AAC stream muxed into video mp4
|
| 132 |
- Video: mp4 at the FPS specified in input
|
|
@@ -166,7 +166,7 @@ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated sys
|
|
| 166 |
|
| 167 |
**Runtime Engine(s):**
|
| 168 |
|
| 169 |
-
- [PyTorch](https://github.com/nvidia/
|
| 170 |
- [vLLM-Omni](https://github.com/vllm-project/vllm-omni)
|
| 171 |
- [Hugging Face Diffusers](https://huggingface.co/docs/diffusers/en/index)
|
| 172 |
- [SGLang](https://github.com/sgl-project/sglang)
|
|
@@ -183,6 +183,8 @@ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated sys
|
|
| 183 |
|
| 184 |
**Precision:** This repository provides the FP8-quantized Cosmos3-Nano checkpoint, including its quantization configuration and ModelOpt state. Components excluded from quantization retain their original precision.
|
| 185 |
|
|
|
|
|
|
|
| 186 |
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.
|
| 187 |
|
| 188 |
## Training, Testing, and Evaluation Datasets
|
|
@@ -264,6 +266,8 @@ Training datasets passed through multiple layers of automated and manual safegua
|
|
| 264 |
|
| 265 |
Please see our [technical paper](https://research.nvidia.com/labs/cosmos-lab/cosmos3/technical-report.pdf) for detailed evaluations of the base model.
|
| 266 |
|
|
|
|
|
|
|
| 267 |
### Overall
|
| 268 |
|
| 269 |

|
|
@@ -296,16 +300,32 @@ All results use one GPU. Generator inference uses vLLM-Omni, Reasoner inference
|
|
| 296 |
| H100 | 3.20 | 1.01 | 129.5 | 173.01 | 38.29 |
|
| 297 |
| H200 | 3.34 | 1.00 | 128.3 | 202.05 | 38.97 |
|
| 298 |
| H20 | 0.91 | 0.24 | 242.0 | 148.21 | 11.17 |
|
| 299 |
-
| DGX Spark |
|
| 300 |
-
| DGX Station |
|
|
|
|
| 301 |
| RTX PRO 6000 Blackwell Server Edition | 2.34 | 0.61 | 150.2 | 112.90 | 24.90 |
|
| 302 |
|
| 303 |
- **Generator:** Image-to-video (i2v) only, at 480p and 720p. Throughput is calculated as **189 frames ÷ average end-to-end generation time in seconds**, following the existing model-card benchmark convention. This measures inference throughput for a generated sequence.
|
| 304 |
-
- **
|
|
|
|
| 305 |
- **Action Policy:** Policy Robot (DROID), 480p, with 32-action chunks. Action throughput is calculated as **32 ÷ average generation time in seconds**.
|
| 306 |
- **GB300 Action Policy estimate (*):** Uses the single-GPU B300 result, assuming similar performance on GB300. This value has not been measured on GB300.
|
| 307 |
-
- **Hardware:** H100 results use the 80GB HBM3 configuration; H200 results use the 141GB
|
| 308 |
-
- **Missing results:** A dash (—) indicates that
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 309 |
|
| 310 |
## Usage
|
| 311 |
|
|
@@ -369,7 +389,7 @@ The example inputs (`assets/`) live in this model repo. Download just this folde
|
|
| 369 |
|
| 370 |
```bash
|
| 371 |
pip install -U "huggingface_hub[cli]"
|
| 372 |
-
hf download nvidia/Cosmos3-Nano-FP8 assets/ --local-dir Cosmos3-Nano-FP8
|
| 373 |
cd Cosmos3-Nano-FP8
|
| 374 |
```
|
| 375 |
|
|
@@ -1005,7 +1025,7 @@ curl -sS -L "http://localhost:30000/v1/videos/${job_id}/content" \
|
|
| 1005 |
-o cosmos3_t2v_output.mp4
|
| 1006 |
```
|
| 1007 |
|
| 1008 |
-
SGLang accepts Cosmos 3 request options including `max_sequence_length`, `flow_shift`, `extra_params.guardrails`, `extra_params.use_resolution_template`, and `extra_params.use_duration_template`.
|
| 1009 |
|
| 1010 |
For complete serving instructions and request examples, see the [Cosmos3 SGLang cookbook](https://docs.sglang.io/cookbook/diffusion/Cosmos/Cosmos3).
|
| 1011 |
|
|
@@ -1021,7 +1041,7 @@ Cosmos3 outputs should not be treated as physically accurate simulation, reliabl
|
|
| 1021 |
|
| 1022 |
**Test Hardware:** GB300, B300, B200, H100, H200, H20, DGX Spark, DGX Station, RTX PRO 6000 Blackwell Server Edition, RTX PRO 5000 72GB Blackwell
|
| 1023 |
|
| 1024 |
-
See [Performance Benchmarking](#performance-benchmarking) for per-hardware results
|
| 1025 |
|
| 1026 |
## Ethical Considerations
|
| 1027 |
|
|
|
|
| 91 |
- Text: String
|
| 92 |
- Image: jpg, png, jpeg, webp
|
| 93 |
- Video (with or without audio): mp4
|
| 94 |
+
- Action: JSON array shaped (T, D)
|
| 95 |
- **Input Parameters**:
|
| 96 |
- Text: One-dimensional (1D)
|
| 97 |
- Image: Two-dimensional (2D)
|
| 98 |
- Video: Three-dimensional (3D)
|
| 99 |
- Audio: One-dimensional (1D)
|
| 100 |
+
- Action trajectory: Two-dimensional (T, D)
|
| 101 |
- **Other Properties Related to Input**:
|
| 102 |
- For video inputs, we accept various resolutions, including 720p, 480p, and 256p.
|
| 103 |
- When using input video with audio muxed into the video MP4 file, the audio should have 2 channels (stereo) and a 48 kHz sample rate.
|
|
|
|
| 116 |
- Image: JPG
|
| 117 |
- Video: MP4
|
| 118 |
- Audio: Advanced Audio Coding (AAC) stream (muxed within the MP4)
|
| 119 |
+
- Action: JSON object containing action data and shape metadata
|
| 120 |
- Text: string
|
| 121 |
- **Output Parameters**:
|
| 122 |
- Image: Two-dimensional (2D)
|
| 123 |
- Video: Three-dimensional (3D)
|
| 124 |
- Audio: One-dimensional (1D)
|
| 125 |
+
- Action: Two-dimensional (T, D)
|
| 126 |
- Text: One-dimensional (1D)
|
| 127 |
- **Other Properties Related to Output**:
|
| 128 |
- The generated video is an MP4 file, with the resolution, frame rate, and duration specified in the input. The generated audio is encoded in AAC format, muxed into the video MP4 file with 2 channels (stereo) and a 48 kHz sample rate.
|
| 129 |
+
- The technical paper lists video generation from 5 to 400 frames at 256p and 480p, and from 5 to 300 frames at 720p, at 10–30 fps. The examples below request 189 frames at 24 fps. Supported request sizes also depend on the inference backend.
|
| 130 |
- The generated action is only supported for compatible embodiments, including general camera motion (9D), autonomous vehicle (9D), egocentric motion (57D), single Franka Panda arm with RobotiQ gripper (10D), dual Franka Panda arm with RobotiQ gripper (20D), Agibot (29D), UR (10D), Google robot (10D), WidowX 250 (10D), UMI (9D).
|
| 131 |
- Audio: 48 kHz stereo AAC stream muxed into video mp4
|
| 132 |
- Video: mp4 at the FPS specified in input
|
|
|
|
| 166 |
|
| 167 |
**Runtime Engine(s):**
|
| 168 |
|
| 169 |
+
- [PyTorch / Cosmos Framework](https://github.com/nvidia/cosmos-framework)
|
| 170 |
- [vLLM-Omni](https://github.com/vllm-project/vllm-omni)
|
| 171 |
- [Hugging Face Diffusers](https://huggingface.co/docs/diffusers/en/index)
|
| 172 |
- [SGLang](https://github.com/sgl-project/sglang)
|
|
|
|
| 183 |
|
| 184 |
**Precision:** This repository provides the FP8-quantized Cosmos3-Nano checkpoint, including its quantization configuration and ModelOpt state. Components excluded from quantization retain their original precision.
|
| 185 |
|
| 186 |
+
The Diffusers transformer configuration also specifies A16 activations for the first three and last three denoising iterations and for the Reasoner. Runtime support for this policy depends on the inference backend; FP8 does not imply that every component and operation runs in FP8.
|
| 187 |
+
|
| 188 |
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.
|
| 189 |
|
| 190 |
## Training, Testing, and Evaluation Datasets
|
|
|
|
| 266 |
|
| 267 |
Please see our [technical paper](https://research.nvidia.com/labs/cosmos-lab/cosmos3/technical-report.pdf) for detailed evaluations of the base model.
|
| 268 |
|
| 269 |
+
The figures below report base-model quality evaluations. They do not establish the quality change caused by FP8 quantization of this checkpoint.
|
| 270 |
+
|
| 271 |
### Overall
|
| 272 |
|
| 273 |

|
|
|
|
| 300 |
| H100 | 3.20 | 1.01 | 129.5 | 173.01 | 38.29 |
|
| 301 |
| H200 | 3.34 | 1.00 | 128.3 | 202.05 | 38.97 |
|
| 302 |
| H20 | 0.91 | 0.24 | 242.0 | 148.21 | 11.17 |
|
| 303 |
+
| DGX Spark | 0.39† | 0.11† | 336.1 | 22.62 | — |
|
| 304 |
+
| DGX Station | 0.68† | 0.38† | 85.4 | 255.99 | — |
|
| 305 |
+
| RTX PRO 5000 72GB Blackwell | 1.26† | 0.32† | — | — | — |
|
| 306 |
| RTX PRO 6000 Blackwell Server Edition | 2.34 | 0.61 | 150.2 | 112.90 | 24.90 |
|
| 307 |
|
| 308 |
- **Generator:** Image-to-video (i2v) only, at 480p and 720p. Throughput is calculated as **189 frames ÷ average end-to-end generation time in seconds**, following the existing model-card benchmark convention. This measures inference throughput for a generated sequence.
|
| 309 |
+
- **Enterprise Generator (†):** Calculated from the measured times below using an assumed workload of 189 frames, as specified for this model-card update. The inspected enterprise report does not independently confirm the frame count or denoising-step count.
|
| 310 |
+
- **Reasoner:** 50 input text tokens plus a 30-second 374×374 video sampled at 1 fps, 100 output tokens, concurrency 1, and 50 requests. Both columns use the same workload. Output token throughput is the reported benchmark metric, including request processing time; it is not decode-only token speed.
|
| 311 |
- **Action Policy:** Policy Robot (DROID), 480p, with 32-action chunks. Action throughput is calculated as **32 ÷ average generation time in seconds**.
|
| 312 |
- **GB300 Action Policy estimate (*):** Uses the single-GPU B300 result, assuming similar performance on GB300. This value has not been measured on GB300.
|
| 313 |
+
- **Hardware:** H100 results use the SXM 80GB HBM3 configuration; H200 results use the SXM 141GB HBM3e configuration. RTX PRO 6000 results use the Blackwell Server Edition.
|
| 314 |
+
- **Missing results:** A dash (—) indicates that a verified value for that metric is not included here. The enterprise forward/inverse dynamics results are different workloads from DROID Action Policy.
|
| 315 |
+
|
| 316 |
+
#### Enterprise Generator measurements
|
| 317 |
+
|
| 318 |
+
The enterprise vLLM-Omni report (PBR 309485), published November 5, 2026 Pacific time, provides the following single-GPU FP8 image-to-video generation times. Values are the reported average generation times, without conversion to frames/s.
|
| 319 |
+
|
| 320 |
+
| Platform (1 GPU) | Generator — 480p (seconds) ↓ | Generator — 720p (seconds) ↓ |
|
| 321 |
+
| --- | ---: | ---: |
|
| 322 |
+
| DGX Spark | 490.73 | 1669.62 |
|
| 323 |
+
| DGX Station | 276.93 | 494.01 |
|
| 324 |
+
| RTX PRO 5000 72GB Blackwell | 150.33 | 585.58 |
|
| 325 |
+
|
| 326 |
+
DGX Spark and RTX PRO 5000 use CUDNN_ATTN; DGX Station uses Flash Attention. The report flags thermal slowdown on many DGX Spark tests and one DGX Station test. The frames/s table uses 189 frames for these workloads by assumption. The denoising-step count remains unverified, limiting direct comparisons with other results.
|
| 327 |
+
|
| 328 |
+
Benchmark sources (NVIDIA network access required): [Generator, PBR 309431](https://prism.nvidia.com/ai-sim-platform/reports/40210), [datacenter Reasoner, PBR 308912](https://prism.nvidia.com/ai-sim-platform/reports/39455), [GB300 Reasoner, PBR 309589](https://prism.nvidia.com/ai-sim-platform/reports/40565), [enterprise Reasoner, PBR 308916](https://prism.nvidia.com/ai-sim-platform/reports/39470), and [enterprise Generator, PBR 309485](https://prism.nvidia.com/ai-sim-platform/reports/40954).
|
| 329 |
|
| 330 |
## Usage
|
| 331 |
|
|
|
|
| 389 |
|
| 390 |
```bash
|
| 391 |
pip install -U "huggingface_hub[cli]"
|
| 392 |
+
hf download nvidia/Cosmos3-Nano-FP8 --include "assets/*" --local-dir Cosmos3-Nano-FP8
|
| 393 |
cd Cosmos3-Nano-FP8
|
| 394 |
```
|
| 395 |
|
|
|
|
| 1025 |
-o cosmos3_t2v_output.mp4
|
| 1026 |
```
|
| 1027 |
|
| 1028 |
+
SGLang accepts Cosmos 3 request options including `max_sequence_length`, `flow_shift`, `extra_params.guardrails`, `extra_params.use_resolution_template`, and `extra_params.use_duration_template`. Modality support depends on the checkpoint and SGLang version. The current cookbook includes video-to-video, joint sound, and action workflows for compatible checkpoints; verify FP8 checkpoint support separately.
|
| 1029 |
|
| 1030 |
For complete serving instructions and request examples, see the [Cosmos3 SGLang cookbook](https://docs.sglang.io/cookbook/diffusion/Cosmos/Cosmos3).
|
| 1031 |
|
|
|
|
| 1041 |
|
| 1042 |
**Test Hardware:** GB300, B300, B200, H100, H200, H20, DGX Spark, DGX Station, RTX PRO 6000 Blackwell Server Edition, RTX PRO 5000 72GB Blackwell
|
| 1043 |
|
| 1044 |
+
See [Performance Benchmarking](#performance-benchmarking) for per-hardware results, measurement units, and the meaning of missing values.
|
| 1045 |
|
| 1046 |
## Ethical Considerations
|
| 1047 |
|