Jaret Burkett 1069dee0e4 Added ui sopport for multi control samples and datasets. Added qwen image edit 5209 to the ui		2025-09-25 11:10:02 -06:00
..
data_transfer_object	Initial support for qwen image edit plus	2025-09-24 11:39:10 -06:00
keymaps	Fixed issue with grad scaling	2024-07-20 08:21:57 -06:00
models	Initial support for qwen image edit plus	2025-09-24 11:39:10 -06:00
optimizers	Added a basic torch profiler that can be used in config during development to find some obvious issues.	2025-06-17 13:03:39 -06:00
orig_configs	Added refiner fine tuning. Works, but needs some polish.	2023-11-05 17:15:03 -07:00
samplers	WIP on mean flow loss. Still a WIP.	2025-06-12 08:00:51 -06:00
timestep_weighing	Added new timestep weighing strategy	2025-06-04 01:16:02 -06:00
util	Add stepped loss type	2025-09-22 15:50:12 -06:00
__init__.py	Initial commit	2023-07-05 16:44:58 -06:00
accelerator.py	Fallback unwrapping logic if fails	2025-05-21 03:10:33 -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 a file signature check on the dataset size caching system to invalidate cached dimensions if the file changes.	2025-04-01 07:39:36 -06:00
buckets.py	Added ability to load video datasets and train with them	2025-03-19 09:54:26 -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	Added ui sopport for multi control samples and datasets. Added qwen image edit 5209 to the ui	2025-09-25 11:10:02 -06:00
control_generator.py	Reworked control generator. It is now significantly faster. Also uses better pose model with better license.	2025-05-08 14:35:55 -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	More work on mean flow loss. Moved it to an adapter. Still not functioning properly though.	2025-06-16 07:17:35 -06:00
data_loader.py	Initial support for finetuning qwen image. Will only work with caching for now, need to add controls everywhere.	2025-08-21 16:41:17 -06:00
dataloader_mixins.py	Initial support for qwen image edit plus	2025-09-24 11:39:10 -06:00
dequantize.py	Improvements for full tuning flux. Added debugging launch config for vscode	2024-10-29 04:54:08 -06:00
ema.py	Increase ema feedback amount. Normalize the dfe 4 image embeds	2025-06-10 08:01:13 -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 support for finetuning OmniGen2.	2025-06-25 13:58:16 -06:00
guidance.py	Add model hooks to polarity loss	2025-04-17 09:00:10 -06:00
image_utils.py	Added ability to load video datasets and train with them	2025-03-19 09:54:26 -06:00
inversion_utils.py	Various bug fixes, wip stuff, and tweaks	2023-11-02 18:19:20 -06:00
ip_adapter.py	Remove ip adapter submodule	2025-04-18 09:59:42 -06:00
job.py	Moved the run job command	2023-08-13 10:25:56 -06:00
kohya_lora.py	Removed all submodules. Submodule free now, yay.	2025-04-18 10:39:15 -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
logging_aitk.py	Pixel shuffle adapter. Some bug fixes thrown in	2025-03-29 21:15:01 -06:00
lora_special.py	Add ability to train a full rank LoRA. (experimental)	2025-09-09 07:36:25 -06:00
lorm.py	Various bug fixes and optimizations for quantized training. Added untested custom adam8bit optimizer. Did some work on LoRM (dont use)	2024-11-20 09:16:55 -07:00
losses.py	Small updates and bug fixes for various things	2025-06-03 20:08:35 -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	Add ability to train a full rank LoRA. (experimental)	2025-09-09 07:36:25 -06:00
optimizer.py	Removed unnessary import	2025-03-25 11:54:42 -06:00
paths.py	Switch to unified text encoder for wan models. Pred for 2.2 14b	2025-08-14 10:07:18 -06: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	Change inpainting mask to zero out on latents instead of image for inpaint area.	2025-03-24 14:16:52 -06:00
pixel_shuffle_encoder.py	Pixel shuffle adapter. Some bug fixes thrown in	2025-03-29 21:15:01 -06:00
print.py	Fix issue where the job would hang in the ui if it failed to start	2025-03-21 09:46:41 -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	Preperation for audio	2025-09-02 07:26:50 -06:00
reference_adapter.py	Removed all submodules. Submodule free now, yay.	2025-04-18 10:39:15 -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	WIP on mean flow loss. Still a WIP.	2025-06-12 08:00:51 -06:00
saving.py	Added training for pixart-a	2024-02-13 16:00:04 -07:00
scheduler.py	Modal cloud training support, fixed typo in toolkit/scheduler.py, Schnell training support for Colab, issue #92 , issue #114 (#115)	2024-08-22 21:25:44 -06:00
sd_device_states_presets.py	Added training for an experimental decoratgor embedding. Allow for turning off guidance embedding on flux (for unreleased model). Various bug fixes and modifications	2024-12-15 08:59:27 -07:00
stable_diffusion_model.py	Initial support for qwen image edit plus	2025-09-24 11:39:10 -06:00
style.py	Bug fixes and little improvements here and there.	2024-06-08 06:24:20 -06:00
timer.py	Show the console log on the job overview in the ui.	2025-03-21 11:45:36 -06:00
train_pipelines.py	Information trainer	2023-07-28 08:16:29 -06:00
train_tools.py	Added initial support for f-lite model	2025-05-01 11:15:18 -06:00
unloader.py	Fix issue with fake text encoder unload	2025-08-14 09:33:44 -06:00