Kohya S. a5a27fe4c3 Merge pull request #1808 from recris/huber-loss-flux ...

Implement pseudo Huber loss for Flux and SD3

2024-12-01 13:15:33 +09:00

..

ipex

Bump crate-ci/typos from 1.19.0 to 1.21.0, fix typos, and updated _typos.toml (Close #1307)

2024-06-05 16:31:55 +09:00

__init__.py

Proposed file structure rework and required file changes

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

adafactor_fused.py

add stochastic rounding, fix single block

2024-08-21 21:04:10 +09:00

attention_processors.py

update diffusers to 1.16 | train_network

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

config_util.py

Update README and clean-up the code for SD3 timesteps

2024-11-07 21:27:12 +09:00

custom_offloading_utils.py

feat: add block swap for FLUX.1/SD3 LoRA training

2024-11-12 21:39:13 +09:00

custom_train_functions.py

Fix training for V-pred and ztSNR

2024-10-21 07:34:33 -04:00

deepspeed_utils.py

make deepspeed_utils

2024-02-27 21:30:46 +09:00

device_utils.py

revert logging #1137

2024-02-25 10:43:14 +09:00

flux_models.py

feat: add block swap for FLUX.1/SD3 LoRA training

2024-11-12 21:39:13 +09:00

flux_train_utils.py

feat: add block swap for FLUX.1/SD3 LoRA training

2024-11-12 21:39:13 +09:00

flux_utils.py

Merge branch 'sd3' into sd3_5_support

2024-10-29 22:30:01 +09:00

huggingface_util.py

Replace print with logger if they are logs (#905)

2024-02-04 18:14:34 +09:00

hypernetwork.py

update diffusers to 1.16 | train_network

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

lpw_stable_diffusion.py

Replace print with logger if they are logs (#905)

2024-02-04 18:14:34 +09:00

model_util.py

Merge branch 'dev' into dev_device_support

2024-02-12 13:01:54 +09:00

original_unet.py

Replace print with logger if they are logs (#905)

2024-02-04 18:14:34 +09:00

sai_model_spec.py

refactor SD3 CLIP to transformers etc.

2024-10-24 19:49:28 +09:00

sd3_models.py

fix: improve pos_embed handling for oversized images and update resolution_area_to_latent_size, when sample image size > train image size

2024-11-30 18:25:50 +09:00

sd3_train_utils.py

feat: add block swap for FLUX.1/SD3 LoRA training

2024-11-12 21:39:13 +09:00

sd3_utils.py

support SD3.5M

2024-10-30 12:51:49 +09:00

sdxl_lpw_stable_diffusion.py

sample generation in SDXL ControlNet training

2024-09-30 23:39:32 +09:00

sdxl_model_util.py

use original ControlNet instead of Diffusers

2024-09-29 23:07:34 +09:00

sdxl_original_control_net.py

use original ControlNet instead of Diffusers

2024-09-29 23:07:34 +09:00

sdxl_original_unet.py

use original ControlNet instead of Diffusers

2024-09-29 23:07:34 +09:00

sdxl_train_util.py

support weighted captions for SD/SDXL

2024-10-11 08:48:55 +09:00

slicing_vae.py

Replace print with logger if they are logs (#905)

2024-02-04 18:14:34 +09:00

strategy_base.py

Support SD3.5M multi resolutional training

2024-10-31 19:58:22 +09:00

strategy_flux.py

Support SD3.5M multi resolutional training

2024-10-31 19:58:22 +09:00

strategy_sd3.py

Support SD3.5M multi resolutional training

2024-10-31 19:58:22 +09:00

strategy_sd.py

Merge branch 'sd3' into fast_image_sizes

2024-10-13 17:31:11 +09:00

strategy_sdxl.py

support weighted captions for SD/SDXL

2024-10-11 08:48:55 +09:00

train_util.py

Fix issues found in review (pt 2)

2024-11-30 11:36:40 +00:00

utils.py

remove unused weight swapping functions from utils.py

2024-11-05 23:27:41 +09:00