Jaret Burkett 19255cdc7c Bugfixes. Added small augmentations to dataloader. Will switch to abluminations soon though. Added ability to adjust step count on start to override what is in the file		2023-09-20 05:30:10 -06:00
..
data_transfer_object	Bugfixes. Added small augmentations to dataloader. Will switch to abluminations soon though. Added ability to adjust step count on start to override what is in the file	2023-09-20 05:30:10 -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	Improve reference slider memory and speed	2023-09-10 18:26:44 -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	Bugfixes. Added small augmentations to dataloader. Will switch to abluminations soon though. Added ability to adjust step count on start to override what is in the file	2023-09-20 05:30:10 -06:00
cuda_malloc.py	Memory optimizations. Default to using cudamalloc when torch 2.0 for mem allocation	2023-09-12 04:30:23 -06:00
data_loader.py	added flipping x and y for dataset loader	2023-09-17 08:42:54 -06:00
dataloader_mixins.py	Bugfixes. Added small augmentations to dataloader. Will switch to abluminations soon though. Added ability to adjust step count on start to override what is in the file	2023-09-20 05:30:10 -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	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -06:00
lora_special.py	implemented device placement preset system more places. Vastly improved speed on setting network multiplier and activating network. Fixed timing issues on progress bar	2023-09-14 08:31:54 -06:00
losses.py	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -06:00
lycoris_special.py	Added some further extendability for plugins	2023-09-19 05:41:44 -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	Added some further extendability for plugins	2023-09-19 05:41:44 -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	implemented device placement preset system more places. Vastly improved speed on setting network multiplier and activating network. Fixed timing issues on progress bar	2023-09-14 08:31:54 -06:00
prompt_utils.py	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -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	t2i training working from what I can tell at least	2023-09-17 15:56:43 -06:00
scheduler.py	Allow special args for schedulers	2023-09-03 20:38:44 -06:00
sd_device_states_presets.py	Added base setup for training t2i adapters. Currently untested, saw something else shiny i wanted to finish sirst. Added content_or_style to the training config. It defaults to balanced, which is standard uniform time step sampling. If style or content is passed, it will use cubic sampling for timesteps to favor timesteps that are beneficial for training them. for style, favor later timesteps. For content, favor earlier timesteps.	2023-09-16 08:30:38 -06:00
stable_diffusion_model.py	t2i training working from what I can tell at least	2023-09-17 15:56:43 -06:00
style.py	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -06:00
train_pipelines.py	Information trainer	2023-07-28 08:16:29 -06:00
train_tools.py	Fixes to esrgan trainer. Moved logic for sd prompt embeddings out of diffusers pipeline so I can manipulate it	2023-09-16 17:41:07 -06:00