Jaret Burkett 58537fc92b Added initial direct vision pixtral support

2024-09-28 10:47:51 -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

DoRA.py

WIP - adding support for flux DoRA and ip adapter training

2024-08-22 04:36:39 -06:00

ilora2.py

Add a mergable linear to the mid of ilora

2024-07-20 21:17:53 -06:00

ilora.py

Fixes for training ilora on flux

2024-08-31 16:55:26 -06:00

LoRAFormer.py

WIP to add the caption_proj weight to pixart sigma TE adapter

2024-07-06 13:00:21 -06:00

pixtral_vision.py

Added initial direct vision pixtral support

2024-09-28 10:47:51 -06:00

RRDB.py

Fixed numerous issues with traing ESRGAN

2023-08-08 20:03:19 -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 initial direct vision pixtral support

2024-09-28 10:47: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