Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ base_model:
|
|
17 |
pipeline_tag: text-generation
|
18 |
---
|
19 |
|
20 |
-
[Phi4-mini](https://huggingface.co/microsoft/Phi-4-mini-instruct) quantized with [torchao](https://huggingface.co/docs/transformers/main/en/quantization/torchao) int4 weight only quantization, using [hqq](https://mobiusml.github.io/hqq_blog/) algorithm for improved accuracy, by PyTorch team. Use it directly or serve using [vLLM](https://docs.vllm.ai/en/latest/) for 67% VRAM reduction and
|
21 |
|
22 |
# Inference with vLLM
|
23 |
Install vllm nightly and torchao nightly to get some recent changes:
|
@@ -281,11 +281,11 @@ print(f"Peak Memory Usage: {mem:.02f} GB")
|
|
281 |
Our int4wo is only optimized for batch size 1, so expect some slowdown with larger batch sizes, we expect this to be used in local server deployment for single or a few users where the decode tokens per second will matters more than the time to first token.
|
282 |
|
283 |
## Results (A100 machine)
|
284 |
-
| Benchmark (Latency) | |
|
285 |
-
|
286 |
-
| | Phi-4 mini-Ins | phi4-mini-int4wo-hqq
|
287 |
-
| latency (batch_size=1) | 2.46s | 2.2s (
|
288 |
-
| serving (num_prompts=1) | 0.87 req/s | 1.05 req/s (
|
289 |
|
290 |
Note the result of latency (benchmark_latency) is in seconds, and serving (benchmark_serving) is in number of requests per second.
|
291 |
Int4 weight only is optimized for batch size 1 and short input and output token length, please stay tuned for models optimized for larger batch sizes or longer token length.
|
|
|
17 |
pipeline_tag: text-generation
|
18 |
---
|
19 |
|
20 |
+
[Phi4-mini](https://huggingface.co/microsoft/Phi-4-mini-instruct) quantized with [torchao](https://huggingface.co/docs/transformers/main/en/quantization/torchao) int4 weight only quantization, using [hqq](https://mobiusml.github.io/hqq_blog/) algorithm for improved accuracy, by PyTorch team. Use it directly or serve using [vLLM](https://docs.vllm.ai/en/latest/) for 67% VRAM reduction and 1.12x-1.2x speedup on A100 GPUs.
|
21 |
|
22 |
# Inference with vLLM
|
23 |
Install vllm nightly and torchao nightly to get some recent changes:
|
|
|
281 |
Our int4wo is only optimized for batch size 1, so expect some slowdown with larger batch sizes, we expect this to be used in local server deployment for single or a few users where the decode tokens per second will matters more than the time to first token.
|
282 |
|
283 |
## Results (A100 machine)
|
284 |
+
| Benchmark (Latency) | | |
|
285 |
+
|----------------------------------|----------------|----------------------------|
|
286 |
+
| | Phi-4 mini-Ins | phi4-mini-int4wo-hqq |
|
287 |
+
| latency (batch_size=1) | 2.46s | 2.2s (1.12x speedup) |
|
288 |
+
| serving (num_prompts=1) | 0.87 req/s | 1.05 req/s (1.20x speedup) |
|
289 |
|
290 |
Note the result of latency (benchmark_latency) is in seconds, and serving (benchmark_serving) is in number of requests per second.
|
291 |
Int4 weight only is optimized for batch size 1 and short input and output token length, please stay tuned for models optimized for larger batch sizes or longer token length.
|