diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2022-10-01 03:53:25 +0000 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2022-10-01 03:53:25 +0000 |
| commit | bdaa36c84470adbdce3e98c01a69af5e95adfb02 (patch) | |
| tree | 54999ce4bcefb0b360f3fe72969c12c248b6aa40 /modules/devices.py | |
| parent | 84e97a98c5233119d0f444e0a3a0f6391da23677 (diff) | |
| download | stable-diffusion-webui-gfx803-bdaa36c84470adbdce3e98c01a69af5e95adfb02.tar.gz stable-diffusion-webui-gfx803-bdaa36c84470adbdce3e98c01a69af5e95adfb02.tar.bz2 stable-diffusion-webui-gfx803-bdaa36c84470adbdce3e98c01a69af5e95adfb02.zip | |
When device is MPS, use CPU for GFPGAN instead
GFPGAN will not work if the device is MPS, so default to CPU instead.
Diffstat (limited to 'modules/devices.py')
| -rw-r--r-- | modules/devices.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/devices.py b/modules/devices.py index 07bb2339..08bb26d6 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -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): |