From 345740fec6e8a2f2a2e2451234745d0016890463 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 18 Mar 2024 23:02:36 +0300 Subject: [PATCH] RealESRGAN use devices.device_esrgan --- modules/postprocess/realesrgan_model_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/postprocess/realesrgan_model_arch.py b/modules/postprocess/realesrgan_model_arch.py index 30b8e65ac..a4f4bd682 100644 --- a/modules/postprocess/realesrgan_model_arch.py +++ b/modules/postprocess/realesrgan_model_arch.py @@ -55,7 +55,7 @@ class RealESRGANer(): self.device = torch.device( f'cuda:{gpu_id}' if torch.cuda.is_available() else 'cpu') if device is None else device else: - self.device = devices.device if device is None else device + self.device = devices.device_esrgan if device is None else device if isinstance(model_path, list): # dni