You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Dec 1, 2024. It is now read-only.
Copy file name to clipboardExpand all lines: README.md
+5-5Lines changed: 5 additions & 5 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -71,11 +71,11 @@ See [example](flexgen/apps/data_wrangle)
71
71
The corresponding effective batch sizes are in parentheses. Please see [here](benchmark/batch_size_table.md) for more details.
72
72
| System | OPT-6.7B | OPT-30B | OPT-175B |
73
73
| ------ | -------- | ------- | -------- |
74
-
| Hugging Face Accelerate | 25.12 (2 on gpu) | 0.62 (8 on cpu ) | 0.01 (2 on disk) |
75
-
| DeepSpeed ZeRO-Inference | 9.28 (16 on cpu) | 0.60 (4 on cpu) | 0.01 (1 on disk) |
76
-
| Petals | 8.25 | 2.84 | 0.08 |
77
-
| FlexGen | 25.26 (2 on gpu) | 7.32 (144 on cpu) | 0.69 (256 on disk) |
78
-
| FlexGen with Compression | **29.12** (72 on gpu) | **8.38** (512 on cpu) | **1.12** (144 on cpu) |
74
+
| Hugging Face Accelerate | 25.12 (2 on GPU) | 0.62 (8 on CPU) | 0.01 (2 on disk) |
75
+
| DeepSpeed ZeRO-Inference | 9.28 (16 on CPU) | 0.60 (4 on CPU) | 0.01 (1 on disk) |
76
+
| Petals\* | - | - | 0.05 |
77
+
| FlexGen | 25.26 (2 on GPU) | 7.32 (144 on CPU) | 0.69 (256 on disk) |
78
+
| FlexGen with Compression | **29.12** (72 on GPU) | **8.38** (512 on CPU) | **1.12** (144 on CPU) |
79
79
80
80
- Hardware: an NVIDIA T4 (16GB) instance on GCP with 208GB of DRAM and 1.5TB of SSD.
81
81
- Workload: input sequence length = 512, output sequence length = 32. The batch size is tuned to **a large value** that maximizes the generation throughput for each system.
0 commit comments