ai-toolkit/toolkit
Jaret Burkett 64a5441832 Fully tested and now supporting locon on sdxl. If you have the ram 2023-09-04 14:05: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 Corrected key saving and loading to better match kohya 2023-09-04 00:22:34 -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 Added LoCON from LyCORIS 2023-09-04 08:48:07 -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 reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl 2023-09-03 07:56:09 -06:00
lora_special.py Fully tested and now supporting locon on sdxl. If you have the ram 2023-09-04 14:05:10 -06:00
losses.py Base for loopback lora training setup, still working on proper sliders 2023-07-21 18:26:02 -06:00
lycoris_special.py Fully tested and now supporting locon on sdxl. If you have the ram 2023-09-04 14:05:10 -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
network_mixins.py Fully tested and now supporting locon on sdxl. If you have the ram 2023-09-04 14:05:10 -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 reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl 2023-09-03 07:56:09 -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
sampler.py reworked samplers. Trying to find what is wrong with diffusers sampling is sdxl 2023-09-03 07:56:09 -06:00
saving.py Corrected key saving and loading to better match kohya 2023-09-04 00:22:34 -06:00
scheduler.py Allow special args for schedulers 2023-09-03 20:38:44 -06:00
stable_diffusion_model.py Fully tested and now supporting locon on sdxl. If you have the ram 2023-09-04 14:05:10 -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