ai-toolkit/toolkit/models
Jaret Burkett fc83eb7691 WIP on mean flow loss. Still a WIP. 2025-06-12 08:00:51 -06:00
..
wan21 Various code to support experiments. 2025-06-09 11:19:21 -06:00
DoRA.py
LoRAFormer.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
RRDB.py
auraflow.py
base_model.py Fixed issue where sampling fails if doing a full finetune for some models 2025-05-17 19:37:55 +00:00
block.py
clip_fusion.py
clip_pre_processor.py
cogview4.py Added ability to add models to finetune as plugins. Also added flux2 new arch via that method. 2025-03-27 16:07:00 -06:00
control_lora_adapter.py Handle inpainting training for control_lora adapter 2025-03-24 13:17:47 -06:00
decorator.py Added training for an experimental decoratgor embedding. Allow for turning off guidance embedding on flux (for unreleased model). Various bug fixes and modifications 2024-12-15 08:59:27 -07:00
diffusion_feature_extraction.py Increase ema feedback amount. Normalize the dfe 4 image embeds 2025-06-10 08:01:13 -06:00
flux.py WIP on mean flow loss. Still a WIP. 2025-06-12 08:00:51 -06:00
flux_sage_attn.py Added support for full finetuning flux with randomized param activation. Examples coming soon 2024-11-21 13:05:32 -07:00
i2v_adapter.py Cleanup 2025-04-18 11:44:49 -06:00
ilora.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
ilora2.py Various experiments and minor bug fixes for edge cases 2025-04-25 13:44:38 -06:00
llm_adapter.py Various experiments and minor bug fixes for edge cases 2025-04-25 13:44:38 -06:00
lokr.py Bugfixes 2025-03-03 08:22:15 -07:00
lumina2.py DFE tweaks. Adding support for more llms as text encoders 2025-02-13 04:31:49 -07:00
pixtral_vision.py Quick patch to scope xformer imports until a better solution 2024-09-28 15:36:42 -06:00
redux.py Added support for training flux redux adapters 2024-11-21 20:01:52 -07:00
single_value_adapter.py Removed all submodules. Submodule free now, yay. 2025-04-18 10:39:15 -06:00
size_agnostic_feature_encoder.py
sref.py Avoid loading state dict for automagic for now until I can sort out some issues 2025-02-26 17:03:14 -07:00
subpixel_adapter.py small bug fixes 2025-03-30 20:09:40 -06:00
te_adapter.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
te_aug_adapter.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
vd_adapter.py Remove ip adapter submodule 2025-04-18 09:59:42 -06:00
zipper_resampler.py