差别
| 后一修订版 | 前一修订版 |
| it:ai:python [2023-08-20 07:12] – 创建 goldentianya | it:ai:python [2023-08-20 11:31] (当前版本) – [编译ONNX] goldentianya |
|---|
| ====== Transformer 项目 ====== | ====== Transformer 项目 ====== |
| |
| 按照 [[https://huggingface.co/docs/transformers/installation|transformers]] 文档说明安装 python 以及依赖的库文件。最后运行 <color #ff7f27>''%%python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('we love you'))"%%''</color> 来测试。 | ===== 安装&测试 ===== |
| | |
| | 按照 [[https://huggingface.co/docs/transformers/installation|transformers]] 文档说明安装 python 以及依赖的库文件。最后运行以及下语句来测试。 |
| | 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/ |
| | |
