| .. |
|
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 |