starcoder2-3b-int4-ov
- Model creator: BigCode
- Original model: bigcode/starcoder2-3b
Description
This is bigcode/starcoder2-3b model converted to the OpenVINO™ IR (Intermediate Representation) format with weights compressed to INT4 by NNCF.
Quantization Parameters
Weight compression was performed using nncf.compress_weights
with the following parameters:
- mode: INT4_SYM
- group_size: 128
- ratio: 1.0
For more information on quantization, check the OpenVINO model optimization guide.
Compatibility
The provided OpenVINO™ IR model is compatible with:
- OpenVINO version 2024.2.0 and higher
- Optimum Intel 1.17.0 and higher
Running Model Inference with Optimum Intel
- Install packages required for using Optimum Intel integration with the OpenVINO backend:
pip install optimum[openvino]
- Run model inference:
from transformers import AutoTokenizer
from optimum.intel.openvino import OVModelForCausalLM
model_id = "OpenVINO/starcoder2-3b-int4-ov"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = OVModelForCausalLM.from_pretrained(model_id)
inputs = tokenizer("What is OpenVINO?", return_tensors="pt")
outputs = model.generate(**inputs, max_length=200)
text = tokenizer.batch_decode(outputs)[0]
print(text)
For more examples and possible optimizations, refer to the OpenVINO Large Language Model Inference Guide.
Running Model Inference with OpenVINO GenAI
- Install packages required for using OpenVINO GenAI.
pip install openvino-genai huggingface_hub
- Download model from HuggingFace Hub
import huggingface_hub as hf_hub
model_id = "OpenVINO/starcoder2-3b-int4-ov"
model_path = "starcoder2-3b-int4-ov"
hf_hub.snapshot_download(model_id, local_dir=model_path)
- Run model inference:
import openvino_genai as ov_genai
device = "CPU"
pipe = ov_genai.LLMPipeline(model_path, device)
print(pipe.generate("def print_hello_world():", max_length=200))
More GenAI usage examples can be found in OpenVINO GenAI library docs and samples
For more examples and possible optimizations, refer to the OpenVINO Large Language Model Inference Guide.
Legal information
The original model is distributed under bigcode-openrail-m license. More details can be found in bigcode/starcoder2-3b.
Disclaimer
Intel is committed to respecting human rights and avoiding causing or contributing to adverse impacts on human rights. See Intel’s Global Human Rights Principles. Intel’s products and software are intended only to be used in applications that do not cause or contribute to adverse impacts on human rights.
- Downloads last month
- 37