ai-toolkit/toolkit
Jaret Burkett bee0b6a235 Added converters for all stable diffusion models to convert back to ldm format from diffusers. 2023-08-28 16:12:32 -06:00
..
keymaps Added converters for all stable diffusion models to convert back to ldm format from diffusers. 2023-08-28 16:12:32 -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 issue with bucket dataloader corpping in too much. Added normalization capabilities to LoRA modules. Testing effects, but should prevent them from burning and also make them more compatable with stacking many LoRAs 2023-08-27 09:40:01 -06:00
data_loader.py Fixed bug with dataloader not seperating mulitple datasets 2023-08-26 18:07:24 -06:00
dataloader_mixins.py Dataloader bug fixes 2023-08-27 14:36:38 -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.py Added Model rescale and prepared a release upgrade 2023-08-01 13:49:54 -06:00
lora_special.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -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 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 shuffeling permutations 2023-08-23 22:02:00 -06:00
saving.py Added converters for all stable diffusion models to convert back to ldm format from diffusers. 2023-08-28 16:12:32 -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 Added converters for all stable diffusion models to convert back to ldm format from diffusers. 2023-08-28 16:12:32 -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