TrNi commited on
Commit
72b7e60
ยท
verified ยท
1 Parent(s): 36a33e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -15,19 +15,19 @@ pipeline_tag: text-to-3d
15
 
16
  # ๐Ÿš€ First INT4 Quantized Cube3D - Run on Half the VRAM
17
 
18
- ![](https://img.shields.io/badge/VRAM_-14.3_GB_(44%25โ†“)-brightgreen)
19
- ![](https://img.shields.io/badge/Loading-88%25_Faster-blue)
20
  ![](https://img.shields.io/badge/Latency-Same-green)
21
 
22
  Presenting the **first INT4 quantized version** of [Cube3D v0.5](https://huggingface.co/Roblox/cube3d-v0.5), a text-to-3D mesh generative model. Quantized via **RTN W4A16** (group_size=128) using [torchao](https://github.com/pytorch/ao), it cuts peak VRAM from **25.4 GB โ†’ 14.3 GB (44%โ†“)** while maintaining the same inference speed and comparable shape fidelity - enabling 3D shape generation on much smaller, more accessible GPUs.
23
 
24
  | | BF16 + Engine | BF16 + EngineFast | **INT4 + EngineFast** |
25
  |---|:-:|:-:|:-:|
26
- | ๐ŸŽฎ Peak VRAM | 21.7 GB | 25.4 GB | **14.3 GB (44%โ†“)** โœจ |
27
- | ๐Ÿ“ฆ Setup time | 19.4 s | 206.9 s | **25.1 s (88%โ†“)** |
28
  | โฑ๏ธ Latency | 90.9 s | 15.0 s | **14.2 s** |
29
 
30
- <mark>๐Ÿ’ก The 44% VRAM reduction means this model now fits on a single 16 GB GPU (e.g. NVIDIA L4, A10, A2 etc.), bringing high-quality text-to-3D generation to individual researchers and end-user hardware.
31
  </mark>
32
 
33
  ### Original BF16 vs Quantized INT4 Comparisons:
 
15
 
16
  # ๐Ÿš€ First INT4 Quantized Cube3D - Run on Half the VRAM
17
 
18
+ ![](https://img.shields.io/badge/VRAM_-11.3_GB_(55%25โ†“)-brightgreen)
19
+ ![](https://img.shields.io/badge/Loading-97%25_Faster-blue)
20
  ![](https://img.shields.io/badge/Latency-Same-green)
21
 
22
  Presenting the **first INT4 quantized version** of [Cube3D v0.5](https://huggingface.co/Roblox/cube3d-v0.5), a text-to-3D mesh generative model. Quantized via **RTN W4A16** (group_size=128) using [torchao](https://github.com/pytorch/ao), it cuts peak VRAM from **25.4 GB โ†’ 14.3 GB (44%โ†“)** while maintaining the same inference speed and comparable shape fidelity - enabling 3D shape generation on much smaller, more accessible GPUs.
23
 
24
  | | BF16 + Engine | BF16 + EngineFast | **INT4 + EngineFast** |
25
  |---|:-:|:-:|:-:|
26
+ | ๐ŸŽฎ Peak VRAM | 21.7 GB | 25.4 GB | **11.3 GB (55%โ†“)** โœจ |
27
+ | ๐Ÿ“ฆ Setup time | 19.4 s | 206.9 s | **6.9 s (97%โ†“)** |
28
  | โฑ๏ธ Latency | 90.9 s | 15.0 s | **14.2 s** |
29
 
30
+ <mark>๐Ÿ’ก The 55% VRAM reduction means this model now fits on a single 15 GB GPU (e.g. NVIDIA L4, A10, A2 etc.), bringing high-quality text-to-3D generation to individual researchers and end-user hardware.
31
  </mark>
32
 
33
  ### Original BF16 vs Quantized INT4 Comparisons: