Instructions to use nvidia/Cosmos3-Edge-FP8 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Cosmos
How to use nvidia/Cosmos3-Edge-FP8 with Cosmos:
# 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
Commit ·
02df5ef
1
Parent(s): cc6b7d7
Update FP8 performance benchmarks and remove unmeasured T2000 row (#6)
Browse files- Update FP8 performance benchmarks and remove unmeasured T2000 row (00213e1b98514e65c963df7fc4806eb72c278ad8)
- Remove B200 from model card test hardware list (c9370cc4e13a4017f18da94603b137f4b7b6811e)
Co-authored-by: Sergiy Fefilatyev <fsergiy@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -298,26 +298,27 @@ The Edge model is a strong initialization for downstream action tasks. For examp
|
|
| 298 |
|
| 299 |
The table reports single-GPU inference performance for Cosmos3-Edge-FP8 using FLASH_ATTN unless otherwise noted.
|
| 300 |
|
| 301 |
-
| Hardware | Generator — 480p (frames/s) | Reasoner — time to first token (ms) | Reasoner — output token throughput (tokens/s) | Action Policy — 480p (actions/s,
|
| 302 |
| --- | ---: | ---: | ---: | ---: |
|
| 303 |
-
| H100 | 13.04 | 108.1 | 312.64 |
|
| 304 |
-
| H200 | 12.96 | 103.0 | 341.18 |
|
| 305 |
-
| H20 | 3.30 | 163.6 | 262.42 | 6.
|
| 306 |
-
| DGX Station |
|
| 307 |
-
| DGX Spark |
|
| 308 |
-
| RTX PRO 6000 | 7.52 |
|
| 309 |
-
| RTX PRO 5000 |
|
| 310 |
-
| Jetson Thor T5000 |
|
| 311 |
-
| Jetson T3000 |
|
| 312 |
-
| Jetson T2000 | - | - | - | - |
|
| 313 |
| Jetson AGX Orin | - | 817.8 | 41.46 | - |
|
| 314 |
|
| 315 |
A dash (-) indicates unavailable benchmark results for the specified configuration.
|
| 316 |
|
| 317 |
-
- **Generator:** FP8 image-to-video (`i2v`) with vLLM-Omni, 832 × 480 resolution, 121 generated frames, 35 denoising steps, concurrency 1, and one warmup request. Each GPU uses its fastest measured backend for this image-to-video workload: FLASH_ATTN for H100 and H200; CUDNN_ATTN for H20
|
| 318 |
-
- **Reasoner:** FP8 vLLM with concurrency 1, 50 input text tokens plus a 30-second, 374 × 374 video sampled at FPS 1 (30 frames), and 100 output tokens.
|
| 319 |
-
- **
|
| 320 |
-
- **
|
|
|
|
|
|
|
| 321 |
- **Embedded hardware:** Reasoner measurements use one integrated GPU: Jetson AGX Thor T5000 (128 GB), a T5000 devkit configured to emulate T3000 (12 SMs, 8 CPUs, 32 GB), and Jetson AGX Orin (64 GB), all in MAXN mode with clocks locked. Orin runs the FP8 checkpoint with weight-only W8A16 compute; Thor uses FP8 compute. T3000 results are emulated measurements. Hardware variants for other rows require matching benchmark results.
|
| 322 |
- Throughputs are calculated before rounding to two decimal places; time to first token is displayed to one decimal place. No estimates are included.
|
| 323 |
|
|
@@ -968,7 +969,7 @@ Cosmos3 outputs should not be treated as physically accurate simulation, reliabl
|
|
| 968 |
|
| 969 |
**Acceleration Engine:** [vLLM-Omni](https://github.com/vllm-project/vllm-omni), [vLLM](https://github.com/vllm-project/vllm), [PyTorch](https://pytorch.org/)
|
| 970 |
|
| 971 |
-
**Test Hardware:**
|
| 972 |
|
| 973 |
## Ethical Considerations
|
| 974 |
|
|
|
|
| 298 |
|
| 299 |
The table reports single-GPU inference performance for Cosmos3-Edge-FP8 using FLASH_ATTN unless otherwise noted.
|
| 300 |
|
| 301 |
+
| Hardware | Generator — 480p (frames/s) | Reasoner — time to first token (ms) | Reasoner — output token throughput (tokens/s) | Action Policy — 480p (actions/s, 32-action chunks) |
|
| 302 |
| --- | ---: | ---: | ---: | ---: |
|
| 303 |
+
| H100 | 13.04 | 108.1 | 312.64 | 20.50 |
|
| 304 |
+
| H200 | 12.96 | 103.0 | 341.18 | 21.11 |
|
| 305 |
+
| H20 | 3.30 | 163.6 | 262.42 | 6.81 |
|
| 306 |
+
| DGX Station | 30.63 | 65.4 | 425.65 | 30.48 |
|
| 307 |
+
| DGX Spark | 1.65 | 213.2 | 69.46 | 4.03 |
|
| 308 |
+
| RTX PRO 6000 | 7.52 | 121.0 | 245.38 | 10.77 |
|
| 309 |
+
| RTX PRO 5000 | 4.51 | 99.7 | 243.16 | 9.58 |
|
| 310 |
+
| Jetson Thor T5000 | 1.92 | 205.7 | 76.69 | 3.79 |
|
| 311 |
+
| Jetson T3000 | 0.88 | 384.3 | 62.16 | 1.81 |
|
|
|
|
| 312 |
| Jetson AGX Orin | - | 817.8 | 41.46 | - |
|
| 313 |
|
| 314 |
A dash (-) indicates unavailable benchmark results for the specified configuration.
|
| 315 |
|
| 316 |
+
- **Generator:** FP8 image-to-video (`i2v`) with vLLM-Omni, 832 × 480 resolution, 121 generated frames, 35 denoising steps, concurrency 1, and one warmup request. Each GPU uses its fastest measured backend for this image-to-video workload: FLASH_ATTN for H100 and H200; CUDNN_ATTN for H20, RTX PRO 6000, DGX Station, DGX Spark, RTX PRO 5000, Thor T5000, and emulated T3000. Throughput is 121 divided by end-to-end server inference time in seconds and measures inference throughput.
|
| 317 |
+
- **Reasoner:** FP8 vLLM with concurrency 1, 50 input text tokens plus a 30-second, 374 × 374 video sampled at FPS 1 (30 frames), and 100 output tokens. H100, H200, H20, RTX PRO 6000, and embedded hardware use this workload and FLASH_ATTN; output throughput copies the reported output-token throughput.
|
| 318 |
+
- **DGX Station, DGX Spark, and RTX PRO 5000 reasoner:** OSL (output sequence length) 100, FPS 1, and concurrency 1. The attention backend, input text length, video duration, and resolution are unspecified for these results.
|
| 319 |
+
- **Action Policy:** FP8 vLLM-Omni, Policy Robot (DROID), 480p, one GPU, 32-action chunks, 30 denoising steps, concurrency 1, and one warmup request. Throughput is 32 divided by end-to-end server inference time in seconds, reported in actions per second. Each GPU uses its fastest measured backend for this workload: FLASH_ATTN for H100 and H200; CUDNN_ATTN for H20, RTX PRO 6000, DGX Station, DGX Spark, RTX PRO 5000, Thor T5000, and emulated T3000.
|
| 320 |
+
- **Hardware:** Generator measurements use one H100 SXM 80 GB, H200 SXM 141 GB, or H20 SXM 96 GB GPU. The RTX PRO 6000 generator measurement uses one RTX PRO 6000 Blackwell Server Edition GPU. The RTX PRO 5000 generator measurement uses the 72 GB variant.
|
| 321 |
+
- **Embedded generator and action policy:** Thor T5000 and emulated T3000 use FP8 with CUDNN_ATTN and one warmup request. Policy requests use 736 × 544 output in the 480p model tier, with 32 actions plus one conditioning frame. Orin generator and policy measurements use a separate BF16 checkpoint and are omitted from this FP8 table.
|
| 322 |
- **Embedded hardware:** Reasoner measurements use one integrated GPU: Jetson AGX Thor T5000 (128 GB), a T5000 devkit configured to emulate T3000 (12 SMs, 8 CPUs, 32 GB), and Jetson AGX Orin (64 GB), all in MAXN mode with clocks locked. Orin runs the FP8 checkpoint with weight-only W8A16 compute; Thor uses FP8 compute. T3000 results are emulated measurements. Hardware variants for other rows require matching benchmark results.
|
| 323 |
- Throughputs are calculated before rounding to two decimal places; time to first token is displayed to one decimal place. No estimates are included.
|
| 324 |
|
|
|
|
| 969 |
|
| 970 |
**Acceleration Engine:** [vLLM-Omni](https://github.com/vllm-project/vllm-omni), [vLLM](https://github.com/vllm-project/vllm), [PyTorch](https://pytorch.org/)
|
| 971 |
|
| 972 |
+
**Test Hardware:** H100, H20, RTX PRO 6000, DGX Station, DGX Spark, Jetson Thor, Jetson AGX Orin
|
| 973 |
|
| 974 |
## Ethical Considerations
|
| 975 |
|