Commit Graph

10 Commits

Author SHA1 Message Date
Kohya S
4b2283491e feat: add VAE chunking and caching options to reduce memory usage 2026-02-11 21:32:00 +09:00
Kohya S
b67cc5a457 feat: update Anima SAI model spec metadata handling 2026-02-11 14:56:23 +09:00
Kohya S
90725eba64 feat: support LORA for Qwen3 2026-02-11 14:21:33 +09:00
Kohya S
dbb40ae4c0 fix: wait all block trasfer before siwtching offloader mode 2026-02-10 21:46:58 +09:00
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