Optimum documentation

Overview

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 TensorFlow models to TFLite in the exporters.tflite module. In addition, models hosted on the Hugging Face Hub with PyTorch weights but having a TensorFlow implementation will also be supported in the export thanks to Transformers’ TFPreTrainedModel.from_pretrained() auto-conversion to TensorFlow.

The TFLite export support provides classes, functions and a command line interface to export a model easily.

Supported architectures:

  • Albert
  • BERT
  • Camembert
  • ConvBert
  • Deberta
  • Deberta V2
  • DistilBert
  • Electra
  • Flaubert
  • MobileBert
  • MPNet
  • ResNet
  • Roberta
  • RoFormer
  • XLM
  • XLMRoberta
< > Update on GitHub