Add pipeline tag, library name, and paper link

#6
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +26 -442
README.md CHANGED
@@ -1,447 +1,42 @@
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
- - vllm-omni
12
- - diffusers
13
- - image-to-video
14
- - video-generation
15
  ---
16
 
17
  # **Cosmos 3: Omnimodal World Models for Physical AI**
18
- **[Model Collection](https://huggingface.co/collections/nvidia/cosmos3)** | **[Code](https://github.com/nvidia/cosmos)** | **[White Paper](https://research.nvidia.com/labs/cosmos-lab/cosmos3/technical-report.pdf)** | **[Website](https://research.nvidia.com/labs/cosmos-lab/cosmos3/)**
19
 
20
- [NVIDIA Cosmos™](https://github.com/nvidia/cosmos) is a world foundation model platform designed to accelerate the development of Physical AI by enabling machines to understand, simulate, and interact with the physical world across robotics, autonomous driving, and smart space environments, including industrial and factory-scale applications.
21
 
22
- # Model Overview: Cosmos3-Super-Image2Video
23
 
24
- ## Description
25
 
26
- 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.
27
 
28
- This model is ready for commercial and non-commercial use.
29
 
30
- **Model Developer:** NVIDIA
31
-
32
- ### Model Versions
33
- - Cosmos3-Nano:
34
- - 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.
35
-
36
- - Cosmos3-Super:
37
- - 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.
38
-
39
- - Cosmos3-Nano-Policy-DROID:
40
- - Given language instructions and visual observations from the DROID robot platform, generate robot action trajectories for manipulation and control tasks.
41
-
42
- - Cosmos3-Super-Image2Video:
43
- - Given one input image and text instructions, generate temporally coherent video sequences that are consistent with the provided visual content.
44
-
45
- - Cosmos3-Super-Text2Image:
46
- - Given text input, generate high-fidelity images that are consistent with the provided description.
47
-
48
- ### License
49
-
50
- This model is released under the [OpenMDW1.1](https://openmdw.ai/license/1-1/)
51
-
52
- ### Deployment Geography
53
-
54
- Global
55
-
56
- ### Use Case
57
-
58
- Physical AI: Encompassing robotics, autonomous vehicles (AV), and smart space environments, including industrial and factory-scale applications.
59
-
60
- ### Release Date
61
-
62
- Hugging Face 05/31/2026 via [https://huggingface.co/collections/nvidia/cosmos3](https://huggingface.co/collections/nvidia/cosmos3)
63
- GitHub 05/31/2026 via [https://github.com/nvidia/cosmos](https://github.com/nvidia/cosmos)
64
-
65
- ## Model Architecture
66
-
67
- **Architecture Type:** Transformer
68
-
69
- **Network Architecture:** Mixture-of-Transformers (MoT)
70
-
71
- Cosmos3 is an Omni-modal foundation model built on a Mixture-of-Transformers (MoT) architecture consisting of two complementary transformer towers: an autoregressive transformer for discrete token generation and a diffusion transformer for continuous multimodal generation. During inference, text is generated through standard next-token autoregressive decoding, while non-text modalities, such as images, video, audio, and actions, are synthesized through iterative denoising. This unified architecture enables Cosmos3 to model heterogeneous modalities within a single framework while preserving generation mechanisms best suited to each modality.
72
-
73
- **This model was developed based on:** [Cosmos Framework](https://github.com/nvidia/cosmos-framework)
74
-
75
- **Number of trainable model parameters:**
76
-
77
- - Cosmos3-Nano: 16B
78
- - Cosmos3-Super: 64B
79
- - Cosmos3-Nano-Policy-DROID: 16B
80
- - Cosmos3-Super-Image2Video: 64B
81
- - Cosmos3-Super-Text2Image: 64B
82
-
83
- ## Input/Output Specifications
84
-
85
- - **Generator Input**
86
- - **Input Type(s)**: Text, Image, Video (with audio or without audio), Action Trajectory
87
- - **Input Format(s)**:
88
- - Text: String
89
- - Image: jpg, png, jpeg, webp
90
- - Video (with or without audio): mp4
91
- - Action: json (1D list)
92
- - **Input Parameters**:
93
- - Text: One-dimensional (1D)
94
- - Image: Two-dimensional (2D)
95
- - Video: Three-dimensional (3D)
96
- - Audio: One-dimensional (1D)
97
- - Action trajectory: One-dimensional (1D)
98
- - **Other Properties Related to Input**:
99
- - For video inputs, we accept various resolutions, including 720p, 480p, and 256p.
100
- - 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.
101
- - Image and video inputs are RGB color (8 bits per channel, sRGB color space); grayscale inputs are not supported.
102
- - Action input is a per-frame sequence of robot/agent state or control values (e.g., joint positions, gripper state, camera pose). The full input is a 2D array shaped (T, D), where T is the number of frames and D is the embodiment-specific dimensionality listed below.
103
- - Input 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).
104
- - **Input Size and Length limits:**
105
- - **Text:** 4096 tokens
106
- - **Image:** 256p, 480p, and 720p resolution at one of these aspect ratios (16:9, 4:3, 1:1, 3:4, 9:16)
107
- - **Video:** 256p, 480p, and 720p resolution at one of these aspect ratios (16:9, 4:3, 1:1, 3:4, 9:16). Max number of frames = 5.
108
- - **Audio:** Max 0.5 second
109
- - **Action:** 16 – 400 video frames
110
- - **Generator Output**
111
- - **Output Type(s)**: Image, video, audio, action, text
112
- - **Output Format(s)**:
113
- - Image: JPG
114
- - Video: MP4
115
- - Audio: Advanced Audio Coding (AAC) stream (muxed within the MP4)
116
- - Action: 1D list (.json)
117
- - Text: string
118
- - **Output Parameters**:
119
- - Image: Two-dimensional (2D)
120
- - Video: Three-dimensional (3D)
121
- - Audio: One-dimensional (1D)
122
- - Action: One-dimensional (1D)
123
- - Text: One-dimensional (1D)
124
- - **Other Properties Related to Output**:
125
- - 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.
126
- - Video generation supports durations from 5 to 400 frames, with 189 frames as the default generation duration.
127
- - 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).
128
- - Audio: 48 kHz stereo AAC stream muxed into video mp4
129
- - Video: mp4 at the FPS specified in input
130
- - Image: JPEG
131
- - **Reasoner Input**
132
- - **Input Type(s)**: Text, Text+Image, Text+Video
133
- - **Input Format(s)**:
134
- - Text: String
135
- - Image: jpg, png, jpeg, webp
136
- - Video: mp4
137
- - **Input Parameters**:
138
- - Text: One-dimensional (1D)
139
- - Image: Two-dimensional (2D)
140
- - Video: Three-dimensional (3D)
141
- - **Other Properties Related to Input**:
142
- - Video inputs are recommended at a frame rate of 4 fps.
143
- - Long-context inputs supported up to 256K tokens.
144
- - **Input Size and Length limits:**
145
- - **Text:** Up to 256K tokens (context window).
146
- - **Image:** Standard input image formats; passed as file or URL.
147
- - **Video:** mp4 at the recommended 4 fps.
148
- - **Reasoner Output**
149
- - **Output Type(s)**: Text
150
- - **Output Format(s)**:
151
- - Text: string
152
- - **Output Parameters**:
153
- - Text: One-dimensional (1D)
154
- - **Other Properties Related to Output**:
155
- - Default `max_tokens=4096+` is recommended for reasoning outputs; longer outputs may be requested.
156
- - Reasoning outputs may include structured chain-of-thought, 2D/3D point localization, and bounding-box coordinates for vision-based tasks.
157
-
158
- The video content visualizes the input text description as a short animated scene, capturing key elements within the specified time constraints.
159
-
160
- 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.
161
-
162
- ## Software Integration
163
-
164
- **Runtime Engine(s):**
165
-
166
- - [PyTorch](https://github.com/nvidia/cosmos3)
167
- - [vLLM-Omni](https://github.com/vllm-project/vllm-omni)
168
- - [Hugging Face Diffusers](https://huggingface.co/docs/diffusers/en/index)
169
-
170
- **Supported Hardware Microarchitecture Compatibility:**
171
-
172
- - NVIDIA Ampere
173
- - NVIDIA Blackwell
174
- - NVIDIA Hopper
175
-
176
- **Operating System(s):**
177
-
178
- - Linux (We have not tested on other operating systems.)
179
-
180
- **Note:** Only BF16 precision is tested. Other precisions like FP4, FP8, and FP16 are not officially supported.
181
-
182
- 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.
183
-
184
- ## Training, Testing, and Evaluation Datasets
185
-
186
- ### Dataset Overview
187
-
188
- - **Total Size:** 1.3B data points
189
- - **Total Number of Datasets:** 393 dataset entries
190
- - **Dataset partition:** Training [100%], Testing [N/A — evaluation benchmarks used separately], Validation [N/A — evaluation benchmarks used separately]
191
- - **Time period for training data collection:** 2024–2026
192
- - **Time period for testing data collection:** N/A (standard public benchmarks)
193
- - **Time period for validation data collection:** N/A (standard public benchmarks)
194
-
195
- 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.
196
-
197
- 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.
198
-
199
- **Data Modality and Training Data Size**
200
-
201
- | Modality | Reasoning Data Sample Count | Generation Data Sample Count |
202
- | -------- | ------------------- | -------------------- |
203
- | Text | 22M | Not Applicable |
204
- | Image | 19M | 767M |
205
- | Video | 1M | 348M |
206
- | Audio | Not Applicable | 139M |
207
- | Action | Not Applicable | 8M |
208
-
209
- **Data Collection Method by dataset**
210
-
211
- - Hybrid: Automatic/Sensors, Synthetic, Automated
212
-
213
- **Labeling Method by dataset**
214
-
215
- - Hybrid: Human, Automated
216
-
217
- **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.
218
-
219
- ### Public Datasets
220
-
221
- | Dataset                                                             | Samples           |
222
- |---|---|
223
- | OpenImage | 1.2M |
224
- | Coyo700M | 100M |
225
- | YouTube Video | 340M |
226
- | UMI | 4.5M |
227
-
228
- ### Private Datasets
229
-
230
- | Dataset                                                             | Samples           |
231
- |---|---|
232
- | Egocentric | 7M |
233
- | Nexar | 0.6M |
234
- | AgiBot | 0.2M |
235
- | HOI | 0.3M |
236
-
237
- ### Synthetic Datasets
238
-
239
- | Dataset | Samples |
240
- |---|---|
241
- | synthetic images generated using HiDream-I1 | 15M |
242
- | synthetic images generated using Qwen-Image-2512 | 14M |
243
- | synthetic captions generated using Qwen3-VL | 1115M |
244
-
245
- ## Evaluation Datasets
246
-
247
- **Data Collection Method by dataset**
248
-
249
- - Hybrid: Automatic/Sensors, Synthetic, Automated
250
-
251
- **Labeling Method by dataset**
252
-
253
- - Hybrid: Human, Automated
254
-
255
- **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.
256
-
257
- ## Benchmarks
258
-
259
- Please see our [technical paper](https://research.nvidia.com/labs/cosmos-lab/cosmos3/technical-report.pdf) for detailed evaluations of the base model.
260
-
261
- ### Artificial Analysis Leaderboard
262
-
263
- #### Open-Source Models [2026/05/28/]
264
-
265
- ![Artificial Analysis Image-to-Video leaderboard (no audio) — open-source models](assets/benchmark-image2video-leaderboard.png)
266
-
267
- #### All Models [2026/05/28/] (Including Closed-Source)
268
-
269
- ![Artificial Analysis Image-to-Video leaderboard (no audio) — all models including closed-source](assets/benchmark-image2video-leaderboard-all-models.png)
270
-
271
- ## Usage
272
-
273
- - See [Cosmos](https://github.com/nvidia/cosmos) for details.
274
-
275
- ### Prompt upsampling
276
-
277
- For optimal quality, text prompts should be upsampled into a specific JSON structure. Description and code can be found [here](https://github.com/nvidia/cosmos-framework/blob/main/docs/prompt_upsampling.md).
278
-
279
- For example, for prompt upsampling using Opus-4.7:
280
-
281
- ```bash
282
- git clone https://github.com/NVIDIA/cosmos-framework.git packages/cosmos-framework
283
- pip install -e packages/cosmos-framework
284
-
285
- export PROMPT_UPSAMPLER_ENDPOINT_URL="https://api.anthropic.com/v1/"
286
- export PROMPT_UPSAMPLER_MODEL_NAME="claude-opus-4-7"
287
- export PROMPT_UPSAMPLER_API_TOKEN="<you_token>"
288
-
289
- python -m cosmos_framework.inference.prompt_upsampling \
290
- --input assets/example_original_prompt.txt \
291
- --image-url assets/example_first_frame.png \
292
- --output /tmp/upsampled_posttrain_i2v/ \
293
- --mode posttrain_image2video \
294
- --endpoint-url "${PROMPT_UPSAMPLER_ENDPOINT_URL}" \
295
- --model "${PROMPT_UPSAMPLER_MODEL_NAME}" \
296
- --api-token "${PROMPT_UPSAMPLER_API_TOKEN}" \
297
- --resolution 480 \
298
- --aspect-ratio "16,9" \
299
- --duration 7s
300
- ```
301
-
302
- The JSON-upsampled version of `assets/example_original_prompt.txt` is saved in `assets/example_prompt.json` for convenience, and is used for the video generation examples below.
303
-
304
- ### vLLM-Omni
305
-
306
- #### Container
307
-
308
- ```
309
- docker pull vllm/vllm-omni:cosmos3
310
- ```
311
-
312
- #### General Invocation
313
-
314
- You can use the release-tested `vllm-omni` package for deploying an OpenAI-compatible API inference endpoint.
315
- The recommended vLLM-Omni serving configuration for nvidia/Cosmos3-Super-Image2Video on 8xH200, 8xH100, or 8xA100 is:
316
-
317
- ```bash
318
- vllm serve nvidia/Cosmos3-Super-Image2Video \
319
- --omni \
320
- --host 0.0.0.0 \
321
- --port 8000 \
322
- --cfg-parallel-size 2 \
323
- --ulysses-degree 4 \
324
- --use-hsdp \
325
- --hsdp-shard-size 8 \
326
- --init-timeout 1800
327
- ```
328
-
329
- With this configuration, video generation with 50 steps should take approximately 55 seconds on H200 GPUs. For 2xH200, one can simply use `--cfg-parallel-size 2 --use-hsdp --hsdp-shard-size 2`, and a video should take approximately 3 minutes to generate. Tensor parallelism is also supported by setting `--tensor-parallel-size`. Setting `--enable-layerwise-offload` can help reduce memory usage on GPUs with less available memory.
330
-
331
- #### Download example prompts and scripts
332
-
333
- The inference scripts (`scripts/`) and example inputs (`assets/`) live in this model repo. Download just those folders with the Hugging Face CLI:
334
-
335
- ```bash
336
- pip install -U "huggingface_hub[cli]"
337
- hf download nvidia/Cosmos3-Super-Image2Video scripts/ assets/ \
338
- --local-dir Cosmos3-Super-Image2Video
339
- cd Cosmos3-Super-Image2Video
340
- ```
341
-
342
- Run all commands below from the downloaded repo root.
343
-
344
- #### Example: image to video generation
345
-
346
- Generate a video from a first-frame image and a JSON-format prompt by calling a vLLM-Omni endpoint:
347
-
348
- ```bash
349
- python scripts/gen_video.py \
350
- --endpoint <endpoint-url> \
351
- --prompt-file assets/example_prompt.json \
352
- --image-path assets/example_first_frame.png \
353
- --output-path scripts/output.mp4
354
- ```
355
-
356
- Or, as a minimal standalone script:
357
-
358
- ```python
359
- import json
360
- import requests
361
-
362
- # 1. Read JSON-upsampled prompt (prompt + negative_prompt)
363
- json_prompt = json.load(open("assets/example_prompt.json"))
364
-
365
- # 2. Build your API payload
366
- payload = {
367
- "prompt": json_prompt["prompt"],
368
- "negative_prompt": json_prompt["negative_prompt"],
369
- "size": "832x480",
370
- "num_frames": 189,
371
- "fps": 24,
372
- "num_inference_steps": 50,
373
- "guidance_scale": 6.0,
374
- "flow_shift": 5.0,
375
- "extra_params": json.dumps(
376
- {
377
- "use_resolution_template": False,
378
- "use_duration_template": False,
379
- "guardrails": True,
380
- }
381
- ),
382
- }
383
- files = {"input_reference": ("input.png", open("assets/example_first_frame.png", "rb"), "image/png")}
384
-
385
- # 3. Send the POST request
386
- url = "http://localhost:8000/v1/videos/sync"
387
- print("Sending request to server...")
388
- response = requests.post(url, data=payload, files=files, headers={"Accept": "video/mp4"})
389
- response.raise_for_status()
390
-
391
- # 4. Save the returned MP4 bytes
392
- with open("/tmp/cosmos3_i2v.mp4", "wb") as video_file:
393
- video_file.write(response.content)
394
- print("Saved video to /tmp/cosmos3_i2v.mp4")
395
- ```
396
-
397
- Example output generated from `assets/example_first_frame.png`:
398
-
399
- <video controls width="832" height="480" src="https://huggingface.co/nvidia/Cosmos3-Super-Image2Video/resolve/main/assets/example_output.mp4"></video>
400
-
401
- #### Inferencing with custom prompts
402
-
403
- Cosmos3-Super-Image2Video uses JSON-format prompts for optimal quality. The recommended way is to utilize [cosmos-framework](#prompt-upsampling). Here we provide a simple proof-of-concept script for convenience. It requires an OpenAI-compatible VLM model like `claude-opus-4.7` and `gpt-5.5`.
404
 
405
  ```bash
406
- export PROMPT_UPSAMPLER_API_KEY="..."
407
- python scripts/upsample_prompt.py \
408
- --model-name <model> \
409
- --base-url <VLM-endpoint-url> \
410
- --image-path assets/example_first_frame.png \
411
- --user-prompt "The ice cream melts and gradually disappears. The camera moves around." \
412
- --output-path scripts/upsampled.json
413
- ```
414
-
415
- ### Diffusers
416
-
417
- Cosmos3 is fully supported within the popular HuggingFace Diffusers package. This integration makes it a supported inference backend, allowing developers to easily incorporate Cosmos3's capabilities - such as text-to-image generation - into their pipelines using the Cosmos3OmniPipeline class, as demonstrated by the provided code examples (see examples for other modalities on the HuggingFace Cosmos3 page).
418
-
419
- #### Installation
420
-
421
- To install diffusers with Cosmos3OmniPipeline:
422
- ```
423
- uv venv --python 3.13 --seed --managed-python
424
- source .venv/bin/activate
425
  uv pip install \
426
  "diffusers @ git+https://github.com/huggingface/diffusers.git" \
427
- accelerate \
428
- av \
429
- cosmos_guardrail \
430
- huggingface_hub \
431
- imageio \
432
- imageio-ffmpeg \
433
- torch \
434
- torchvision \
435
- transformers
436
  ```
437
 
438
- #### Example: image to video generation with Diffusers
439
-
440
- The following example generates a video in approximately 170 seconds on a single GB200.
441
 
442
  ```python
443
  import json
444
-
445
  import torch
446
  from diffusers import Cosmos3OmniPipeline, UniPCMultistepScheduler
447
  from diffusers.utils import export_to_video, load_image
@@ -456,7 +51,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow
456
 
457
  image = load_image("assets/example_first_frame.png")
458
 
459
- # JSON-format prompt (see scripts/upsample_prompt.py to build your own).
460
  spec = json.load(open("assets/example_prompt.json"))
461
  prompt = spec["prompt"]
462
  negative_prompt = spec["negative_prompt"]
@@ -478,28 +73,17 @@ result = pipe(
478
  export_to_video(result.video, "output.mp4", fps=24, quality=7, macro_block_size=1)
479
  ```
480
 
481
- Example output generated by Diffusers:
482
-
483
- <video controls width="832" height="480" src="https://huggingface.co/nvidia/Cosmos3-Super-Image2Video/resolve/main/assets/example_output_diffusers.mp4"></video>
484
-
485
- ## Limitations
486
-
487
- Cosmos3 may produce imperfect outputs in challenging scenarios. Generation artifacts include temporal inconsistency, unstable camera or object motion, imprecise physical interactions, inaccurate audio-video synchronization, and action-state drift — especially in long-horizon or high-resolution outputs. Reasoning may also be incorrect: object states, causal relationships, spatial geometry, temporal ordering, agent intent, and future outcomes can be misinferred, and complex or long-context inputs may yield hallucinated entities, inconsistent interpretations, or implausible predictions. Because the model lacks an explicit physics simulator, 3D geometry, 4D space-time evolution, object permanence, contact dynamics, and physical laws are only approximated — producing artifacts such as disappearing or morphing objects, unrealistic collisions, and physically implausible motions. Quality further degrades in out-of-distribution environments, safety-critical edge cases, and domains underrepresented in training.
488
-
489
- Cosmos3 outputs should not be treated as physically accurate simulation, reliable ground-truth reasoning, or safety-certified decision making. Applications involving robotics control, autonomous systems, scientific simulation, or safety-critical planning require additional validation, external constraints, system-level safety analysis, and domain-specific guardrails before deployment.
490
-
491
- ## Inference
492
-
493
- **Acceleration Engine:** [PyTorch](https://pytorch.org/), [vLLM](https://github.com/vllm-project/vllm), [vLLM-Omni](https://github.com/vllm-project/vllm-omni), [Hugging Face Diffusers](https://github.com/huggingface/diffusers)
494
 
495
- **Test Hardware:** GB200 and H100
 
496
 
497
- ## Ethical Considerations
498
 
499
- 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.
500
 
501
- 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.
502
 
503
- 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.
504
 
505
- For more detailed information on ethical considerations for this model, please see the Model Card++ [Explainability](EXPLAINABILITY.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/).
 
1
  ---
2
+ library_name: diffusers
3
  license: other
4
  license_name: openmdw1.1-license
5
+ license_link: https://openmdw.ai/license/1-1/
6
+ pipeline_tag: image-to-video
 
7
  tags:
8
+ - nvidia
9
+ - cosmos
10
+ - cosmos3
11
+ - vllm-omni
12
+ - video-generation
 
 
13
  ---
14
 
15
  # **Cosmos 3: Omnimodal World Models for Physical AI**
16
+ **[Model Collection](https://huggingface.co/collections/nvidia/cosmos3)** | **[Code](https://github.com/nvidia/cosmos)** | **[Paper](https://huggingface.co/papers/2606.02800)** | **[Website](https://research.nvidia.com/labs/cosmos-lab/cosmos3/)**
17
 
18
+ [NVIDIA Cosmos™](https://github.com/nvidia/cosmos) is a world foundation model platform designed to accelerate the development of Physical AI by enabling machines to understand, simulate, and interact with the physical world across robotics, autonomous driving, and smart space environments.
19
 
20
+ ## Model Overview: Cosmos3-Super-Image2Video
21
 
22
+ **Cosmos3-Super-Image2Video** is a 64B parameter model designed for generating temporally coherent video sequences from a single input image and text instructions. It is part of the Cosmos 3 family, which uses a unified Mixture-of-Transformers (MoT) architecture to process and generate multimodal content.
23
 
24
+ ## Sample Usage
25
 
26
+ Cosmos 3 is fully supported within the Hugging Face `diffusers` library.
27
 
28
+ ### Installation
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
 
30
  ```bash
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  uv pip install \
32
  "diffusers @ git+https://github.com/huggingface/diffusers.git" \
33
+ accelerate av cosmos_guardrail huggingface_hub imageio imageio-ffmpeg torch torchvision transformers
 
 
 
 
 
 
 
 
34
  ```
35
 
36
+ ### Inference with Diffusers
 
 
37
 
38
  ```python
39
  import json
 
40
  import torch
41
  from diffusers import Cosmos3OmniPipeline, UniPCMultistepScheduler
42
  from diffusers.utils import export_to_video, load_image
 
51
 
52
  image = load_image("assets/example_first_frame.png")
53
 
54
+ # JSON-format prompt (see the GitHub repository to build your own).
55
  spec = json.load(open("assets/example_prompt.json"))
56
  prompt = spec["prompt"]
57
  negative_prompt = spec["negative_prompt"]
 
73
  export_to_video(result.video, "output.mp4", fps=24, quality=7, macro_block_size=1)
74
  ```
75
 
76
+ ## Model Architecture
 
 
 
 
 
 
 
 
 
 
 
 
77
 
78
+ **Architecture Type:** Transformer
79
+ **Network Architecture:** Mixture-of-Transformers (MoT)
80
 
81
+ Cosmos3 is an omnimodal foundation model built on a Mixture-of-Transformers (MoT) architecture consisting of two complementary transformer towers: an autoregressive transformer for discrete token generation and a diffusion transformer for continuous multimodal generation.
82
 
83
+ ## Limitations
84
 
85
+ Cosmos3 may produce imperfect outputs in challenging scenarios. Artifacts can include temporal inconsistency, unstable camera or object motion, imprecise physical interactions, and action-state drift. Because the model approximates physical laws without an explicit physics simulator, users may see disappearing objects or unrealistic collisions.
86
 
87
+ ## License
88
 
89
+ NVIDIA Cosmos source code and models are released under the [OpenMDW-1.1](https://openmdw.ai/license/1-1/) License.