Isotr0py db84530074 Fix gradients synchronization for multi-GPUs training (#989) ...

* delete DDP wrapper

* fix train_db vae and train_network

* fix train_db vae and train_network unwrap

* network grad sync

---------

Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>

2023-12-07 22:01:42 +09:00

..

ipex

IPEX fix torch.UntypedStorage.is_cuda

2023-12-05 22:18:47 +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

Fix typo again

2023-10-30 20:19:51 +08:00

custom_train_functions.py

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

2023-11-07 23:02:25 +01: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

fix sampling in multi GPU training

2023-07-15 11:21:14 +09:00

model_util.py

Intel ARC support with IPEX

2023-09-19 18:05:05 +03:00

original_unet.py

Merge pull request #961 from rockerBOO/attention-processor

2023-12-03 21:24:12 +09: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

Add workaround for clip's bug for pooled output

2023-08-04 08:38:27 +09:00

sdxl_model_util.py

add temporary workaround for playground-v2

2023-12-06 23:08:02 +09:00

sdxl_original_unet.py

make separate U-Net for inference

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

sdxl_train_util.py

Fix gradients synchronization for multi-GPUs training (#989)

2023-12-07 22:01:42 +09:00

slicing_vae.py

make slicing vae compatible with latest diffusers

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

train_util.py

Fix gradients synchronization for multi-GPUs training (#989)

2023-12-07 22:01:42 +09:00

utils.py

small fix

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