ai-toolkit/extensions_built_in/diffusion_models/hidream
Jaret Burkett 5f04ae7ad5 Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning. 2026-06-25 13:19:38 -06:00
..
src Fixed issue with hidream-01 that could cause a weird nan state. Took forever to track down as it was 1 in 10 starts. 2026-05-28 12:48:24 -06:00
__init__.py Add support for fine-tuning Hidream O1 (#831) 2026-05-12 11:15:16 -06:00
hidream_e1_model.py Added initial support for Hidream E1 training 2025-07-27 15:12:56 -06:00
hidream_model.py Added initial support for Hidream E1 training 2025-07-27 15:12:56 -06:00
hidream_o1_model.py Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning. 2026-06-25 13:19:38 -06:00