ai-toolkit/toolkit
Jaret Burkett 34bfeba229 Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -06:00
..
data_transfer_object Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -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 Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -06:00
buckets.py allow smaller images in buckets and bucket them 2023-09-10 03:43:02 -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 Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -06:00
data_loader.py Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -06:00
dataloader_mixins.py Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -06:00
embedding.py Bug fixes and improvements to token injection 2023-09-08 06:10:59 -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 Save embeddings as their trigger to match auto and comfy style loading. Also, FINALLY found why gradients were wonkey and fixed it. The root problem is dropping out of network state before backward pass. 2023-09-09 12:02:07 -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 Save embeddings as their trigger to match auto and comfy style loading. Also, FINALLY found why gradients were wonkey and fixed it. The root problem is dropping out of network state before backward pass. 2023-09-09 12:02:07 -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 Save embeddings as their trigger to match auto and comfy style loading. Also, FINALLY found why gradients were wonkey and fixed it. The root problem is dropping out of network state before backward pass. 2023-09-09 12:02:07 -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
progress_bar.py Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -06:00
prompt_utils.py Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -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 Massive speed increase. Added latent caching both to disk and to memory 2023-09-10 08:54:49 -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