ai-toolkit

data_transfer_object

Hude rework to move the batch to a DTO to make it far more modular to the future ui

2023-08-29 10:22:19 -06:00

keymaps

Corrected key saving and loading to better match kohya

2023-09-04 00:22:34 -06:00

models

Fixed numerous issues with traing ESRGAN

2023-08-08 20:03:19 -06:00

__init__.py

Initial commit

2023-07-05 16:44:58 -06:00

basic.py

Worked on reference slider script. It is working well currently. Still going to tune it a bit before a writeup though

2023-08-12 17:59:24 -06:00

buckets.py

Switched to new bucket system that matched sdxl trained buckets. Fixed requirements. Updated embeddings to work with sdxl. Added method to train lora with an embedding at the trigger. Still testing but works amazingly well from what I can see

2023-09-07 13:06:18 -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

config_modules.py

Switched to new bucket system that matched sdxl trained buckets. Fixed requirements. Updated embeddings to work with sdxl. Added method to train lora with an embedding at the trigger. Still testing but works amazingly well from what I can see

2023-09-07 13:06:18 -06:00

config.py

Added a way to run as a library by passing job dict

2023-08-13 09:54:39 -06:00

data_loader.py

Reworked bucket loader to scale buckets to pixels amounts not just minimum size. Makes the network more consistant

2023-08-30 14:52:12 -06:00

dataloader_mixins.py

Bug fixes and improvements to token injection

2023-09-08 06:10:59 -06:00

embedding.py

Bug fixes and improvements to token injection

2023-09-08 06:10:59 -06:00

esrgan_utils.py

Added training for a custom version of ERSGAN arcitecture. Testing training now

2023-08-07 18:04:23 -06:00

extension.py

Added inbuild plugins and made one for image referenced. WIP

2023-08-10 16:02:44 -06:00

image_utils.py

Reworked bucket loader to scale buckets to pixels amounts not just minimum size. Makes the network more consistant

2023-08-30 14:52:12 -06:00

job.py

Moved the run job command

2023-08-13 10:25:56 -06:00

kohya_model_util.py

Hotfix to handle latest transformers clip model missing key suddenly

2023-08-02 12:26:18 -06:00

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

reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl

2023-09-03 07:56:09 -06:00

lora_special.py

Added experimental concept replacer, replicate converter, bucket maker, and other goodies

2023-09-06 18:50:32 -06:00

losses.py

Base for loopback lora training setup, still working on proper sliders

2023-07-21 18:26:02 -06:00

lycoris_special.py

Fully tested and now supporting locon on sdxl. If you have the ram

2023-09-04 14:05:10 -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

Added Model rescale and prepared a release upgrade

2023-08-01 13:49:54 -06:00

network_mixins.py

Switched to new bucket system that matched sdxl trained buckets. Fixed requirements. Updated embeddings to work with sdxl. Added method to train lora with an embedding at the trigger. Still testing but works amazingly well from what I can see

2023-09-07 13:06:18 -06:00

optimizer.py

Cleanup and small bug fixes

2023-08-29 05:45:49 -06:00

paths.py

Added a converter back to ldm from diffusers for sdxl. Can finally get to training it properly

2023-08-21 16:22:01 -06:00

pipelines.py

reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl

2023-09-03 07:56:09 -06:00

prompt_utils.py

Fixed issue with loadin models after resume function added. Added additional flush if not training text encoder to clear out vram before grad accum

2023-08-28 17:56:30 -06:00

sampler.py

reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl

2023-09-03 07:56:09 -06:00

saving.py

Corrected key saving and loading to better match kohya

2023-09-04 00:22:34 -06:00

scheduler.py

Allow special args for schedulers

2023-09-03 20:38:44 -06:00

stable_diffusion_model.py

Bug fixes and improvements to token injection

2023-09-08 06:10:59 -06:00

style.py

Added Critic support to VAE training. Still tweaking and working on it. Many other fixes

2023-07-19 15:57:32 -06:00

train_pipelines.py

Information trainer

2023-07-28 08:16:29 -06:00

train_tools.py

Added multiplier jitter, min_snr, ability to choose sdxl encoders to use, shuffle generator, and other fun

2023-08-19 05:54:22 -06:00