ai-toolkit

data_transfer_object

Fixed bug when loading video datasets

2025-04-10 08:16:05 -06:00

keymaps

Fixed issue with grad scaling

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

models

Increase ema feedback amount. Normalize the dfe 4 image embeds

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

optimizers

Reworked automagic optimizer and did more testing. Starting to really like it. Working well.

2025-04-28 08:01:10 -06:00

orig_configs

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

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

samplers

Added new timestep weighing strategy

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

timestep_weighing

Added new timestep weighing strategy

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

util

Various bug fixes

2025-04-29 09:30:33 -06:00

__init__.py

…

accelerator.py

Fallback unwrapping logic if fails

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

assistant_lora.py

fixed huge flux training bug. Added ability to use an assistatn lora

2024-08-14 10:14:13 -06:00

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

Varous bug fixes

2024-07-10 15:20:04 -06:00

config_modules.py

Various code to support experiments.

2025-06-09 11:19:21 -06:00

config.py

Added specialized scaler training to ip adapters

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

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

Removed all submodules. Submodule free now, yay.

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

data_loader.py

Added methods to the dataloader to automatically generate controls for line, mask, inpainting, depth, and pose.

2025-04-09 13:35:04 -06:00

dataloader_mixins.py

Added new timestep weighing strategy

2025-06-04 01:16:02 -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

Increase ema feedback amount. Normalize the dfe 4 image embeds

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

embedding.py

Work on ipadapters and custom adapters

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

esrgan_utils.py

…

extension.py

…

guidance.py

Add model hooks to polarity loss

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

image_utils.py

Added ability to load video datasets and train with them

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

inversion_utils.py

Various bug fixes, wip stuff, and tweaks

2023-11-02 18:19:20 -06:00

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

Bug fixes. Added some functionality to help with private extensions

2023-10-05 07:09:34 -06:00

logging_aitk.py

Pixel shuffle adapter. Some bug fixes thrown in

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

lora_special.py

Small updates and bug fixes for various things

2025-06-03 20:08:35 -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

Improved lorm extraction and training

2023-10-28 08:21:59 -06:00

lycoris_utils.py

…

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

Reworked automagic optimizer and did more testing. Starting to really like it. Working well.

2025-04-28 08:01:10 -06:00

optimizer.py

Removed unnessary import

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

paths.py

Removed all submodules. Submodule free now, yay.

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

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

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

Hidream is training, but has a memory leak

2025-04-13 23:28:18 +00:00

reference_adapter.py

Removed all submodules. Submodule free now, yay.

2025-04-18 10:39:15 -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

Added initial support for training i2v adapter WIP

2025-04-09 08:06:29 -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

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

Allow fine tuning pruned versions of chroma. Allow flash attention 2 for chroma if it is installed.

2025-05-21 07:02:50 -06:00

style.py

Bug fixes and little improvements here and there.

2024-06-08 06:24:20 -06:00

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