Jaret Burkett 4ca819a05e Fixes for dataloader		2023-08-31 04:54:10 -06:00
..
data_transfer_object	Hude rework to move the batch to a DTO to make it far more modular to the future ui	2023-08-29 10:22:19 -06:00
keymaps	Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol	2023-08-29 15:20:03 -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	Worked on reference slider script. It is working well currently. Still going to tune it a bit before a writeup though	2023-08-12 17:59:24 -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.py	Added a way to run as a library by passing job dict	2023-08-13 09:54:39 -06:00
config_modules.py	Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol	2023-08-29 15:20:03 -06:00
data_loader.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
dataloader_mixins.py	Fixes for dataloader	2023-08-31 04:54:10 -06:00
embedding.py	Tied in ant tested TI script	2023-08-23 13:26:28 -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	Moved SD batch processing to a shared method and added it for use in slider training. Still testing if it affects quality over sampling	2023-08-26 08:55:00 -06:00
lora.py	Added Model rescale and prepared a release upgrade	2023-08-01 13:49:54 -06:00
lora_special.py	Cleanup and small bug fixes	2023-08-29 05:45:49 -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	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	Added sd1.5 and 2.1 do the diffusers pipeline flow	2023-07-27 12:34:48 -06:00
prompt_utils.py	Fixed issue with loadin models after resume function added. Added additional flush if not training text encoder to clear out vram before grad accum	2023-08-28 17:56:30 -06:00
saving.py	fixed issues with converting and saving models. Cleaned keys. Improved testing for cycle load saving.	2023-08-29 12:31:19 -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	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
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