VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 14:45:19 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,317 Commits 25 Branches 38 Tags
2ca4d0c8313213c7fc6e9781078467bbf69d6712
Commit Graph

8 Commits

Author SHA1 Message Date
Aarni Koskela
478156b4f7 Refactor device determination to function; add MPS fallback 2024-01-23 14:29:03 +02:00
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
Kohya S
314a364f61 restore sd_model arg for backward compat 2023-04-19 21:11:12 +09:00
A2va
87163cff8b Fix missing pretrained_model_name_or_path 2023-04-17 09:16:07 +02:00
A2va
683680e5c8 Fixes 2023-04-09 21:52:02 +02:00
Robert Smieja
eb66e5ebac Extract parser setup to helper function 
- Allows users who `import` the scripts to examine the parser programmatically
 2023-03-20 00:06:47 -04:00
Kohya S
3f3502fb57 add message 2023-02-11 13:20:58 +09:00
Kohya S
9239fefa52 add lora interrogator with text encoder 2023-02-11 13:15:57 +09:00