hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add --use-cpu command line option

Browse Source 
Remove MPS detection to use CPU for GFPGAN / CodeFormer and add a --use-cpu command line option.
This commit is contained in:
brkirch
2022-10-04 04:24:35 -04:00
parent b88e4ea7d6
commit eeab7aedf5
4 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/devices.py
View File

@@ -1,8 +1,8 @@
import torch
# has_mps is only available in nightly pytorch (for now), `getattr` for compatibility
from modules import errors
# has_mps is only available in nightly pytorch (for now), `getattr` for compatibility
has_mps = getattr(torch, 'has_mps', False)
cpu = torch.device("cpu")
@@ -32,8 +32,7 @@ def enable_tf32():
errors.run(enable_tf32, "Enabling TF32")
device = get_optimal_device()
device_gfpgan = device_codeformer = cpu if device.type == 'mps' else device
device = device_gfpgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device()
dtype = torch.float16
def randn(seed, shape):