-
Notifications
You must be signed in to change notification settings - Fork 5
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support onnx #1
Support onnx #1
Conversation
良いですね!! このリポジトリにプルリクエストを送る配慮もとてもありがたいです。 Cythonの動作確認はC++版実装ができてからになると思います。 |
positional encodingは動的生成させました
👍 Cythonの件はここではパスします |
https://github.com/Hiroshiba/voicevox_core/pull/34 こちらのREADMEではあまり言及していませんがonnx変換方法がこっちで書かれていた方が良いのでマージお願いします。 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!!!
related: https://github.com/Hiroshiba/voicevox_engine/issues/69
TODO
Description
python run.py --yukarin_s_model_dir "model/yukarin_s" --yukarin_sa_model_dir "model/yukarin_sa" --yukarin_sosoa_model_dir "model/yukarin_sosoa" --hifigan_model_dir "model/hifigan" --speaker_ids 5 --method=convert
でonnxへの変換が可能。modelフォルダのyukarin_s, yukarin_sa, yukarin_sosoaにonnxが保存されるyukarin_sosoaにはhifi_ganと合わせた
decode.onnx
が保存されるonnxで実行したい場合は
--method=onnx
とする;python run.py --yukarin_s_model_dir "model/yukarin_s" --yukarin_sa_model_dir "model/yukarin_sa" --yukarin_sosoa_model_dir "model/yukarin_sosoa" --hifigan_model_dir "model/hifigan" --speaker_ids 5 --method=onnx
おそらくtorchをimportしていない
テストの結果波形の相対誤差が1e-3くらいになったがもっと小さくできるかは不明。decodeで何故か誤差が出てきてしまう
実際の重みで聞いてみる必要がある
テストコマンド:
python test.py --yukarin_s_model_dir "model/yukarin_s" --yukarin_sa_model_dir "model/yukarin_sa" --yukarin_sosoa_model_dir "model/yukarin_sosoa" --hifigan_model_dir "model/hifigan" --speaker_ids 6 --texts "おはようございます、こんにちは、こんばんは"