Jaret Burkett 836fee47a6 Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol

2023-08-29 15:20:03 -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_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

config.py

Added a way to run as a library by passing job dict

2023-08-13 09:54:39 -06:00

data_loader.py

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

dataloader_mixins.py

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

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

Added bucketting capabilities to dataloader. Finally have full planned capability. noice

2023-08-26 16:36:32 -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_special.py

Cleanup and small bug fixes

2023-08-29 05:45:49 -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

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

Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol

2023-08-29 15:20:03 -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