Jaret Burkett ff2c9f3d04 Merge branch 'main' of github.com:ostris/ai-toolkit

2023-08-07 18:05:03 -06:00

..

models

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

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

__init__.py

Initial commit

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

config_modules.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

config.py

Various windows bug fixes

2023-08-04 05:51:58 -06:00

data_loader.py

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

2023-08-07 18:04:23 -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 extensions and an example extension that merges models

2023-08-04 09:37:24 -06:00

job.py

Added extensions and an example extension that merges models

2023-08-04 09:37:24 -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

Added my good ole pattern loss. God I love that thing, conv transpose pattern instantly wiped from vae

2023-07-20 15:44:16 -06:00

lora_special.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

lora.py

Added Model rescale and prepared a release upgrade

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

losses.py

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

2023-07-21 18:26:02 -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

optimizer.py

Added Model rescale and prepared a release upgrade

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

paths.py

Added example for slider training that will run as is

2023-07-23 11:24:12 -06:00

pipelines.py

Added sd1.5 and 2.1 do the diffusers pipeline flow

2023-07-27 12:34:48 -06:00

prompt_utils.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

scheduler.py

Big refactor of SD runner and added image generator

2023-08-03 14:51:25 -06:00

stable_diffusion_model.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

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

Fix some windows dependency issues

2023-08-05 20:28:49 -06:00