Cosmos-Predict2: A Suite of Diffusion-based World Foundation Models Available in 0.6B, 2B, and 14B

Cosmos | Code | Website

Model Overview

Description

Cosmos-Predict2: A family of highly performant pre-trained world foundation models purpose-built for generating physics-aware images, videos and world states for physical AI development.

Cosmos-Predict2 diffusion models are a collection of diffusion based world foundation models that generate dynamic, high quality images and videos from text, image, or video inputs. It can serve as the building block for various applications or research that are related to world generation. The models are ready for commercial use under NVIDIA Open Model license agreement.

Model Developer: NVIDIA

Model Versions

The Cosmos-Predict2 diffusion-based model family includes the following models:

• Cosmos-Predict2-0.6B-Text2Image
  • Given a text description, predict an output image.
• Cosmos-Predict2-2B-Text2Image
  • Given a text description, predict an output image.
• Cosmos-Predict2-14B-Text2Image
  • Given a text description, predict an output image.
• Cosmos-Predict2-2B-Video2World
  • Given a text description and an image as the first frame, predict the future frames.
  • We have four variants for this model to support different use cases:
    • variant that produces 720P video with 16FPS (this is our default model)
    • variant that produces 720P video with 10FPS
    • variant that produces 480P video with 16FPS
    • variant that produces 480P video with 10FPS
• Cosmos-Predict2-14B-Video2World
  • Given a text description and an image as the first frame, predict the future frames.
  • We have four variants for this model to support different use cases:
    • variant that produces 720P video with 16FPS (this is our default model)
    • variant that produces 720P video with 10FPS
    • variant that produces 480P video with 16FPS
    • variant that produces 480P video with 10FPS
• Cosmos-Predict2-14B-Video2World-Sample-GR00T-Dreams-GR1:
  • Video + Text based future visual world generation, post trained on GR00T GR1 data
• Cosmos-Predict2-14B-Video2World-Sample-GR00T-Dreams-DROID:
  • Video + Text based future visual world generation, post trained on GR00T DROID data
• Cosmos-Predict2-2B-Action-Conditioned-Sample
  • Given an image as the first frame and next 12 actions, predict the future 12 frames.

License

This model is released under the NVIDIA Open Model License. For a custom license, please contact cosmos-license@nvidia.com.

Under the NVIDIA Open Model License, NVIDIA confirms:

• Models are commercially usable.
• You are free to create and distribute Derivative Models.
• NVIDIA does not claim ownership to any outputs generated using the Models or Derivative Models.

Important Note: If you bypass, disable, reduce the efficacy of, or circumvent any technical limitation, safety guardrail or associated safety guardrail hyperparameter, encryption, security, digital rights management, or authentication mechanism contained in the Model, your rights under NVIDIA Open Model License Agreement will automatically terminate.

Deployment Geography

Global

Model Architecture

Cosmos-Predict2-0.6B-Text2Image is a diffusion transformer model designed for image denoising in the latent space. The network is composed of interleaved self-attention, cross-attention and feedforward layers as its building blocks. The cross-attention layers allow the model to condition on input text throughout the denoising process. Before each layer, adaptive layer normalization is applied to embed the time information for denoising.

Input/Output Specifications

• Input

  • Input Type(s): Text
  • Input Format(s): String
  • Input Parameters: One-dimensional (1D)
  • Other Properties Related to Input:
    • The input string should contain fewer than 300 words and should provide descriptive content for world generation, such as a scene description, key objects or characters, background, and any specific actions or motions to be depicted within the 5-second duration.

• Output

  • Output Type(s): Image
  • Output Format(s): jpg
  • Output Parameters: Two-dimensional (2D)
  • Other Properties Related to Output: By default, the generated image is with a resolution of 1280x704 pixels and RGB color. The image content visualizes the input text description, capturing key elements within the specified time constraints.

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.

Software Integration

Runtime Engine(s):

• Cosmos-Predict2
• Diffusers

import torch
from diffusers import Cosmos2TextToImagePipeline

# Available checkpoints: nvidia/Cosmos-Predict2-0.6B-Text2Image, nvidia/Cosmos-Predict2-2B-Text2Image, nvidia/Cosmos-Predict2-14B-Text2Image
model_id = "nvidia/Cosmos-Predict2-0.6B-Text2Image"
pipe = Cosmos2TextToImagePipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
pipe.to("cuda")

prompt = "A close-up shot captures a vibrant yellow scrubber vigorously working on a grimy plate, its bristles moving in circular motions to lift stubborn grease and food residue. The dish, once covered in remnants of a hearty meal, gradually reveals its original glossy surface. Suds form and bubble around the scrubber, creating a satisfying visual of cleanliness in progress. The sound of scrubbing fills the air, accompanied by the gentle clinking of the dish against the sink. As the scrubber continues its task, the dish transforms, gleaming under the bright kitchen lights, symbolizing the triumph of cleanliness over mess."
negative_prompt = "The video captures a series of frames showing ugly scenes, static with no motion, motion blur, over-saturation, shaky footage, low resolution, grainy texture, pixelated images, poorly lit areas, underexposed and overexposed scenes, poor color balance, washed out colors, choppy sequences, jerky movements, low frame rate, artifacting, color banding, unnatural transitions, outdated special effects, fake elements, unconvincing visuals, poorly edited content, jump cuts, visual noise, and flickering. Overall, the video is of poor quality."

output = pipe(
    prompt=prompt, negative_prompt=negative_prompt, generator=torch.Generator().manual_seed(1)
).images[0]
output.save("output.png")

Supported Hardware Microarchitecture Compatibility:

• NVIDIA Ampere
• NVIDIA Blackwell
• NVIDIA Hopper

Note: Only BF16 precision is tested. Other precisions like FP16 or FP32 are not officially supported.

Inference

Acceleration Engine: PyTorch, Transformer Engine

Operating System(s):

• Linux (We have not tested on other operating systems.)

System Requirements and Performance: We will publish the requirements shortly.

Quality Benchmarks: We will publish the evaluation shortly. |

Usage

• See Cosmos-Predict2 for details.

Limitations

Despite various improvements in world generation for Physical AI, Cosmos-Predict2 text2image models still face technical and application limitations for world prediction. In particular, they struggle to generate high-resolution images without artifacts. Common issues include camera and object motion instability, and imprecise interactions. The models may inaccurately represent 3D space, or physical laws in the generated images, leading to artifacts such as unrealistic interactions and implausible motions. As a result, applying these models for applications that require simulating physical law-grounded environments or complex multi-agent dynamics remains challenging.

Ethical Considerations

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. When downloaded or used in accordance with our terms of service, 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.

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.

For more detailed information on ethical considerations for this model, please see the subcards of Explainability, Bias, Safety & Security, and Privacy below. Please report security vulnerabilities or NVIDIA AI Concerns here.

Plus Plus (++) Promise

We value you, the datasets, the diversity they represent, and what we have been entrusted with. This model and its associated data have been:

• Verified to comply with current applicable disclosure laws, regulations, and industry standards.
• Verified to comply with applicable privacy labeling requirements.
• Annotated to describe the collector/source (NVIDIA or a third-party).
• Characterized for technical limitations.
• Reviewed to ensure proper disclosure is accessible to, maintained for, and in compliance with NVIDIA data subjects and their requests.
• Reviewed before release.
• Tagged for known restrictions and potential safety implications.

Bias

Explainability

Privacy

Safety