Optimum documentation

Overview

You are viewing v1.14.0 version. A newer version v1.23.3 is available.
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Overview

🤗 Optimum handles the export of PyTorch or TensorFlow models to ONNX in the exporters.onnx module. It provides classes, functions, and a command line interface to perform the export easily.

Supported architectures:

  • AST
  • Audio Spectrogram Transformer
  • Albert
  • Bart
  • Beit
  • Bert
  • BlenderBot
  • BlenderBotSmall
  • Bloom
  • Camembert
  • CLIP
  • CodeGen
  • ConvBert
  • ConvNext
  • Data2VecAudio
  • Data2VecText
  • Data2VecVision
  • Deberta
  • Deberta-v2
  • Deit
  • Detr
  • DistilBert
  • Donut-Swin
  • Electra
  • Encoder Decoder
  • Falcon
  • Flaubert
  • GPT-2
  • GPT-BigCode
  • GPT-J
  • GPT-Neo
  • GPT-NeoX
  • OPT
  • GroupVit
  • Hubert
  • IBert
  • LayoutLM
  • LayoutLM-v3
  • Lilt
  • Levit
  • LongT5
  • Llama
  • M2-M100
  • Marian
  • MBart
  • Mistral
  • MobileBert
  • MobileVit
  • MobileNet v1
  • MobileNet v2
  • MPNet
  • MT5
  • Nystromformer
  • OWL-ViT
  • Pegasus
  • Perceiver
  • Pix2Struct
  • PoolFormer
  • RegNet
  • ResNet
  • Roberta
  • Roformer
  • SAM
  • Segformer
  • SEW
  • SEW-D
  • Speech2Text
  • SpeechT5
  • Splinter
  • SqueezeBert
  • Stable Diffusion
  • Swin
  • T5
  • TROCR
  • UniSpeech
  • UniSpeech SAT
  • Vision Encoder Decoder
  • Vit
  • Wav2Vec2
  • Wav2Vec2 Conformer
  • WavLM
  • Whisper
  • XLM
  • XLM-Roberta
  • Yolos

Supported architectures (Timm):

  • Resnext50-32x4d
  • Resnext50d-32x4d
  • Resnext101-32x4d
  • Resnext101-32x8d
  • Resnext101-64x4d