概述
🤗 Optimum 在 exporters.tflite 模块中处理 TensorFlow 模型到 TFLite 的导出。此外，托管在 Hugging Face Hub 上并具有 PyTorch 权重但具有 TensorFlow 实现的模型也将通过 Transformers’ 的 TFPreTrainedModel.from_pretrained() 自动转换为 TensorFlow 而获得支持。
TFLite 导出支持提供类、函数和命令行界面，以便轻松导出模型。
支持的架构
- Albert
- BERT
- Camembert
- ConvBert
- Deberta
- Deberta V2
- DistilBert
- Electra
- Flaubert
- MobileBert
- MPNet
- ResNet
- Roberta
- RoFormer
- XLM
- XLMRoberta