Jaret Burkett c2a4b8e058 More timestep fixes		2023-11-28 10:34:08 -07:00
..
data_transfer_object	diffirential guidance is WORKING (from what I can tell)	2023-11-07 19:24:12 -07:00
keymaps	Fixed weight mapping for refiner	2023-11-06 07:37:47 -07:00
models	Fixed numerous issues with traing ESRGAN	2023-08-08 20:03:19 -06:00
orig_configs	Added refiner fine tuning. Works, but needs some polish.	2023-11-05 17:15:03 -07:00
samplers	Added timestep modifications to lcm scheduler for more evenly spaced timesteps	2023-11-17 23:26:52 -07:00
__init__.py	Initial commit	2023-07-05 16:44:58 -06:00
basic.py	Fixed cleanup of emebddings.	2023-11-16 20:26:11 -07:00
buckets.py	Fixed issues with dataloader bucketing. Allow using standard base image for t2i adapters.	2023-09-24 05:19:57 -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 dataset tagging and management tools using llava	2023-10-24 10:11:47 -06:00
config_modules.py	More timestep fixes	2023-11-28 10:34:08 -07: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	diffirential guidance is WORKING (from what I can tell)	2023-11-07 19:24:12 -07:00
dataloader_mixins.py	Numerous fixes for time sampling. Still not perfect	2023-11-28 07:34:43 -07:00
embedding.py	Varous bug fixes. Finalized targeted guidance algo	2023-11-10 12:18:08 -07: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 some features for an LCM condenser plugin	2023-11-15 08:56:45 -07:00
inversion_utils.py	Various bug fixes, wip stuff, and tweaks	2023-11-02 18:19:20 -06:00
ip_adapter.py	Added IP adapter training. Not functioning correctly yet	2023-09-24 02:39:43 -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	Bug fixes. Added some functionality to help with private extensions	2023-10-05 07:09:34 -06:00
lora_special.py	Improved lorm extraction and training	2023-10-28 08:21:59 -06:00
lorm.py	Added gradient accumulation finally	2023-10-28 13:14:29 -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	Improved lorm extraction and training	2023-10-28 08:21:59 -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	Bug fixes, negative prompting during training, hardened catching	2023-11-24 07:25:11 -07:00
network_mixins.py	Bug fixes, negative prompting during training, hardened catching	2023-11-24 07:25:11 -07:00
optimizer.py	Added some split prompting started code, adamw8bit, replacements improving, learnable snr gos. A lot of good stuff.	2023-11-01 06:52:21 -06:00
paths.py	Added refiner fine tuning. Works, but needs some polish.	2023-11-05 17:15:03 -07:00
pipelines.py	Added refiner fine tuning. Works, but needs some polish.	2023-11-05 17:15:03 -07: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	fixed issue with down block residuals when doing slider cfg on sdxl with t2i adapter assisted training	2023-10-01 07:32:48 -06:00
sampler.py	Added timestep modifications to lcm scheduler for more evenly spaced timesteps	2023-11-17 23:26:52 -07:00
saving.py	Added refiner fine tuning. Works, but needs some polish.	2023-11-05 17:15:03 -07:00
scheduler.py	Added doffusers schedulers	2023-10-29 12:39:50 -06:00
sd_device_states_presets.py	Added a way pass refiner ratio to sample config	2023-11-06 09:22:58 -07:00
stable_diffusion_model.py	More timestep fixes	2023-11-28 10:34:08 -07: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
timer.py	Massive speed increases and ram optimizations	2023-10-10 06:07:55 -06:00
train_pipelines.py	Information trainer	2023-07-28 08:16:29 -06:00
train_tools.py	Numerous fixes for time sampling. Still not perfect	2023-11-28 07:34:43 -07:00