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