From 79728faa8da0443f35e83320aa3a5f979a75309e Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 12:46:39 -0800 Subject: [PATCH] Update devices.py --- modules/devices.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modules/devices.py b/modules/devices.py index 53c37a9b..22b5808c 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -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()