Support V210 train and fix some bugs and improve

This commit is contained in:
litagin02
2023-12-23 12:59:57 +09:00
parent 3090f5837d
commit 6c171fcea0
8 changed files with 1365 additions and 900 deletions

View File

@@ -1,3 +1,17 @@
# 変更点
- `keep_ckpts`のバグを修正
- 圧縮モデルを保存する設定を追加: `config.json``save_compressed_models``true`にすると、modelの保存時に圧縮モデルも保存されるこれと`keep_ckpts``1`とかにしとけばかなり容量の節約に)
- Ver 2.1での学習をサポート(`train_ms_V210.py`
# TODO
- [ ] Ver 2.2での学習をサポート
- [ ] Ver 2.1, 2.2での学習でのbf16対応
- [ ] 推論のWebUIでのバージョンに応じた感情指定のサポート
- [ ] より良い推論WebUI
- [ ] 学習のWebUI?
以下本家のREADME.md
<div align="center">
<img alt="LOGO" src="https://cdn.jsdelivr.net/gh/fishaudio/fish-diffusion@main/images/logo_512x512.png" width="256" height="256" />