Style-Bert-ViTS2のモデルをONNX化する方法

AivisSpeech Engin を使いたい方へ

こんにちは！葉加瀬あいです。
前回こちらについて少し解説したので、開発者の方向けにその補足をしたいと思います！

なお、要するにこちらでONNXモデルが作成できないときの対処法です。

Style-Bert-ViTS2をONNX化する方法

Style-Bert-ViTS2をONNX形式に変換する手順を紹介します。以下の手順に従って、スムーズに作業を進めてください！

1.　まず、Style-Bert-ViTS2のフォルダに移動します。

cd 【Style-Bert-VITS2のフォルダ】

2.　次に、開発ブランチに切り替えます。

git checkout dev

3.　最新の変更を取得します。

git pull origin dev

4.　`Update-Style-Bert-VITS2.bat`を実行します。この際、onnxruntime-gpuなどの必要なパッケージがインストールされていることを確認してください。

5.　再度、Style-Bert-ViTS2のフォルダに移動します。

cd 【Style-Bert-VITS2のフォルダ】

6.　仮想環境をアクティブにします。

venv\Scripts\activate

7.　最後に、以下のコマンドを実行してONNX形式に変換します。

python convert_onnx.py --model 【SVBS2のモデルフォルダ】

実行しますとこんな感じで作成されます！

参考リンク: Style-Bert-VITS2 GitHub