sylwia-kuros's picture
Create README.md
bb2fe88 verified
|
raw
history blame
2.1 kB
metadata
license: mit

starcoder2-7b-fp16-ov

Description

This is bigcode/starcoder2-7b model converted to the OpenVINO™ IR (Intermediate Representation) format.

Compatibility

The provided OpenVINO™ IR model is compatible with:

  • OpenVINO version 2024.1.0 and higher
  • Optimum Intel 1.16.0 and higher

Running Model Inference

  1. Install packages required for using Optimum Intel integration with the OpenVINO backend:
pip install optimum[openvino]
  1. Run model inference:
from transformers import AutoTokenizer
from optimum.intel.openvino import OVModelForCausalLM

model_id = "OpenVINO/starcoder2-7b-fp16-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.

Legal information

The original model is distributed under bigcode-openrail-m license. More details can be found in bigcode/starcoder2-7b.

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.