Skip to content

Latest commit

 

History

History
407 lines (392 loc) · 18.1 KB

SUPPORTED_MODELS.md

File metadata and controls

407 lines (392 loc) · 18.1 KB

OpenVINO™ GenAI: Supported Models

Large language 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

Note

LoRA adapters are supported.

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.

Image generation models

Architecture Text 2 image Image 2 image LoRA support Example HuggingFace Models
Latent Consistency Model Supported Supported Supported
Stable Diffusion Supported Supported Supported
Stable Diffusion XL Supported Supported Supported
Stable Diffusion 3 Supported Not supported Not supported
Flux Supported Not supported Not supported

Inpainting models

In addition to image generation models, InpaintingPipeline supports specialized inpainting models

Architecture LoRA support Example HuggingFace Models
Stable Diffusion Supported
Stable Diffusion XL Supported

Visual language models

Architecture Models LoRA support Example HuggingFace Models
InternVL2 InternVL2 Not supported
LLaVA LLaVA-v1.5 Not supported
LLaVA-NeXT LLaVa-v1.6 Not supported
MiniCPMV MiniCPM-V-2_6 Not supported

Whisper models

Architecture Models LoRA support Example HuggingFace Models
WhisperForConditionalGeneration Whisper Not supported
Distil-Whisper Not supported
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.