tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add: Test for ONNX inference code (without PyTorch dependency)

Browse Source
This commit is contained in:
tsukumi
2024-09-23 10:04:02 +09:00
parent 5a82105df3
commit eb5d245903
6 changed files with 104 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
convert_onnx.py
View File

@@ -1,5 +1,5 @@
# usage: .venv/bin/python convert_onnx.py --model model_assets/koharune-ami/koharune-ami.safetensors
# usage: .venv/bin/python convert_onnx.py --model model_assets/ (All models in the directory will be converted)
# Usage: .venv/bin/python convert_onnx.py --model model_assets/koharune-ami/koharune-ami.safetensors
# Usage: .venv/bin/python convert_onnx.py --model model_assets/ (All models in the directory will be converted)
# ref: https://github.com/tuna2134/sbv2-api/blob/main/convert/convert_model.py
import time