michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

added the warning suppresion when the workflow runs, instead of as an env variable

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-27 15:49:00 -04:00
parent 3722078b2d
commit f8793ba2a3
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.riahub/workflows/workflow.yaml
View File

@ -66,12 +66,10 @@ jobs:
path: checkpoint_files/inference_recognition_model.ckpt path: checkpoint_files/inference_recognition_model.ckpt
- name: 3. Convert to ONNX file - name: 3. Convert to ONNX file
env:
MKL_DISABLE_FAST_MM=1
run: | run: |
PYTHONPATH=. python onnx_scripts/convert_to_onnx.py MKL_DISABLE_FAST_MM=1 PYTHONPATH=. python onnx_scripts/convert_to_onnx.py
echo "building inference app" echo "building inference app"
- name: Upload ONNX file - name: Upload ONNX file
uses: actions/upload-artifact@v3 uses: actions/upload-artifact@v3
with: with:
@ -88,7 +86,7 @@ jobs:
name: profile-data name: profile-data
path: '**/onnxruntime_profile_*.json' path: '**/onnxruntime_profile_*.json'
- name: 4. Convert to ORT file - name: 5. Convert to ORT file
run: | run: |
python -m onnxruntime.tools.convert_onnx_models_to_ort \ python -m onnxruntime.tools.convert_onnx_models_to_ort \
/workspace/qoherent/modrec-workflow/onnx_files/inference_recognition_model.onnx \ /workspace/qoherent/modrec-workflow/onnx_files/inference_recognition_model.onnx \