| .. |
|
audio
|
Add support for ACE-Step 1.5 and ACE-Step 1.5 XL. Also added dataset captioning through the UI. (#785)
|
2026-04-09 15:02:03 -06:00 |
|
data_transfer_object
|
Fix audio losses for DOP and other preservation losses
|
2026-08-04 22:08:48 -06:00 |
|
keymaps
|
…
|
|
|
memory_management
|
Fix race condition that can corrupt grads under certain conditions.
|
2026-08-07 14:52:47 -06:00 |
|
models
|
Add support for MiniMax H3 T2V and I2V training
|
2026-08-03 10:17:39 -06:00 |
|
optimizers
|
Add code for automagic experiment
|
2026-08-01 10:04:46 -06:00 |
|
orig_configs
|
…
|
|
|
samplers
|
fix custom_flowmatch_sampler (#783)
|
2026-04-13 09:42:05 -06:00 |
|
timestep_weighing
|
Added new timestep weighing strategy
|
2025-06-04 01:16:02 -06:00 |
|
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
|
Fallback unwrapping logic if fails
|
2025-05-21 03:10:33 -06:00 |
|
advanced_prompt_embeds.py
|
Add support for fine-tuning Hidream O1 (#831)
|
2026-05-12 11:15:16 -06:00 |
|
assistant_lora.py
|
…
|
|
|
basic.py
|
Add Mac OS support for Apple Silicon (#770)
|
2026-03-30 09:37:47 -06:00 |
|
buckets.py
|
Fixed issue with new bucket scaler
|
2026-05-28 11:34:16 -06:00 |
|
civitai.py
|
…
|
|
|
clip_vision_adapter.py
|
…
|
|
|
config.py
|
…
|
|
|
config_modules.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 |
|
control_generator.py
|
Add a control generation script.
|
2026-06-25 10:14:24 -06:00 |
|
cuda_malloc.py
|
…
|
|
|
custom_adapter.py
|
Prep for future breaking changes in newer versions of transformers library
|
2026-04-10 12:04:32 -06:00 |
|
data_loader.py
|
Merge branch 'dev' of github.com:ostris/ai-toolkit into dev
|
2026-07-29 17:50:50 -06:00 |
|
dataloader_mixins.py
|
Add dataset flag to cache the raw tensors
|
2026-08-04 15:32:48 -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
|
…
|
|
|
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
|
Adjust dataloader tester to handle videos to test them
|
2025-10-21 14:47:23 -06:00 |
|
inversion_utils.py
|
…
|
|
|
ip_adapter.py
|
Prep for future breaking changes in newer versions of transformers library
|
2026-04-10 12:04:32 -06:00 |
|
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
|
Remove future steps from loss log if resuming from an earlier step
|
2026-05-18 09:23:55 -06:00 |
|
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
|
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
|
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.py
|
…
|
|
|
photomaker_pipeline.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
|
Pixel shuffle adapter. Some bug fixes thrown in
|
2025-03-29 21:15:01 -06:00 |
|
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
|
…
|
|
|
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
|
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
|
Remove k-sampler requirement and remove it form the cold. not used anymore anyway
|
2026-07-27 14:58:31 -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 text encoder offload bug when caching embeddings (#868)
|
2026-06-07 14:09:32 -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 |
f4e9130547