Jaret Burkett f5aa4232fa Added ability to quantize with torchao

2025-03-20 16:28:54 -06:00

..

wan21

Added ability to quantize with torchao

2025-03-20 16:28:54 -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

Added ability to train control loras. Other important bug fixes thrown in

2025-03-14 18:03:00 -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

DoRA.py

WIP - adding support for flux DoRA and ip adapter training

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

flex2.py

Work on supporting flex.2 potential arch

2025-02-17 14:10:25 -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

flux.py

Experimental features and bug fixes

2025-02-04 13:36:34 -07: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

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

LoRAFormer.py

WIP to add the caption_proj weight to pixart sigma TE adapter

2024-07-06 13:00:21 -06: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

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

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