..
data_transfer_object
Added caching to image sizes so we dont do it every time.
2024-07-15 19:07:41 -06:00
keymaps
Fixed issue with grad scaling
2024-07-20 08:21:57 -06:00
models
WIP - adding support for flux DoRA and ip adapter training
2024-08-22 04:36:39 -06:00
orig_configs
Added refiner fine tuning. Works, but needs some polish.
2023-11-05 17:15:03 -07:00
samplers
Fixed breaking change with diffusers. Allow flowmatch on normal stable diffusion models.
2024-08-22 14:36:22 -06:00
util
Added support for pixart sigma loras
2024-06-16 11:56:30 -06:00
__init__.py
Initial commit
2023-07-05 16:44:58 -06:00
assistant_lora.py
fixed huge flux training bug. Added ability to use an assistatn lora
2024-08-14 10:14:13 -06:00
basic.py
Added reference adapters, many bug fixes, more ip adapter work and customizability
2024-01-01 17:15:53 -07:00
buckets.py
A lot of pixart sigma training tweaks
2024-07-28 11:23:18 -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
clip_vision_adapter.py
Varous bug fixes
2024-07-10 15:20:04 -06:00
config.py
Added specialized scaler training to ip adapters
2024-04-05 08:17:09 -06:00
config_modules.py
Add `push_to_hub` to the trainer ( #109 )
2024-08-22 21:18:56 -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
custom_adapter.py
Switch back to old ilora
2024-07-29 07:22:05 -06:00
data_loader.py
Bug fix with dataloader. Added a flag to completly disable sampling
2024-08-12 09:19:40 -06:00
dataloader_mixins.py
Change flux latent spact if so it will not use old cache
2024-08-14 11:27:40 -06:00
ema.py
Adjustments to loading of flux. Added a feedback to ema
2024-08-07 13:17:26 -06:00
embedding.py
Work on ipadapters and custom adapters
2024-05-13 06:37:54 -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
guidance.py
Added support for polarity guidance for flow matching models
2024-08-15 12:22:00 -06:00
image_utils.py
Change img multiplier math
2024-07-30 11:33:41 -06:00
inversion_utils.py
Various bug fixes, wip stuff, and tweaks
2023-11-02 18:19:20 -06:00
ip_adapter.py
Fixed breaking change with diffusers. Allow flowmatch on normal stable diffusion models.
2024-08-22 14:36:22 -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
fixed huge flux training bug. Added ability to use an assistatn lora
2024-08-14 10:14:13 -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
Many bug fixes. Ip adapter bug fixes. Added noise to unconditional, it works better. added an ilora adapter for 1 shotting LoRAs
2024-01-28 08:20:03 -07:00
network_mixins.py
WIP - adding support for flux DoRA and ip adapter training
2024-08-22 04:36:39 -06:00
optimizer.py
Bug fixes and little improvements here and there.
2024-06-08 06:24:20 -06:00
paths.py
Added refiner fine tuning. Works, but needs some polish.
2023-11-05 17:15:03 -07:00
photomaker.py
Added an experimental clip fusion model that is showing promise for embedding concepts
2024-01-17 13:13:04 -07:00
photomaker_pipeline.py
Initial training script for photomaker training. Needs a little more work.
2024-01-15 18:46:26 -07:00
pipelines.py
Added flux training instructions
2024-08-10 14:10:02 -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
Added training for pixart-a
2024-02-13 16:00:04 -07:00
reference_adapter.py
Some work on sd3 training. Not working
2024-06-13 12:19:16 -06:00
resampler.py
Added a clip vision adapter trainer. Only works for sd15 for now
2023-12-24 13:26:04 -07:00
sampler.py
New image generation img2img. various tweaks and fixes
2024-07-24 04:13:41 -06:00
saving.py
Added training for pixart-a
2024-02-13 16:00:04 -07:00
scheduler.py
Added doffusers schedulers
2023-10-29 12:39:50 -06:00
sd_device_states_presets.py
More guidance work. Improved LoRA module resolver for unet. Added vega mappings and LoRA training for it. Various other bigfixes and changes
2023-12-15 06:02:10 -07:00
stable_diffusion_model.py
Fixed breaking change with diffusers. Allow flowmatch on normal stable diffusion models.
2024-08-22 14:36:22 -06:00
style.py
Bug fixes and little improvements here and there.
2024-06-08 06:24:20 -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
Added new experimental time step weighing that should solve a lot of issues with distribution. Updated example. Removed a warning
2024-08-13 12:02:11 -06:00
4d35a29c97