ai-toolkit

audio

…

data_transfer_object

Add support for video references in MiniMax H3 ref2va

2026-08-15 06:18:09 -06:00

keymaps

…

memory_management

Fixed issue with offloading text encoder on ltx 2.5

2026-08-12 10:46:58 -06:00

models

Rework img/video reference in Minimax H3 to more closely match the comfy ui implementation.

2026-08-15 07:13:54 -06:00

optimizers

Add code for automagic experiment

2026-08-01 10:04:46 -06:00

orig_configs

…

samplers

…

timestep_weighing

…

util

Fix backwards compatability for older versions of triton

2026-08-04 21:07:57 -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

…

advanced_prompt_embeds.py

When dropping out, caching, and doing DOP, make sure we select a cahced trigger word when dropped out so DOP matches the drop out embeddings.

2026-08-11 06:59:20 -06:00

assistant_lora.py

…

basic.py

…

buckets.py

Fixed issue with new bucket scaler

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

civitai.py

…

clip_vision_adapter.py

…

config_modules.py

Make contrastive guidance loss do constant instead of sigma loss schedule by default

2026-08-13 09:52:02 -06:00

config.py

…

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

Generate thumbnails for dataset items so things load faster and the ui is more stable. Especially good for videos.

2026-08-12 20:33:42 -06:00

dataloader_mixins.py

Rework img/video reference in Minimax H3 to more closely match the comfy ui implementation.

2026-08-15 07:13:54 -06:00

dequantize.py

…

ema.py

…

embedding.py

…

esrgan_utils.py

…

extension.py

…

guidance.py

…

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

Restore device runtime scales for compiled adapters (#963 )

2026-07-18 08:19:57 -06:00

kohya_model_util.py

…

layers.py

…

llvae.py

…

logging_aitk.py

…

lora_special.py

Reworked merge in out of loras with convrot weights for better roundtrip accuracy.

2026-08-02 08:29:17 -06:00

lorm.py

…

losses.py

…

lycoris_special.py

Restore device runtime scales for compiled adapters (#963 )

2026-07-18 08:19:57 -06:00

lycoris_utils.py

…

metadata.py

…

network_mixins.py

Apply sigma to contrastive guidance to balance loss better. Prevent noise grads on images/non audio datasets. Prep for training adapters on MiniMax H3

2026-08-05 13:27:12 -06:00

optimizer.py

Fixed experiment opt name match

2026-08-01 17:43:36 -06:00

paths.py

Add support for MiniMax H3 T2V and I2V training

2026-08-03 10:17:39 -06:00

photomaker_pipeline.py

…

photomaker.py

…

pipelines.py

Remove k-sampler requirement and remove it form the cold. not used anymore anyway

2026-07-27 14:58:31 -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

…

resampler.py

…

sampler.py

Remove k-sampler requirement and remove it form the cold. not used anymore anyway

2026-07-27 14:58:31 -06:00

saving.py

…

scheduler.py

…

sd_device_states_presets.py

…

stable_diffusion_model.py

Rework img/video reference in Minimax H3 to more closely match the comfy ui implementation.

2026-08-15 07:13:54 -06:00

style.py

…

timer.py

…

train_pipelines.py

…

train_tools.py

…

unloader.py

Reworked freeing memory manager for removing text encoder completly when not needed.

2026-08-09 07:08:24 -06:00

win_console.py

Rework windows start and stop methods so that command windows dont appear. Stop with signal since we cannot signint.

2026-07-27 17:43:01 -06:00