VOOZH about

URL: https://huggingface.co/RedHatAI/granite-3.1-8b-instruct-quantized.w4a16

⇱ RedHatAI/granite-3.1-8b-instruct-quantized.w4a16 · Hugging Face


granite-3.1-8b-instruct-quantized.w4a16 👁 Model Icon

👁 Validated Badge

Model Overview

  • Model Architecture: granite-3.1-8b-instruct
    • Input: Text
    • Output: Text
  • Model Optimizations:
    • Weight quantization: INT4
    • Activation quantization: INT4
  • Release Date: 1/8/2025
  • Version: 1.0
  • Validated on: RHOAI 2.20, RHAIIS 3.0, RHELAI 1.5
  • Model Developers: Neural Magic

Quantized version of ibm-granite/granite-3.1-8b-instruct. It achieves an average score of 69.81 on the OpenLLM benchmark (version 1), whereas the unquantized model achieves 70.30.

Model Optimizations

This model was obtained by quantizing the weights of ibm-granite/granite-3.1-8b-instruct to INT4 data type, ready for inference with vLLM >= 0.5.2. This optimization reduces the number of bits per parameter from 16 to 4, reducing the disk size and GPU memory requirements by approximately 75%. Only the weights of the linear operators within transformers blocks are quantized.

Deployment

Use with vLLM

This model can be deployed efficiently using the vLLM backend, as shown in the example below.

from transformers import AutoTokenizer
from vllm import LLM, SamplingParams

max_model_len, tp_size = 4096, 1
model_name = "neuralmagic/granite-3.1-8b-instruct-quantized.w4a16"
tokenizer = AutoTokenizer.from_pretrained(model_name)
llm = LLM(model=model_name, tensor_parallel_size=tp_size, max_model_len=max_model_len, trust_remote_code=True)
sampling_params = SamplingParams(temperature=0.3, max_tokens=256, stop_token_ids=[tokenizer.eos_token_id])

messages_list = [
 [{"role": "user", "content": "Who are you? Please respond in pirate speak!"}],
]

prompt_token_ids = [tokenizer.apply_chat_template(messages, add_generation_prompt=True) for messages in messages_list]

outputs = llm.generate(prompt_token_ids=prompt_token_ids, sampling_params=sampling_params)

generated_text = [output.outputs[0].text for output in outputs]
print(generated_text)

vLLM also supports OpenAI-compatible serving. See the documentation for more details.

Creation

This model was created with llm-compressor by running the code snippet below.

Evaluation

The model was evaluated on OpenLLM Leaderboard V1, OpenLLM Leaderboard V2 and on HumanEval, using the following commands:

Accuracy

Category Metric ibm-granite/granite-3.1-8b-instruct neuralmagic/granite-3.1-8b-instruct-quantized.w4a16 Recovery (%)
OpenLLM V1 ARC-Challenge (Acc-Norm, 25-shot) 66.81 66.81 100.00
GSM8K (Strict-Match, 5-shot) 64.52 65.66 101.77
HellaSwag (Acc-Norm, 10-shot) 84.18 83.62 99.33
MMLU (Acc, 5-shot) 65.52 64.25 98.06
TruthfulQA (MC2, 0-shot) 60.57 60.17 99.34
Winogrande (Acc, 5-shot) 80.19 78.37 97.73
Average Score 70.30 69.81 99.31
OpenLLM V2 IFEval (Inst Level Strict Acc, 0-shot) 74.01 73.14 98.82
BBH (Acc-Norm, 3-shot) 53.19 51.52 96.86
Math-Hard (Exact-Match, 4-shot) 14.77 16.66 112.81
GPQA (Acc-Norm, 0-shot) 31.76 29.91 94.17
MUSR (Acc-Norm, 0-shot) 46.01 45.75 99.44
MMLU-Pro (Acc, 5-shot) 35.81 34.23 95.59
Average Score 42.61 41.87 98.26
Coding HumanEval Pass@1 71.00 70.50 99.30

Inference Performance

This model achieves up to 2.7x speedup in single-stream deployment and up to 1.5x speedup in multi-stream asynchronous deployment, depending on hardware and use-case scenario. The following performance benchmarks were conducted with vLLM version 0.6.6.post1, and GuideLLM.

Single-stream performance (measured with vLLM version 0.6.6.post1)

Latency (s)
GPU class Model Speedup Code Completion
prefill: 256 tokens
decode: 1024 tokens
Docstring Generation
prefill: 768 tokens
decode: 128 tokens
Code Fixing
prefill: 1024 tokens
decode: 1024 tokens
RAG
prefill: 1024 tokens
decode: 128 tokens
Instruction Following
prefill: 256 tokens
decode: 128 tokens
Multi-turn Chat
prefill: 512 tokens
decode: 256 tokens
Large Summarization
prefill: 4096 tokens
decode: 512 tokens
A5000 granite-3.1-8b-instruct 28.3 3.7 28.8 3.8 3.6 7.2 15.7
granite-3.1-8b-instruct-quantized.w8a8 1.60 17.7 2.3 18.0 2.4 2.2 4.5 10.0
granite-3.1-8b-instruct-quantized.w4a16
(this model)
2.61 10.3 1.5 10.7 1.5 1.3 2.7 6.6
A6000 granite-3.1-8b-instruct 25.8 3.4 26.2 3.4 3.3 6.5 14.2
granite-3.1-8b-instruct-quantized.w8a8 1.50 17.4 2.3 16.9 2.2 2.2 4.4 9.8
granite-3.1-8b-instruct-quantized.w4a16
(this model)
2.48 10.0 1.4 10.4 1.5 1.3 2.5 6.2
A100 granite-3.1-8b-instruct 13.6 1.8 13.7 1.8 1.7 3.4 7.3
granite-3.1-8b-instruct-quantized.w8a8 1.31 10.4 1.3 10.5 1.4 1.3 2.6 5.6
granite-3.1-8b-instruct-quantized.w4a16
(this model)
1.80 7.3 1.0 7.4 1.0 0.9 1.9 4.3
L40 granite-3.1-8b-instruct 25.1 3.2 25.3 3.2 3.2 6.3 13.4
granite-3.1-8b-instruct-FP8-dynamic 1.47 16.8 2.2 17.1 2.2 2.1 4.2 9.3
granite-3.1-8b-instruct-quantized.w4a16
(this model)
2.72 8.9 1.2 9.2 1.2 1.1 2.3 5.3

Multi-stream asynchronous performance (measured with vLLM version 0.6.6.post1)

Maximum Throughput (Queries per Second)
GPU class Model Speedup Code Completion
prefill: 256 tokens
decode: 1024 tokens
Docstring Generation
prefill: 768 tokens
decode: 128 tokens
Code Fixing
prefill: 1024 tokens
decode: 1024 tokens
RAG
prefill: 1024 tokens
decode: 128 tokens
Instruction Following
prefill: 256 tokens
decode: 128 tokens
Multi-turn Chat
prefill: 512 tokens
decode: 256 tokens
Large Summarization
prefill: 4096 tokens
decode: 512 tokens
A5000 granite-3.1-8b-instruct 0.8 3.1 0.4 2.5 6.7 2.7 0.3
granite-3.1-8b-instruct-quantized.w8a8 1.71 1.3 5.2 0.9 4.0 10.5 4.4 0.5
granite-3.1-8b-instruct-quantized.w4a16
(this model)
1.46 1.3 3.9 0.8 2.9 8.2 3.6 0.5
A6000 granite-3.1-8b-instruct 1.3 5.1 0.9 4.0 0.3 4.3 0.6
granite-3.1-8b-instruct-quantized.w8a8 1.39 1.8 7.0 1.3 5.6 14.0 6.3 0.8
granite-3.1-8b-instruct-quantized.w4a16
(this model)
1.09 1.9 4.8 1.0 3.8 10.0 5.0 0.6
A100 granite-3.1-8b-instruct 3.1 10.7 2.1 8.5 20.6 9.6 1.4
granite-3.1-8b-instruct-quantized.w8a8 1.23 3.8 14.2 2.1 11.4 25.9 12.1 1.7
granite-3.1-8b-instruct-quantized.w4a16
(this model)
0.96 3.4 9.0 2.6 7.2 18.0 8.8 1.3
L40 granite-3.1-8b-instruct 1.4 7.8 1.1 6.2 15.5 6.0 0.7
granite-3.1-8b-instruct-FP8-dynamic 1.12 2.1 7.4 1.3 5.9 15.3 6.9 0.8
granite-3.1-8b-instruct-quantized.w4a16
(this model)
1.29 2.4 8.9 1.4 7.1 17.8 7.8 1.0
Downloads last month
993
Safetensors
Model size
8B params
Tensor type
I64
·
I32
·
BF16
·

Model tree for RedHatAI/granite-3.1-8b-instruct-quantized.w4a16

Quantized
(30)
this model

Collections including RedHatAI/granite-3.1-8b-instruct-quantized.w4a16