ai-toolkit

…

Added a flag to keep loading the image when latents are cached. Useful for DFE and other methods that target pixelspace losses.

2026-06-15 05:31:48 -06:00

keymaps

Fixed issue with grad scaling

2024-07-20 08:21:57 -06:00

memory_management

WIP working on convrot offloading

2026-07-11 15:37:28 -06:00

models

Fix issue where the log lone for samples would double up sometimes

2026-07-17 08:11:32 -06:00

optimizers

Add min and max LR or automagic to prefent runaway edge cases.

2026-07-17 14:19:53 -06:00

orig_configs

Added refiner fine tuning. Works, but needs some polish.

2023-11-05 17:15:03 -07:00

samplers

…

timestep_weighing

Added new timestep weighing strategy

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

util

Fix issue with qwen image edit models.

2026-07-17 09:14:04 -06:00

__init__.py

Force HF downloads to show progress bars when output is not a TTY (#909 )

2026-06-24 07:55:38 -06:00

accelerator.py

Fallback unwrapping logic if fails

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

advanced_prompt_embeds.py

…

assistant_lora.py

…

basic.py

…

buckets.py

Fixed issue with new bucket scaler

2026-05-28 11:34:16 -06:00

civitai.py

Added ability to use civit ai url ar model name and built a model downloader and cache manager for it

2023-08-13 13:09:51 -06:00

clip_vision_adapter.py

…

config_modules.py

feat: Add Anima support (#860 )

2026-07-15 11:59:01 -06:00

config.py

Added specialized scaler training to ip adapters

2024-04-05 08:17:09 -06:00

control_generator.py

Add a control generation script.

2026-06-25 10:14:24 -06:00

cuda_malloc.py

…

custom_adapter.py

…

data_loader.py

Add ability to recover from a truncated image file in the dataset

2026-06-22 05:07:30 -06:00

dataloader_mixins.py

Add ability to pull control images from same folder group

2026-07-08 05:21:31 -06:00

dequantize.py

Improvements for full tuning flux. Added debugging launch config for vscode

2024-10-29 04:54:08 -06:00

ema.py

…

embedding.py

Work on ipadapters and custom adapters

2024-05-13 06:37:54 -06:00

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

ideogram_caption.py

Add a fallback bbox adjustment when json parsing fails on ideogram4 captioner

2026-06-19 07:25:54 -06:00

image_utils.py

…

inversion_utils.py

…

ip_adapter.py

…

job.py

…

kohya_lora.py

Fix adapter scalar handling under torch.compile (#946 )

2026-07-15 11:07:55 -06:00

kohya_model_util.py

…

layers.py

Complete reqork of how slider training works and optimized it to hell. Can run entire algorythm in 1 batch now with less VRAM consumption than a quarter of it used to take

2023-08-05 18:46:08 -06:00

llvae.py

…

logging_aitk.py

…

lora_special.py

Fix adapter scalar handling under torch.compile (#946 )

2026-07-15 11:07:55 -06:00

lorm.py

…

losses.py

Small updates and bug fixes for various things

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

lycoris_special.py

Fix adapter scalar handling under torch.compile (#946 )

2026-07-15 11:07:55 -06:00

lycoris_utils.py

Updated module names to be compatable with ney koyha unet

2023-07-31 13:16:29 -06:00

metadata.py

Many bug fixes. Ip adapter bug fixes. Added noise to unconditional, it works better. added an ilora adapter for 1 shotting LoRAs

2024-01-28 08:20:03 -07:00

network_mixins.py

Fix adapter scalar handling under torch.compile (#946 )

2026-07-15 11:07:55 -06:00

optimizer.py

Added Automagic v3

2026-06-07 12:06:42 -06:00

paths.py

…

photomaker_pipeline.py

Initial training script for photomaker training. Needs a little more work.

2024-01-15 18:46:26 -07:00

photomaker.py

Added an experimental clip fusion model that is showing promise for embedding concepts

2024-01-17 13:13:04 -07:00

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

…

print.py

Remove print buffer. Add new line after sampling.

2026-07-16 11:37:23 -06:00

progress_bar.py

…

prompt_utils.py

feat: Add Anima support (#860 )

2026-07-15 11:59:01 -06:00

reference_adapter.py

Prep for future breaking changes in newer versions of transformers library

2026-04-10 12:04:32 -06:00

resampler.py

Added a clip vision adapter trainer. Only works for sd15 for now

2023-12-24 13:26:04 -07:00

sampler.py

WIP on mean flow loss. Still a WIP.

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

saving.py

Added training for pixart-a

2024-02-13 16:00:04 -07:00

scheduler.py

Modal cloud training support, fixed typo in toolkit/scheduler.py, Schnell training support for Colab, issue #92 , issue #114 (#115 )

2024-08-22 21:25:44 -06:00

sd_device_states_presets.py

…

stable_diffusion_model.py

Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning.

2026-06-25 13:19:38 -06:00

style.py

…

timer.py

…

train_pipelines.py

…

train_tools.py

…

unloader.py

Fix text encoder offload bug when caching embeddings (#868 )

2026-06-07 14:09:32 -06:00