Skip to content

Add NIM inference path for the Cosmos 3 Generator#258

Merged
rickzw merged 7 commits into
NVIDIA:mainfrom
ekrivovnv:egor/generator_tutorial
Jul 14, 2026
Merged

Add NIM inference path for the Cosmos 3 Generator#258
rickzw merged 7 commits into
NVIDIA:mainfrom
ekrivovnv:egor/generator_tutorial

Conversation

@ekrivovnv

Copy link
Copy Markdown
Contributor

Added tutorial on using NIM for the Generator path.

Also, added benchmarks for the Generator, overall they show perfomance improvement over vllm-omni.

Comment thread cookbooks/cosmos3/README.md Outdated
| [vLLM](#vllm) | OpenAI-compatible reasoning server (image/video understanding) | Reasoner |
| [vLLM-Omni](#vllm-omni) | OpenAI-compatible generation server (image/video/audio/action) | Generator (Audiovisual, Action) |
| [NIM](#nim) | Prebuilt OpenAI-compatible reasoning server (image/video understanding); no venv | Reasoner |
| [NIM](#nim) | Prebuilt NGC containers; Reasoner serving plus Generator audiovisual T2V/I2V only | Reasoner, Generator (Audiovisual) |

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Use the same naming conventions as above (image/video/audio). Since this reflects two separate NIMs for now, use (video to text reasoning) for reasoner.

Comment thread cookbooks/cosmos3/README.md Outdated
- [`uv`](https://docs.astral.sh/uv/getting-started/installation/), `git`, and `git-lfs` installed.
- Hugging Face access to the gated Cosmos3 model repos. Generator also requires
access to the gated
- Hugging Face access to the gated Cosmos3 model repos for Hugging Face based

@nv-asotelo nv-asotelo Jul 2, 2026

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

be more explicit, or state all supported backends except NIM.

Make sure to remember SGLang and TRT-LLM if making access claims.

Comment thread cookbooks/cosmos3/README.md Outdated

A prebuilt container that serves **Cosmos3-Generator Text2Video and Image2Video
only** through `POST /v1/infer`. The NIM infers mode from the request fields:
non-empty `prompt` with no `image` means T2V; `image` provided means I2V. The

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Explicit - Text-to-video, image-to-video

Comment thread README.md Outdated
- **World generation:** Produce images, videos, synchronized sound, and action-conditioned rollouts from text, image, video, or action inputs.
- **Action modeling:** Predict policy actions, inverse dynamics, and forward dynamics for robotics, camera motion, egocentric motion, and autonomous-driving settings.
- **Research and production paths:** Use Diffusers and Transformers for Python-first development, then vLLM-Omni and vLLM for OpenAI-compatible serving.
- **Research and production paths:** Use Diffusers and Transformers for Python-first development, vLLM-Omni and vLLM for OpenAI-compatible serving, and NIM containers for turnkey Reasoner serving plus Generator T2V/I2V deployment.

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

explicitly, "text to video" and "image to video". This is a different readme file and the first time the acronym is presented.

@nv-asotelo

Copy link
Copy Markdown

@ekrivovnv I added some comments, just minor nits. Please address and LGTM

@ekrivovnv

Copy link
Copy Markdown
Contributor Author

@nv-asotelo I think all your comments were addressed

@ekrivovnv
ekrivovnv requested a review from nv-asotelo July 3, 2026 17:21

@Dinghow Dinghow left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@qmiao-hub
qmiao-hub requested review from rickzw and yaoxu-crypto and removed request for nv-asotelo July 14, 2026 22:07

@rickzw rickzw left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@rickzw
rickzw merged commit aefdf85 into NVIDIA:main Jul 14, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants