RealESRGAN use devices.device_esrgan

2026-09-18 16:54:33 +02:00 · 2024-03-18 23:02:36 +03:00
parent f3b1ba3a5d
commit 345740fec6
1 changed files with 1 additions and 1 deletions
@@ -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