ai-toolkit

data_transfer_object

Initial support for qwen image edit plus

2025-09-24 11:39:10 -06:00

keymaps

…

memory_management

Offload ARA with the layer if doing layer offloading. Add support to offload the LoRA. Still needs optimizer support

2025-10-21 06:03:27 -06:00

models

Added initial support for layer offloading wit Wan 2.2 14B models.

2025-10-20 14:54:30 -06:00

optimizers

Added a basic torch profiler that can be used in config during development to find some obvious issues.

2025-06-17 13:03:39 -06:00

orig_configs

…

samplers

WIP on mean flow loss. Still a WIP.

2025-06-12 08:00:51 -06:00

timestep_weighing

Added new timestep weighing strategy

2025-06-04 01:16:02 -06:00

util

Fix issue where ARA was not working when using memory manager

2025-10-07 13:39:44 -06:00

__init__.py

…

accelerator.py

Fallback unwrapping logic if fails

2025-05-21 03:10:33 -06:00

assistant_lora.py

…

basic.py

Added a file signature check on the dataset size caching system to invalidate cached dimensions if the file changes.

2025-04-01 07:39:36 -06:00

buckets.py

Added ability to load video datasets and train with them

2025-03-19 09:54:26 -06:00

civitai.py

…

clip_vision_adapter.py

…

config_modules.py

Offload ARA with the layer if doing layer offloading. Add support to offload the LoRA. Still needs optimizer support

2025-10-21 06:03:27 -06:00

config.py

…

control_generator.py

Reworked control generator. It is now significantly faster. Also uses better pose model with better license.

2025-05-08 14:35:55 -06:00

cuda_malloc.py

…

custom_adapter.py

More work on mean flow loss. Moved it to an adapter. Still not functioning properly though.

2025-06-16 07:17:35 -06:00

data_loader.py

Initial support for finetuning qwen image. Will only work with caching for now, need to add controls everywhere.

2025-08-21 16:41:17 -06:00

dataloader_mixins.py

Remove dropout from cached text embeddings even if used specifies it so blank prompts are not cached.

2025-09-26 11:50:53 -06:00

dequantize.py

…

ema.py

Increase ema feedback amount. Normalize the dfe 4 image embeds

2025-06-10 08:01:13 -06:00

embedding.py

…

esrgan_utils.py

…

extension.py

Added support for finetuning OmniGen2.

2025-06-25 13:58:16 -06:00

guidance.py

Add model hooks to polarity loss

2025-04-17 09:00:10 -06:00

image_utils.py

Adjust dataloader tester to handle videos to test them

2025-10-21 14:47:23 -06:00

inversion_utils.py

…

ip_adapter.py

Remove ip adapter submodule

2025-04-18 09:59:42 -06:00

job.py

…

kohya_lora.py

Removed all submodules. Submodule free now, yay.

2025-04-18 10:39:15 -06:00

kohya_model_util.py

…

layers.py

…

llvae.py

…

logging_aitk.py

Pixel shuffle adapter. Some bug fixes thrown in

2025-03-29 21:15:01 -06:00

lora_special.py

Fix issue where ARA was not working when using memory manager

2025-10-07 13:39:44 -06:00

lorm.py

Various bug fixes and optimizations for quantized training. Added untested custom adam8bit optimizer. Did some work on LoRM (dont use)

2024-11-20 09:16:55 -07:00

losses.py

Small updates and bug fixes for various things

2025-06-03 20:08:35 -06:00

lycoris_special.py

…

lycoris_utils.py

…

metadata.py

…

network_mixins.py

Offload ARA with the layer if doing layer offloading. Add support to offload the LoRA. Still needs optimizer support

2025-10-21 06:03:27 -06:00

optimizer.py

Removed unnessary import

2025-03-25 11:54:42 -06:00

paths.py

Switch to unified text encoder for wan models. Pred for 2.2 14b

2025-08-14 10:07:18 -06:00

photomaker_pipeline.py

…

photomaker.py

…

pipelines.py

Change inpainting mask to zero out on latents instead of image for inpaint area.

2025-03-24 14:16:52 -06:00

pixel_shuffle_encoder.py

Pixel shuffle adapter. Some bug fixes thrown in

2025-03-29 21:15:01 -06:00

print.py

Fix issue where the job would hang in the ui if it failed to start

2025-03-21 09:46:41 -06:00

progress_bar.py

…

prompt_utils.py

Preperation for audio

2025-09-02 07:26:50 -06:00

reference_adapter.py

Removed all submodules. Submodule free now, yay.

2025-04-18 10:39:15 -06:00

resampler.py

…

sampler.py

WIP on mean flow loss. Still a WIP.

2025-06-12 08:00:51 -06:00

saving.py

…

scheduler.py

…

sd_device_states_presets.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

stable_diffusion_model.py

Initial support for RamTorch. Still a WIP

2025-10-05 13:03:26 -06:00

style.py

…

timer.py

Show the console log on the job overview in the ui.

2025-03-21 11:45:36 -06:00

train_pipelines.py

…

train_tools.py

Added initial support for f-lite model

2025-05-01 11:15:18 -06:00

unloader.py

Fix issue with fake text encoder unload

2025-08-14 09:33:44 -06:00