概述

🤗 Optimum 处理 TensorFlow 模型到 exporters.tflite 模块中 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