差别
两侧同时换到之前的修订记录前一修订版后一修订版 | 前一修订版 |
it:ai:python [2023-08-20 07:13] – goldentianya | it:ai:python [2023-08-20 11:31] (当前版本) – [编译ONNX] goldentianya |
---|
| |
按照 [[https://huggingface.co/docs/transformers/installation|transformers]] 文档说明安装 python 以及依赖的库文件。最后运行以及下语句来测试。 | 按照 [[https://huggingface.co/docs/transformers/installation|transformers]] 文档说明安装 python 以及依赖的库文件。最后运行以及下语句来测试。 |
python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('we love you'))" | python3 -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('we love you'))" |
| |
注意: | 注意: |
- pip 以及 pip3 命令也要更新 <color #ff7f27>''%%pip install --upgrade pip%%''</color> (目前version 23 以上是OK的) | - pip 以及 pip3 命令也要更新 <color #ff7f27>''%%pip install --upgrade pip%%''</color> (目前version 23 以上是OK的) |
- model 会被安装到路径 <color #ff7f27>''%%~/.cache/huggingface/hub/...%%''</color> | - model 会被安装到路径 <color #ff7f27>''%%~/.cache/huggingface/hub/...%%''</color> |
| |
| ===== 编译ONNX ===== |
| [[https://huggingface.co/docs/transformers/serialization|文档]] {{fa>book}} 下面例子会自动下载模型到系统的cache,然后编译到指定路径 |
| |
| optimum-cli export onnx --model facebook/m2m100_418M ~/openai/models/m2m100_418M/ |
| |
回到顶部