TrNi commited on
Commit
40bd7a8
ยท
verified ยท
1 Parent(s): aade19b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -7
README.md CHANGED
@@ -13,21 +13,25 @@ library_name: torchao
13
  pipeline_tag: text-to-3d
14
  ---
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 โ†’ 11.3 GB (55%โ†“)** 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:
 
13
  pipeline_tag: text-to-3d
14
  ---
15
 
16
+ # ๐Ÿš€ First INT4 Quantized Efficient-Cube3D - Run on Half the VRAM
17
+ ![](https://img.shields.io/badge/Model_Size-1.26_GB_(82%25โ†“)-blueviolet)
18
+ ![](https://img.shields.io/badge/Peak_VRAM-11.3_GB_(55%25โ†“)-brightgreen)
19
+ ![](https://img.shields.io/badge/Setup_Time-6.9s_(97%25โ†“)-blue)
20
+ ![](https://img.shields.io/badge/Latency-14.2s_(on_par)-yellowgreen)
21
 
22
+ Presenting the **first INT4 quantized efficient version** of [Cube3D v0.5](https://huggingface.co/Roblox/cube3d-v0.5), a text-to-3D mesh generative model.
23
+ Quantized via **RTN W4A16** (group_size=128) using [torchao](https://github.com/pytorch/ao), it cuts the model size from **7.2GB โ†’ 1.3GB (82%โ†“)**
24
+ and peak VRAM from **25.4 GB โ†’ 11.3 GB (55%โ†“)** while maintaining the same inference speed and comparable shape fidelity -
25
+ enabling 3D shape generation on much smaller, more accessible GPUs.
 
26
 
27
  | | BF16 + Engine | BF16 + EngineFast | **INT4 + EngineFast** |
28
  |---|:-:|:-:|:-:|
29
+ | ๐Ÿ’พ Model size | 7.17 GB | 7.17 GB | **1.26 GB (82%โ†“)** |
30
  | ๐ŸŽฎ Peak VRAM | 21.7 GB | 25.4 GB | **11.3 GB (55%โ†“)** โœจ |
31
  | ๐Ÿ“ฆ Setup time | 19.4 s | 206.9 s | **6.9 s (97%โ†“)** |
32
  | โฑ๏ธ Latency | 90.9 s | 15.0 s | **14.2 s** |
33
 
34
+ <mark>๐Ÿ’ก The 82% size reduction and 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.
35
  </mark>
36
 
37
  ### Original BF16 vs Quantized INT4 Comparisons: