lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-27 10:13:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update devices.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 12:46:39 -08:00
parent 92682b4dbf
commit 79728faa8d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules/devices.py
View File

@@ -46,10 +46,10 @@ def enable_tf32():
cpu: torch.device = torch.device("cpu")
fp8: bool = False
device: torch.device = model_management.get_torch_device()
device_interrogate: torch.device = model_management.text_encoder_device()
device_gfpgan: torch.device = model_management.get_torch_device()
device_esrgan: torch.device = model_management.get_torch_device()
device_codeformer: torch.device = model_management.get_torch_device()
device_interrogate: torch.device = cpu # not used
device_gfpgan: torch.device = cpu
device_esrgan: torch.device = cpu # not used
device_codeformer: torch.device = cpu
dtype: torch.dtype = model_management.unet_dtype()
dtype_vae: torch.dtype = model_management.vae_dtype()
dtype_unet: torch.dtype = model_management.unet_dtype()