cosmos3-inference

bởi nvidia

Hướng dẫn người dùng chạy suy luận Cosmos3 — tạo batch ngoại tuyến, phục vụ trực tuyến với Ray và Gradio, các tùy chọn song song, định dạng đầu vào, lấy mẫu…

npx skills add https://github.com/nvidia/cosmos-framework --skill cosmos3-inference

Cosmos3 Inference

When to use this skill

  • Use when a user wants to generate images or videos with Cosmos3
  • Use when a user asks about inference parameters, input formats, or parallelism
  • Use when a user wants to set up online serving (Ray Serve, Gradio)
  • Use when a user asks about prompt engineering or upsampling
  • For environment or import errors, hand off to cosmos3-env-troubleshoot

Path convention

All paths below are relative to the cosmos3 package root (../../../ from this skill file). All uv run / python commands should also be run from there.

Where to find answers

User questionGo to
How do I run inference? (single-GPU, multi-GPU)README.md § Inference
Which model should I use? (Nano vs Super, memory, shift)README.md § Models
Which modality? (t2i, t2v, i2v, examples)README.md § Modalities
What parallelism preset? (latency vs throughput)README.md § Inference
What input fields are available? (prompt, vision_path, num_frames, ...)docs/inference.md § Sample Arguments
What are the default parameter values?cosmos_framework/inference/defaults/<model_mode>/sample_args.json (per-modality JSON)
How do I use custom defaults?docs/inference.md § Custom Defaults
How do I override a parameter? (precedence)docs/faq.md § How do I override a default parameter?
What is the shift parameter?docs/faq.md § What is the shift parameter?
How many frames can I generate? (resolution caps)docs/faq.md § How many frames can I generate?
How do I start Ray Serve / Gradio / submit requests?docs/faq.md § How do I run online inference with Ray?
How do I upsample short prompts?docs/faq.md § Prompt upsampling
How do I use the low-level API? (examples/)examples/inference.py (model API) / examples/inference_pipeline.py (pipeline API)
All CLI flagsuv run --all-extras --group=cu130 python -m cosmos_framework.scripts.inference --help

Things not obvious from the docs

  • Path resolution: relative paths in input JSON files are resolved relative to the JSON file's directory, not the working directory.
  • Seed: always pass --seed for reproducible results. Without it, a random seed is used each time.
  • Resume: interrupted runs can be resumed by re-running the same command — existing outputs are skipped automatically.
  • --keep-going: continues processing remaining samples after a per-sample failure (e.g. guardrail rejection). Used in online serving by default.
  • Unique names: every sample in a run must have a unique name field, or the script will error.

Related skills

SkillWhen to use
../cosmos3-setup/SKILL.mdInstallation and environment setup
../cosmos3-codebase-nav/SKILL.mdFinding files, parameters, and configs in code
../cosmos3-env-troubleshoot/SKILL.mdDebugging environment and runtime errors

Thêm skills từ nvidia

compileiq-debug
nvidia
Sử dụng khi có điều gì đó không ổn: Search() bị treo, tất cả các đánh giá đều trả về INVALID_SCORE, điểm số không cải thiện, mọi cấu hình đều trả về cùng một số, lỗi ptxas…
create-github-pr
nvidia
Tạo pull request GitHub bằng cách sử dụng gh CLI. Sử dụng khi người dùng muốn tạo PR mới, gửi mã để xem xét, hoặc mở pull request. Từ khóa kích hoạt -…
nemoclaw-maintainer-cross-issue-sweep
nvidia
Quét các vấn đề đang mở khác để tìm những vấn đề mà một PR nhất định có thể sửa hoặc vô tình làm hỏng. Đưa ra các cơ hội sửa lỗi liền kề và rủi ro mâu thuẫn với file:dòng…
fhir-basics
nvidia
Dạy các tác nhân cách hoạt động của API FHIR R4, những tài nguyên có sẵn, cách truy vấn chúng với tham số tìm kiếm, và cách phân tích chính xác tất cả các định dạng phản hồi…
compileiq-validate-result
nvidia
Sử dụng SAU KHI tìm kiếm hoàn tất và TRƯỚC KHI yêu cầu tăng tốc hoặc gửi ACF. Tải tệp CSV dump_results, trích xuất các ứng viên top-K (đơn mục tiêu)…
changelog-audit
nvidia
Kiểm tra Warp CHANGELOG.md trước khi phát hành: khôi phục các mục bị mất, sắp xếp theo tác động người dùng, tinh chỉnh ngôn ngữ mục, xuống dòng và (chế độ nhánh phát hành) so sánh bump…
maintain-dynamic-plugins
nvidia
Duy trì các bộ nạp plugin động NeMo Relay, tệp kê khai, SDK gốc Rust, giao thức worker gRPC, SDK worker Python, tài liệu, kiểm thử và phạm vi quy trình phát hành
dgx-diagnose
nvidia
Chẩn đoán các sự cố thường gặp của DGX Station GB300 — lỗi CUDA, nhắm sai GPU, lỗi container vLLM/SGLang, vấn đề trạng thái MIG, lỗi NVLink/Fabric Manager,…