ai-toolkit/toolkit/models
Jaret Burkett 45be82d5d6 Handle inpainting training for control_lora adapter 2025-03-24 13:17:47 -06:00
..
wan21 Added ability to quantize with torchao 2025-03-20 16:28:54 -06:00
DoRA.py WIP - adding support for flux DoRA and ip adapter training 2024-08-22 04:36:39 -06:00
LoRAFormer.py WIP to add the caption_proj weight to pixart sigma TE adapter 2024-07-06 13:00:21 -06:00
RRDB.py Fixed numerous issues with traing ESRGAN 2023-08-08 20:03:19 -06:00
auraflow.py Partial implementation for training auraflow. 2024-07-12 12:11:38 -06:00
base_model.py Added ability to quantize with torchao 2025-03-20 16:28:54 -06:00
block.py Added training for a custom version of ERSGAN arcitecture. Testing training now 2023-08-07 18:04:23 -06:00
clip_fusion.py Work on additional image embedding methods. Finalized zipper resampler. It works amazing 2024-02-10 09:00:05 -07:00
clip_pre_processor.py Adjustments to the clip preprocessor. Allow merging in new weights for ip adapters so you can change the arcitecture while maintaining as much data as possible 2024-01-06 11:56:53 -07:00
cogview4.py Added ability to quantize with torchao 2025-03-20 16:28:54 -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 DFE tweaks. Adding support for more llms as text encoders 2025-02-13 04:31:49 -07:00
flex2.py Work on supporting flex.2 potential arch 2025-02-17 14:10:25 -07:00
flux.py Experimental features and bug fixes 2025-02-04 13:36:34 -07: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
ilora.py Fixes for training ilora on flux 2024-08-31 16:55:26 -06:00
ilora2.py Add a mergable linear to the mid of ilora 2024-07-20 21:17:53 -06:00
llm_adapter.py WIP Flex 2 pipeline 2025-02-16 14:54:29 -07: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 Added single value adapter training 2024-04-28 06:04:47 -06:00
size_agnostic_feature_encoder.py WIP on SAFE encoder. Work on fp16 training improvements. Various other tweaks and improvements 2024-05-27 10:50:24 -06:00
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
te_adapter.py Amall bug fixes 2024-07-18 10:39:55 -06:00
te_aug_adapter.py Added te aug adapter 2024-02-21 21:30:26 -07:00
vd_adapter.py Fixed issue with adapters not providing gradients with new grad activator 2024-10-29 14:22:10 -06:00
zipper_resampler.py Work on additional image embedding methods. Finalized zipper resampler. It works amazing 2024-02-10 09:00:05 -07:00