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

細かな修正 (#92 を参照) #95

Merged
merged 6 commits into from
Mar 13, 2024

Conversation

tsukumijima
Copy link

@tsukumijima tsukumijima commented Mar 12, 2024

前回のプルリク (#92) で漏れてしまった、下記の細かな修正を行いました。

  • open() 関数の呼び出しフォーマットを統一
  • bert_models.load_model/tokenizer() に cache_dir / revision 引数を追加
  • hatch build でビルドした sdist に余計なファイルが大量に入るのを修正

Tip

Revert されてるコミットがありますが、これは Web UI で学習を開始する際「app.py 起動時に BERT モデルが事前ロードされると VRAM が圧迫されるし明示的に音声合成された時にのみ事前ロードするようにしないといけないのでは…?」と思っていたものの、実際にはロードだけでは通常のメモリしか消費せず VRAM の節約にならない事が分かったためです…。

…consumption during training in the Web UI

Since the BERT features of the dataset are pre-extracted by bert_gen.py, there is no need to load the BERT model at training time.
Include in sdist only the minimum required files for style-bert-vits2 as a library.
…PU VRAM consumption during training in the Web UI"

This reverts commit e8a76e5.
@litagin02
Copy link
Owner

ありがとうございます!
問題なさそうなのでマージします。

@litagin02 litagin02 merged commit 30d9f2e into litagin02:dev-refactor Mar 13, 2024
@litagin02
Copy link
Owner

Revertされていた
9e52226
ですが、VRAMの問題というよりも、学習やデータセットづくりのために立ち上げたときのも不要なBERTモデルロードの処理が走り単純に時間がかかってストレスだなあと思っていたので、取り入れることにします。

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