Overview

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

Last updated