mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 04:11:15 -08:00
Code Issues Projects Releases Packages Wiki Activity

When device is MPS, use CPU for GFPGAN instead

Browse source 
GFPGAN will not work if the device is MPS, so default to CPU instead.
This commit is contained in:
brkirch 2022-09-30 23:53:25 -04:00
parent 84e97a98c5
commit bdaa36c844
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/devices.py
View file

@ -34,7 +34,7 @@ errors.run(enable_tf32, "Enabling TF32")
device = get_optimal_device()
device_codeformer = cpu if has_mps else device
device_gfpgan = device_codeformer = cpu if device.type == 'mps' else device
def randn(seed, shape):