ai-toolkit

data_transfer_object

Memory optimizations. Default to using cudamalloc when torch 2.0 for mem allocation

2023-09-12 04:30:23 -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

Massive speed increase. Added latent caching both to disk and to memory

2023-09-10 08:54:49 -06:00

buckets.py

Improve reference slider memory and speed

2023-09-10 18:26:44 -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

Upgraded to dev for t2i on diffusers. Minor migrations to make it work.

2023-09-11 14:46:06 -06:00

config.py

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

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

cuda_malloc.py

Memory optimizations. Default to using cudamalloc when torch 2.0 for mem allocation

2023-09-12 04:30:23 -06:00

data_loader.py

big fixes

2023-09-12 18:48:39 -06:00

dataloader_mixins.py

Dont load image if we are cachine latents

2023-09-12 18:39:41 -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

Save embeddings as their trigger to match auto and comfy style loading. Also, FINALLY found why gradients were wonkey and fixed it. The root problem is dropping out of network state before backward pass.

2023-09-09 12:02:07 -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

Save embeddings as their trigger to match auto and comfy style loading. Also, FINALLY found why gradients were wonkey and fixed it. The root problem is dropping out of network state before backward pass.

2023-09-09 12:02:07 -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

Upgraded to dev for t2i on diffusers. Minor migrations to make it work.

2023-09-11 14:46:06 -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

progress_bar.py

Massive speed increase. Added latent caching both to disk and to memory

2023-09-10 08:54:49 -06:00

prompt_utils.py

Massive speed increase. Added latent caching both to disk and to memory

2023-09-10 08:54:49 -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

Massive speed increase. Added latent caching both to disk and to memory

2023-09-10 08:54:49 -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