diff options
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 ff82f2f6..5d9c7a07 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -33,7 +33,7 @@ def enable_tf32(): 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 dtype = torch.float16 def randn(seed, shape): |