Instructions to use nvidia/Cosmos-H-Dreams with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NV-MedTech
How to use nvidia/Cosmos-H-Dreams with NV-MedTech:
# 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
- Notebooks
- Google Colab
- Kaggle
Explainability Subcard
intended_domain
Real-time interactive surgical simulation, surgeon/trainee rehearsal, closed-loop surgical policy evaluation, and synthetic data generation.
Model Type
Diffusion Transformer (bidirectional teacher distilled into a causal, few-step, streaming student)
Intended Users
Medical Robotics Engineers, Surgical Robotics Researchers, Healthcare AI Developers, Surgeons and Surgical Trainees
Output
Types: A stream of generated video frames (12-frame autoregressive blocks). Formats: Red, Green, Blue (RGB)
Describe how the model works:
The model accepts a surgical context frame plus a stream of unified 44-dimensional kinematic action vectors (for the released dVRK tabletop checkpoint, a 20-dimensional dual-arm content vector — per manipulator: 3D translation delta, 6D rotation, and gripper — zero-padded to 44D). It autoregressively generates the next 12-frame block conditioned on the current actions and on recent history maintained in a streaming key/value cache. Actions modulate generation through the timestep/AdaLN pathway. Because it is a causal, few-step (2–4 step) distilled student, it responds immediately to live actions and streams frames in real time on a single GPU, so a human operator (keyboard or Meta Quest headset) or a learned policy can act inside the synthesized scene and observe the consequences live.
Name the adversely impacted groups this has been tested to deliver comparable outcomes regardless of:
None
Technical Limitations & Mitigation:
The model may underperform in poor or variable lighting conditions, occlusions from instruments or blood, and specular reflections, which can degrade visual predictions. The model may not perform well for out-of-distribution scenarios including novel procedures, unusual anatomies, or emergency situations not well-represented in training data; rapid motions or long-horizon rollouts where autoregressive drift accumulates errors (mitigated but not eliminated by the self-forcing distillation, the rolling KV-cache window, and the appearance-sink frame); actions beyond the trained kinematic range or near joint limits; and generalization challenges across different camera placements, surgical sites, or surgeon styles. As a few-step distilled student optimized for real-time streaming, it may also trade some per-frame fidelity relative to the many-step bidirectional teacher.
Mitigation: To mitigate these limitations, we recommend performing data augmentation with lighting and occlusion variations; uncertainty estimation and out-of-distribution detection to flag anomalous states; limiting autoregressive rollout length with periodic ground-truth re-anchoring/reset; enforcing kinematic and safety constraints; multi-site training data collection; and maintaining strict human oversight with multiple safety layers. Longer-horizon teacher supervision during distillation is used specifically to reduce long-rollout shifting and drift artifacts.
Verified to have met prescribed NVIDIA quality standards:
Yes
Performance Metrics:
Robust L1 and SSIM vs. number of generated frames. Real-time system quality is assessed with end-to-end input-to-pixel latency and effective FPS, and interactivity is assessed with closed-loop action-controllability probes.
Potential Known Risks:
The model may generate videos that contain artifacts. The model may inaccurately represent 3D space, 4D space-time, or physical laws in the generated videos, leading to artifacts such as disappearing or morphing objects, unrealistic interactions, implausible motions, and physically inconsistent outcomes. As a real-time autoregressive model, errors can compound over long rollouts.
Licensing:
Governing Terms: Use of this model is governed by the NVIDIA Open Model License Agreement.