Commit Graph

11 Commits

Author SHA1 Message Date
Kohya S.
57aa70ea9e Update library/anima_train_utils.py
Remove unused import.

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
2026-02-12 08:02:05 +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
a7cd38dcaf feat: split CFG processing in do_sample function to reduce memory usage 2026-02-11 18:00:10 +09:00
Kohya S
9349c91c89 fix: remove unused code 2026-02-11 17:23:58 +09:00
Kohya S
b67cc5a457 feat: update Anima SAI model spec metadata handling 2026-02-11 14:56:23 +09:00
Kohya S
dbb40ae4c0 fix: wait all block trasfer before siwtching offloader mode 2026-02-10 21:46:58 +09:00
Kohya S
6d08c93b23 feat: enhance block swap functionality for inference and training in Anima model 2026-02-10 21:26:50 +09:00
Kohya S
0f413974b7 feat: support LoRA format without net. prefix 2026-02-09 23:21:04 +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
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