| .. |
|
__init__.py
|
Initial commit
|
2023-07-05 16:44:58 -06:00 |
|
config.py
|
Allow name to be passed through command line
|
2023-07-31 11:56:57 -06:00 |
|
config_modules.py
|
Set gradient checkpointing on unet enabled by default. Help out immensly with sdxl backprop spikes
|
2023-08-01 15:43:27 -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
|
Added Model rescale and prepared a release upgrade
|
2023-08-01 13:49:54 -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
|
Added Model rescale and prepared a release upgrade
|
2023-08-01 13:49:54 -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
|
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 |
|
stable_diffusion_model.py
|
WIP diffusers pipeline is weird. Starting to hate sdxl
|
2023-07-27 17:35:24 -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 Model rescale and prepared a release upgrade
|
2023-08-01 13:49:54 -06:00 |
2bf3e529ce