| .. |
|
data_transfer_object
|
added flipping x and y for dataset loader
|
2023-09-17 08:42:54 -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
|
added flipping x and y for dataset loader
|
2023-09-17 08:42:54 -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
|
added flipping x and y for dataset loader
|
2023-09-17 08:42:54 -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
|
Prevent lycoris network moduels if not training that part of network. Skew timesteps to favor later steps. It performs better
|
2023-09-14 15:13:24 -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 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 |
|
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
|
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 |
|
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
|
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 |
|
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 |
181f237a7b