Skip to content
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

aivmファイルの読み込み機能追加 #189

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

yamichi77
Copy link

お世話になります。
機能追加のPRになりますので、レビューいただければ幸いです。

追加機能

先週公開された AivisSpeech で使用されるaivmファイル(safetensorsベース)を読み込む機能の追加

aivmファイルについて

aivmlib のリポジトリ上に説明がありますが、safetensorsのヘッダーにスタイルベクトルやハイパーパラメータを入れ1つのファイルにした拡張子のようです。このPRではaivmlibから必要な情報を取得するようにしております。

変更点

tts_model.pyファイルへaivmファイル読み込み機能の追加

  • スタイルベクトルやハイパーパラメータを読み込みため、依存関係にaivmlibを追加しております。

@litagin02
Copy link
Owner

PRありがとうございます。ありがたいです。
現在、aivm形式の対応やexport等は(AivisProjectとこのSBV2リポジトリとの棲み分けを含めて)現在AivisProject様と相談してどうするかを検討中です。
devブランチにてonnx形式の対応を行っておりますので、もし読み込み可能とした場合はaivmだけでなくaivmx形式へも対応させる予定であります。
申し訳ありませんが一旦保留させていただいて、aivm(x)形式に対応することが決定した場合は、このプルリクをベースに対応をさせていただくことにします。

@yamichi77
Copy link
Author

対応についてはすでに相談中でしたか。すみません、余計な対応だったようです。
aivm関連以外に、onnx形式の対応も進んでいるとのことなので、今後も楽しみにさせていただきます!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants