diff options
author | brkirch <brkirch@users.noreply.github.com> | 2022-10-14 04:42:53 -0400 |
---|---|---|
committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-14 16:31:39 +0300 |
commit | fdef8253a43ca5135923092ca9b85e878d980869 (patch) | |
tree | db93cdfdbd1719de8d94bdabd6188041f1f69edc /modules/devices.py | |
parent | fdecb636855748e03efc40c846a0043800aadfcc (diff) |
Add 'interrogate' and 'all' choices to --use-cpu
* Add 'interrogate' and 'all' choices to --use-cpu
* Change type for --use-cpu argument to str.lower, so that choices are case insensitive
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 03ef58f1..eb422583 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -34,7 +34,7 @@ def enable_tf32(): errors.run(enable_tf32, "Enabling TF32") -device = device_gfpgan = device_bsrgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device() +device = device_interrogate = device_gfpgan = device_bsrgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device() dtype = torch.float16 dtype_vae = torch.float16 |