Isotr0py e1143caf38 Fix DDP issues and Support DDP for all training scripts (#448) ...

* Fix DDP bugs

* Fix DDP bugs for finetune and db

* refactor model loader

* fix DDP network

* try to fix DDP network in train unet only

* remove unuse DDP import

* refactor DDP transform

* refactor DDP transform

* fix sample images bugs

* change DDP tranform location

* add autocast to train_db

* support DDP in XTI

* Clear DDP import

2023-05-03 10:37:47 +09:00

..

check_lora_weights.py

Extract parser setup to helper function

2023-03-20 00:06:47 -04:00

dylora.py

fix load_state_dict failed in dylora

2023-04-14 22:13:26 +09:00

extract_lora_from_dylora.py

fix typos

2023-04-13 21:18:18 +09:00

extract_lora_from_models.py

Add block dim(rank) feature

2023-04-03 21:19:49 +09:00

lora_interrogator.py

Fix DDP issues and Support DDP for all training scripts (#448)

2023-05-03 10:37:47 +09:00

lora.py

Add attension couple+reginal LoRA

2023-04-06 08:11:54 +09:00

merge_lora_old.py

Extract parser setup to helper function

2023-03-20 00:06:47 -04:00

merge_lora.py

fix conv2d3x3 is not merged

2023-04-01 09:17:37 +09:00

resize_lora.py

update resize_lora.py (fix out of bounds and index)

2023-03-25 19:56:14 -05:00

svd_merge_lora.py

Extract parser setup to helper function

2023-03-20 00:06:47 -04:00