Skip to content

Commit d0f55e8

Browse files
authored
ci: update CI pipeline
as model conversion library changed update CI pipeline code
1 parent d5a6169 commit d0f55e8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

.github/workflows/java-app.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,11 @@ jobs:
2424
- name: Install dependencies
2525
run: |
2626
python -m pip install --upgrade pip
27-
pip install -q transformers[onnx] transformers[sentencepiece] torch
27+
pip install optimum-onnx onnx onnxruntime -q
2828
- name: Hugging Face Model to ONNX
2929
run: |
3030
# 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/
31+
optimum-cli export onnx --opset 18 --atol 0.005 \ --task=token-classification \ --model=xlm-roberta-large-finetuned-conll03-english onnx/
3232
3333
# Upload the ONNX model as an artifact
3434
- name: Upload ONNX model

0 commit comments

Comments
 (0)