diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 15:06:33 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 15:06:33 +0000 |
| commit | b6e5edd74657e3fd1fbd04f341b7a84625d4aa7a (patch) | |
| tree | bac9ddea7cafb768e064b75281edcfeac3a52ca1 /modules/devices.py | |
| parent | 46b0d230e7c13e247eabb22e1103ce512e7ed6b1 (diff) | |
| download | stable-diffusion-webui-gfx803-b6e5edd74657e3fd1fbd04f341b7a84625d4aa7a.tar.gz stable-diffusion-webui-gfx803-b6e5edd74657e3fd1fbd04f341b7a84625d4aa7a.tar.bz2 stable-diffusion-webui-gfx803-b6e5edd74657e3fd1fbd04f341b7a84625d4aa7a.zip | |
add built-in extension system
add support for adding upscalers in extensions
move LDSR, ScuNET and SwinIR to built-in extensions
Diffstat (limited to 'modules/devices.py')
| -rw-r--r-- | modules/devices.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/modules/devices.py b/modules/devices.py index d6a76844..f8cffae1 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -44,6 +44,15 @@ def get_optimal_device(): return cpu +def get_device_for(task): + from modules import shared + + if task in shared.cmd_opts.use_cpu: + return cpu + + return get_optimal_device() + + def torch_gc(): if torch.cuda.is_available(): with torch.cuda.device(get_cuda_device_string()): @@ -67,7 +76,7 @@ def enable_tf32(): errors.run(enable_tf32, "Enabling TF32") cpu = torch.device("cpu") -device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None +device = device_interrogate = device_gfpgan = device_esrgan = device_codeformer = None dtype = torch.float16 dtype_vae = torch.float16 |