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

2023-07-19 15:57:32 -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

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

losses.py

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

2023-07-19 15:57:32 -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 Critic support to VAE training. Still tweaking and working on it. Many other fixes

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

optimizer.py

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

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

paths.py

Made a very basic vae trainer.

2023-07-17 19:03: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 support for traditional LoRa extract using LoCon script

2023-07-12 19:51:40 -06:00