Jaret Burkett a767b82b60 Fixed issue with new logger when ooming		2025-12-25 16:57:34 +00:00
..
data_transfer_object	Fixed issue with new logger when ooming	2025-12-25 16:57:34 +00:00
keymaps	…
memory_management	Do not copy pin memory if it fails, just move	2025-11-17 18:04:00 +00:00
models	Fixed issue with DOP when using Z-Image	2025-11-28 09:36:21 -07:00
optimizers	…
orig_configs	…
samplers	…
timestep_weighing	…
util	Fix issue where ARA was not working when using memory manager	2025-10-07 13:39:44 -06:00
__init__.py	…
accelerator.py	…
assistant_lora.py	…
basic.py	…
buckets.py	…
civitai.py	…
clip_vision_adapter.py	…
config.py	…
config_modules.py	Added initial support to initiate lora training from an existing lora	2025-12-22 12:49:15 -07:00
control_generator.py	Remove easy-dwpose from the default requierments	2025-12-20 07:16:20 -07:00
cuda_malloc.py	…
custom_adapter.py	…
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	Fix check for making sure vae is on the right device.	2025-10-21 14:49:20 -06:00
dequantize.py	…
ema.py	…
embedding.py	…
esrgan_utils.py	…
extension.py	…
guidance.py	…
image_utils.py	Adjust dataloader tester to handle videos to test them	2025-10-21 14:47:23 -06:00
inversion_utils.py	…
ip_adapter.py	…
job.py	…
kohya_lora.py	…
kohya_model_util.py	…
layers.py	…
llvae.py	…
logging_aitk.py	Add loss graph to the ui	2025-12-18 10:08:59 -07:00
lora_special.py	Fix issue where ARA was not working when using memory manager	2025-10-07 13:39:44 -06:00
lorm.py	…
losses.py	…
lycoris_special.py	…
lycoris_utils.py	…
metadata.py	…
network_mixins.py	Add support for training Z-Image Turbo with a de-distill training adapter	2025-11-28 08:08:53 -07:00
optimizer.py	Fixed adafactor eps	2025-10-26 05:47:25 -06:00
paths.py	…
photomaker.py	…
photomaker_pipeline.py	…
pipelines.py	…
pixel_shuffle_encoder.py	…
print.py	…
progress_bar.py	…
prompt_utils.py	Fixed issue with DOP when using Z-Image	2025-11-28 09:36:21 -07:00
reference_adapter.py	…
resampler.py	…
sampler.py	…
saving.py	…
scheduler.py	…
sd_device_states_presets.py	…
stable_diffusion_model.py	Added Differential Guidance training target	2025-11-10 09:38:25 -07:00
style.py	…
timer.py	…
train_pipelines.py	…
train_tools.py	…
unloader.py	Fix issue where text encoder was not fully unloaded in some instances	2025-11-19 09:01:00 -07:00