ai-toolkit/toolkit/models
Jaret Burkett f85ad452c6 Added initial support for pixtral vision as a vision encoder. 2024-09-21 15:21:14 -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
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
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
pixtral_vision.py Added initial support for pixtral vision as a vision encoder. 2024-09-21 15:21:14 -06: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
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 Added support to disable single transformers in vision direct adapter 2024-09-11 08:54:51 -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