Commit Graph

8 Commits

Author SHA1 Message Date
Kohya S
cc1f57bc70 fix: review with Copilot 2026-02-12 08:05:41 +09:00
Kohya S.
3612bedda6 Update anima_train.py
Remove unused import.

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
2026-02-12 08:01:38 +09:00
Kohya S
4b2283491e feat: add VAE chunking and caching options to reduce memory usage 2026-02-11 21:32:00 +09:00
Kohya S
8d2d286a13 feat: refactor Anima training script 2026-02-10 21:27:22 +09:00
kohya-ss
06dcb30016 fix: update argument names in anima_train_utils to align with other archtectures 2026-02-09 12:46:04 +09:00
kohya-ss
4f6511bf28 feat: kept caption dropout rate in cache and handle in training script 2026-02-08 15:35:53 +09:00
kohya-ss
d992037984 format: format 2026-02-08 12:22:54 +09:00
duongve13112002
e21a7736f8 Support Anima model (#2260)
* Support Anima model

* Update document and fix bug

* Fix latent normlization

* Fix typo

* Fix cache embedding

* fix typo in tests/test_anima_cache.py

* Remove redundant argument apply_t5_attn_mask

* Improving caching with argument caption_dropout_rate

* Fix W&B logging bugs

* Fix discrete_flow_shift default value
2026-02-08 10:18:55 +09:00