Jaret Burkett f500b9f240 Add ability to do more advanced sample prompt objects to prepart for a UI rework on control images and other things.		2025-07-17 07:13:35 -06:00
..
data_transfer_object	Fix bug that can happen with fast processing dataset	2025-06-25 14:01:08 -06:00
keymaps	…
models	Fix naming with wan i2v new keys in lora	2025-07-14 07:34:01 -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	…
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	…
__init__.py	…
accelerator.py	Fallback unwrapping logic if fails	2025-05-21 03:10:33 -06:00
assistant_lora.py	…
basic.py	…
buckets.py	…
civitai.py	…
clip_vision_adapter.py	…
config.py	…
config_modules.py	Add ability to do more advanced sample prompt objects to prepart for a UI rework on control images and other things.	2025-07-17 07:13:35 -06:00
control_generator.py	…
cuda_malloc.py	…
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	Allow finetuning tiny autoencoder in vae trainer	2025-07-16 07:13:30 -06:00
dataloader_mixins.py	Fix `NameError`.	2025-06-11 15:07:20 +02:00
dequantize.py	…
ema.py	Increase ema feedback amount. Normalize the dfe 4 image embeds	2025-06-10 08:01:13 -06:00
embedding.py	…
esrgan_utils.py	…
extension.py	Added support for finetuning OmniGen2.	2025-06-25 13:58:16 -06:00
guidance.py	…
image_utils.py	…
inversion_utils.py	…
ip_adapter.py	…
job.py	…
kohya_lora.py	…
kohya_model_util.py	…
layers.py	…
llvae.py	…
logging_aitk.py	…
lora_special.py	Small updates and bug fixes for various things	2025-06-03 20:08:35 -06:00
lorm.py	…
losses.py	Small updates and bug fixes for various things	2025-06-03 20:08:35 -06:00
lycoris_special.py	…
lycoris_utils.py	…
metadata.py	…
network_mixins.py	…
optimizer.py	…
paths.py	…
photomaker.py	…
photomaker_pipeline.py	…
pipelines.py	…
pixel_shuffle_encoder.py	…
print.py	…
progress_bar.py	…
prompt_utils.py	…
reference_adapter.py	…
resampler.py	…
sampler.py	WIP on mean flow loss. Still a WIP.	2025-06-12 08:00:51 -06:00
saving.py	…
scheduler.py	…
sd_device_states_presets.py	…
stable_diffusion_model.py	Fixed an issue training lumina 2	2025-06-24 10:29:47 -06:00
style.py	…
timer.py	…
train_pipelines.py	…
train_tools.py	…