概述

🤗 Optimum 在 exporters.tflite 模块中处理 TensorFlow 模型到 TFLite 的导出。此外，由于 Transformers 的 TFPreTrainedModel.from_pretrained() 自动转换为 TensorFlow，托管在 Hugging Face Hub 上且具有 PyTorch 权重但具有 TensorFlow 实现的模型也将支持导出。

TFLite 导出支持提供了类、函数和命令行界面，以便轻松导出模型。

支持的架构

• Albert
• BERT
• Camembert
• ConvBert
• Deberta
• Deberta V2
• DistilBert
• Electra
• Flaubert
• MobileBert
• MPNet
• ResNet
• Roberta
• RoFormer
• XLM
• XLMRoberta