将 Huggingface 模型转换为 CoreML 的脚本和信息

拥抱脸到 CoreML

此存储库包含将 Huggingface 模型转换为 CoreML 的脚本和信息。

信息

  • 要转换的模型:“facebook/blenderbot-400M-distill”使用示例通过在命令行中执行来尝试$ python3 try.py

链接

初始步骤

  1. 克隆回购:$ git clone https://github.com/madcato/huggingface-coreml.git
  2. 执行聊天机器人下载模型,$ python3 try.py
  3. 安装 huggingface exporters 子模块, $ git submodule init && git submodule update
  4. $ cd exporters && pip3 install -e . && cd ..
  5. python3 -m exporters.coreml --model=facebook/blenderbot-400M-distill exported/ --feature=text2text-generation

最后一步将在文件夹mlpackage中创建一个文件exported这个文件可以用 Xcode 打开,模型可以在 playground 或项目中进行测试。

出口商特点

  • '特征提取',
  • '过去的特征提取',
  • '填充掩码',
  • '图像分类',
  • '蒙面人',
  • '多项选择',
  • '下一句预测',
  • '物体检测',
  • “问答”,
  • '语义分割',
  • '文本分类',
  • '文本生成',
  • '过去的文本生成',
  • 'text2text-generation',
  • '令牌分类',
  • '序列分类',
  • '因果-lm',
  • '因果关系 lm-with-past',
  • 'seq2seq-lm',
  • 'seq2seq-lm-with-past',
  • 'speech2seq-lm',
  • 'speech2seq-lm-with-past',
  • '蒙面 lm',
  • 'vision2seq-lm',
  • '默认',
  • '默认与过去',
  • '自动语音识别',
  • 'CTC'

GitHub

查看 Github