Transformer 项目
安装&测试
按照 transformers 文档说明安装 python 以及依赖的库文件。最后运行以及下语句来测试。
python3 -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('we love you'))"
注意:
- 存在两个 python 和 python3 的话要注意用第二个
- pip 以及 pip3 命令也要更新
pip install --upgrade pip
(目前version 23 以上是OK的) - model 会被安装到路径
~/.cache/huggingface/hub/...
编译ONNX
文档 下面例子会自动下载模型到系统的cache,然后编译到指定路径
optimum-cli export onnx --model facebook/m2m100_418M ~/openai/models/m2m100_418M/