Commit Graph

6 Commits

Author SHA1 Message Date
Kohya S
02a75944b3 fix: keep latents 4D except DiT call 2026-02-10 21:26:20 +09:00
Kohya S
58db77a488 fix: update to work fp8_scaled option 2026-02-09 23:31:33 +09:00
kohya-ss
8f5b298906 feat: simplify Anima training script and update empty caption handling 2026-02-09 12:46:53 +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