VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor device determination to function; add MPS fallback

Browse Source
This commit is contained in:
Aarni Koskela
2024-01-16 15:01:59 +02:00
parent afc38707d5
commit 478156b4f7
10 changed files with 48 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gen_img_diffusers.py
View File

@@ -66,7 +66,7 @@ import diffusers
import numpy as np
import torch
from library.device_utils import clean_memory
from library.device_utils import clean_memory, get_preferred_device
from library.ipex_interop import init_ipex
init_ipex()
@@ -2324,7 +2324,7 @@ def main(args):
scheduler.config.clip_sample = True
# deviceを決定する
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # "mps"を考量してない
device = get_preferred_device()
# custom pipelineをコピったやつを生成する
if args.vae_slices: