Kohya S e24d9606a2 add clean_memory_on_device and use it from training

2024-02-12 11:10:52 +09:00

..

ipex

Fix IPEX support and add XPU device to device_utils

2024-01-31 17:32:37 +03:00

__init__.py

Proposed file structure rework and required file changes

2022-12-18 15:24:46 -05:00

attention_processors.py

update diffusers to 1.16 | train_network

2023-06-01 20:39:26 +09:00

config_util.py

Add network_multiplier for dataset and train LoRA

2024-01-20 16:24:43 +09:00

custom_train_functions.py

Fix min-snr-gamma for v-prediction and ZSNR.

2023-11-07 23:02:25 +01:00

device_utils.py

add clean_memory_on_device and use it from training

2024-02-12 11:10:52 +09:00

huggingface_util.py

fix error in huggingface_path_in_repo=None

2023-08-06 22:08:30 +09:00

hypernetwork.py

update diffusers to 1.16 | train_network

2023-06-01 20:39:26 +09:00

lpw_stable_diffusion.py

Update dependencies ref #1024

2024-01-04 19:53:25 +09:00

model_util.py

Fix IPEX support and add XPU device to device_utils

2024-01-31 17:32:37 +03:00

original_unet.py

Fix typo --spda (it's --sdpa)

2024-01-16 14:32:48 +02:00

sai_model_spec.py

fix crashing in saving lora with clipskip

2023-08-08 21:03:16 +09:00

sdxl_lpw_stable_diffusion.py

Fix dtype issue in PyTorch 2.0 for generating samples in training sdxl network

2024-01-22 12:43:37 +09:00

sdxl_model_util.py

Update dependencies ref #1024

2024-01-04 19:53:25 +09:00

sdxl_original_unet.py

make separate U-Net for inference

2023-11-26 18:11:30 +09:00

sdxl_train_util.py

add clean_memory_on_device and use it from training

2024-02-12 11:10:52 +09:00

slicing_vae.py

make slicing vae compatible with latest diffusers

2023-11-26 18:12:04 +09:00

train_util.py

add clean_memory_on_device and use it from training

2024-02-12 11:10:52 +09:00

utils.py

small fix

2023-04-02 23:21:17 +09:00