u-haru db6752901f 画像のアルファチャンネルをlossのマスクとして使用するオプションを追加 (#1223) ...

* Add alpha_mask parameter and apply masked loss

* Fix type hint in trim_and_resize_if_required function

* Refactor code to use keyword arguments in train_util.py

* Fix alpha mask flipping logic

* Fix alpha mask initialization

* Fix alpha_mask transformation

* Cache alpha_mask

* Update alpha_masks to be on CPU

* Set flipped_alpha_masks to Null if option disabled

* Check if alpha_mask is None

* Set alpha_mask to None if option disabled

* Add description of alpha_mask option to docs

2024-05-19 19:07:25 +09:00

..

ipex

IPEX fix pin_memory

2024-03-28 21:08:16 +03:00

__init__.py

Proposed file structure rework and required file changes

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

adafactor_fused.py

Fused backward pass

2024-04-14 09:56:58 -05:00

attention_processors.py

update diffusers to 1.16 | train_network

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

config_util.py

画像のアルファチャンネルをlossのマスクとして使用するオプションを追加 (#1223)

2024-05-19 19:07:25 +09:00

custom_train_functions.py

画像のアルファチャンネルをlossのマスクとして使用するオプションを追加 (#1223)

2024-05-19 19:07:25 +09: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

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

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

2024-02-04 18:14:34 +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 readme and help message etc.

2024-05-12 17:55:08 +09:00

sdxl_original_unet.py

fix sdxl timestep embedding

2024-03-15 21:05:00 +09:00

sdxl_train_util.py

update readme and help message etc.

2024-05-12 17:55:08 +09:00

slicing_vae.py

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

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

train_util.py

画像のアルファチャンネルをlossのマスクとして使用するオプションを追加 (#1223)

2024-05-19 19:07:25 +09:00

utils.py

fix to work --console_log_xxx options

2024-02-12 14:49:29 +09:00