Optimum 文档

概述

您正在查看 main 版本,该版本需要从源代码安装. 如果您想要常规 pip 安装,请查看最新的稳定版本 (v1.24.0)。
Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

概述

🤗 Optimum 处理 PyTorch 或 TensorFlow 模型到 ONNX 的导出,模块为 exporters.onnx。它提供了类、函数和命令行界面,以便轻松执行导出。

🤗 Transformers 支持的架构

  • AST
  • Audio Spectrogram Transformer
  • Albert
  • Bart
  • Beit
  • Bert
  • BlenderBot
  • BlenderBotSmall
  • Bloom
  • Camembert
  • CLIP
  • CodeGen
  • ConvBert
  • ConvNext
  • ConvNextV2
  • Data2VecAudio
  • Data2VecText
  • Data2VecVision
  • Deberta
  • Deberta-v2
  • Decision Transformer
  • Deit
  • Detr
  • DINOv2
  • DistilBert
  • Donut-Swin
  • Electra
  • Encoder Decoder
  • ESM
  • Falcon
  • Flaubert
  • GPT-2
  • GPT-BigCode
  • GPT-J
  • GPT-Neo
  • GPT-NeoX
  • OPT
  • GroupVit
  • Hiera
  • Hubert
  • IBert
  • LayoutLM
  • LayoutLM-v3
  • Lilt
  • Levit
  • LongT5
  • Llama
  • M2-M100
  • Marian
  • MarkupLM
  • MaskFormer
  • MBart
  • MGP-STR
  • Mistral
  • MobileBert
  • MobileVit
  • MobileNet v1
  • MobileNet v2
  • ModernBert
  • MPNet
  • MT5
  • Musicgen (仅限文本条件)
  • Nystromformer
  • OLMo
  • OLMo2
  • OWL-ViT
  • PatchTST
  • PatchTSMixer
  • Pegasus
  • Perceiver
  • Phi
  • Phi3
  • Pix2Struct
  • PoolFormer
  • PVT
  • Qwen2(Qwen1.5)
  • RegNet
  • RemBERT
  • ResNet
  • Roberta
  • Roformer
  • SAM
  • Segformer
  • SEW
  • SEW-D
  • Speech2Text
  • SigLIP
  • SpeechT5
  • Splinter
  • SqueezeBert
  • Swin
  • SwinV2
  • T5
  • Table Transformer
  • TROCR
  • UniSpeech
  • UniSpeech SAT
  • Vision Encoder Decoder
  • Vit
  • VitMAE
  • VitMSN
  • Wav2Vec2
  • Wav2Vec2 Conformer
  • WavLM
  • Whisper
  • XLM
  • XLM-Roberta
  • Yolos

🤗 Diffusers 支持的架构

  • Stable Diffusion

🤗 Timm 支持的架构

  • Adversarial Inception v3
  • AdvProp (EfficientNet)
  • Big Transfer (BiT)
  • CSP-DarkNet
  • CSP-ResNet
  • CSP-ResNeXt
  • DenseNet
  • Deep Layer Aggregation
  • Dual Path Network (DPN)
  • ECA-ResNet
  • EfficientNet
  • EfficientNet (Knapsack Pruned)
  • Ensemble Adversarial Inception ResNet v2
  • ESE-VoVNet (静态形状的部分支持)
  • FBNet
  • (Gluon) Inception v3
  • (Gluon) ResNet
  • (Gluon) ResNeXt
  • (Gluon) SENet
  • (Gluon) SE-ResNeXt
  • (Gluon) Xception
  • HRNet
  • Instagram ResNeXt WSL
  • Inception ResNet v2
  • Inception v3
  • Inception v4
  • (Legacy) SE-ResNet
  • (Legacy) SE-ResNeXt
  • (Legacy) SENet
  • MixNet
  • MnasNet
  • MobileNet v2
  • MobileNet v3
  • NASNet
  • Noisy Student (EfficientNet)
  • PNASNet
  • RegNetX
  • RegNetY
  • Res2Net
  • Res2NeXt
  • ResNeSt
  • ResNet
  • ResNet-D
  • ResNeXt
  • RexNet
  • SE-ResNet
  • SelecSLS
  • SE-ResNeXt
  • SK-ResNet
  • SK-ResNeXt
  • SPNASNet
  • SSL ResNet
  • SWSL ResNet
  • SWSL ResNeXt
  • (Tensorflow) EfficientNet
  • (Tensorflow) EfficientNet CondConv
  • (Tensorflow) EfficientNet Lite
  • (Tensorflow) Inception v3
  • (Tensorflow) MixNet
  • (Tensorflow) MobileNet v3
  • TResNet
  • Wide ResNet
  • Xception

Sentence Transformers 支持的架构

  • 所有基于 Transformer 和 CLIP 的模型。
< > 更新 在 GitHub 上