We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d5a6169 commit d0f55e8Copy full SHA for d0f55e8
1 file changed
.github/workflows/java-app.yml
@@ -24,11 +24,11 @@ jobs:
24
- name: Install dependencies
25
run: |
26
python -m pip install --upgrade pip
27
- pip install -q transformers[onnx] transformers[sentencepiece] torch
+ pip install optimum-onnx onnx onnxruntime -q
28
- name: Hugging Face Model to ONNX
29
30
# Convert Hugging Face Model to ONNX
31
- python -m transformers.onnx --opset 16 --atol 0.005 --feature=token-classification --model=xlm-roberta-large-finetuned-conll03-english onnx_model/
+ optimum-cli export onnx --opset 18 --atol 0.005 \ --task=token-classification \ --model=xlm-roberta-large-finetuned-conll03-english onnx/
32
33
# Upload the ONNX model as an artifact
34
- name: Upload ONNX model
0 commit comments