from transformers import AutoTokenizer
from optimum.intel.openvino import OVModelForCausalLM

model_id = "OpenVINO/qwen2-1.5b-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 Inference with Optimum Intel.

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/qwen2-1.5b-fp16-ov"
model_path = "qwen2-1.5b-fp16-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("What is OpenVINO?", max_length=200))

More GenAI usage examples can be found in OpenVINO GenAI library docs and samples

You can find more detaild usage examples in OpenVINO Notebooks:

Limitations

Check the original model card for limitations.

Legal information

The original model is distributed under Apache License Version 2.0 license. More details can be found in Qwen2-1.5B.

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.

OpenVINO/Qwen2-1.5B-fp16-ov

Qwen2-1.5B-fp16-ov

Description

Compatibility

Running Model Inference with Optimum Intel

Running Model Inference with OpenVINO GenAI

Limitations

Legal information

Disclaimer

OpenVINO/
Qwen2-1.5B-fp16-ov