Skip to content

Latest commit

 

History

History
160 lines (154 loc) · 6.44 KB

SUPPORTED_MODELS.md

File metadata and controls

160 lines (154 loc) · 6.44 KB

OpenVINO™ GenAI: Supported Models

Architecture Models Example HuggingFace Models
ChatGLMModel ChatGLM
GemmaForCausalLM Gemma
GPTNeoXForCausalLM Dolly
RedPajama
LlamaForCausalLM Llama 3
Llama 2
OpenLLaMA
TinyLlama
MistralForCausalLM Mistral
Notus
Zephyr
PhiForCausalLM Phi
QWenLMHeadModel Qwen

The pipeline can work with other similar topologies produced by optimum-intel with the same model signature. The model is required to have the following inputs after the conversion:

  1. input_ids contains the tokens.
  2. attention_mask is filled with 1.
  3. beam_idx selects beams.
  4. position_ids (optional) encodes a position of currently generating token in the sequence and a single logits output.

Note

Models should belong to the same family and have the same tokenizers.

Some models may require access request submission on the Hugging Face page to be downloaded.

If https://huggingface.co/ is down, the conversion step won't be able to download the models.