目录

Transformer 项目

安装&测试

按照 transformers 文档说明安装 python 以及依赖的库文件。最后运行以及下语句来测试。

python3 -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('we love you'))"

注意:

  1. 存在两个 python 和 python3 的话要注意用第二个
  2. pip 以及 pip3 命令也要更新 pip install --upgrade pip (目前version 23 以上是OK的)
  3. model 会被安装到路径 ~/.cache/huggingface/hub/...

编译ONNX

文档 下面例子会自动下载模型到系统的cache,然后编译到指定路径

optimum-cli export onnx --model facebook/m2m100_418M ~/openai/models/m2m100_418M/