ai-toolkit/toolkit
Jaret Burkett d3ad195b51 Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird 2023-07-26 16:19:50 -06:00
..
__init__.py Initial commit 2023-07-05 16:44:58 -06:00
config.py Added Critic support to VAE training. Still tweaking and working on it. Many other fixes 2023-07-19 15:57:32 -06:00
config_modules.py Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird 2023-07-26 16:19:50 -06:00
data_loader.py Default to only training mse. Did a lot of cleanup with script. Added logging via tensorboard. 2023-07-18 09:40:51 -06:00
job.py Made a very basic vae trainer. 2023-07-17 19:03:50 -06:00
kohya_model_util.py Handle conversions back to ldm for saving 2023-07-18 19:34:35 -06:00
layers.py Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird 2023-07-26 16:19:50 -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.py Base for loopback lora training setup, still working on proper sliders 2023-07-21 18:26:02 -06:00
lora_special.py SDXL should be working, but I broke something where it is not converging. 2023-07-25 13:50:59 -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 Adjusted some names and allow to ignore text encoder on extract 2023-07-16 12:52:39 -06:00
metadata.py Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird 2023-07-26 16:19:50 -06:00
optimizer.py Added better optimizer chooised and param support 2023-07-24 09:21:58 -06:00
paths.py Added example for slider training that will run as is 2023-07-23 11:24:12 -06:00
stable_diffusion_model.py Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird 2023-07-26 16:19:50 -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_tools.py Added rescaling, locon, sdxl, all kinds of stuff. sdxl is still weird 2023-07-26 16:19:50 -06:00