tangyue0820 fsergiy commited on
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>

Files changed (1) hide show
  1. README.md +17 -16
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, 16-action chunks) |
302
  | --- | ---: | ---: | ---: | ---: |
303
- | H100 | 13.04 | 108.1 | 312.64 | 16.33 |
304
- | H200 | 12.96 | 103.0 | 341.18 | 16.39 |
305
- | H20 | 3.30 | 163.6 | 262.42 | 6.05 |
306
- | DGX Station | - | - | - | - |
307
- | DGX Spark | - | - | - | - |
308
- | RTX PRO 6000 | 7.52 | - | - | 6.91 |
309
- | RTX PRO 5000 | - | - | - | - |
310
- | Jetson Thor T5000 | - | 205.7 | 76.69 | - |
311
- | Jetson T3000 | - | 384.3 | 62.16 | - |
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 and RTX PRO 6000. Throughput is 121 divided by end-to-end server inference time in seconds and measures inference throughput. The reported times are 9.277 s (H100), 9.339 s (H200), 36.677 s (H20), and 16.085 s (RTX PRO 6000).
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. Both metrics use this workload and FLASH_ATTN; output throughput copies the reported output-token throughput.
319
- - **Action Policy:** FP8 vLLM-Omni, Policy Robot (DROID), 480p, one GPU, 16-action chunks, 30 denoising steps, concurrency 1, and one warmup request. Throughput is 16 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: CUDNN_ATTN for H100, H200, H20, and RTX PRO 6000. The reported times are 0.980 s (H100), 0.976 s (H200), 2.645 s (H20), and 2.315 s (RTX PRO 6000).
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.
 
 
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:** B200, H100, H20, RTX PRO 6000, DGX Station, DGX Spark, Jetson Thor, Jetson AGX Orin
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