From 945c57c4336c19d42a871099a62d826039c60893 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 31 Dec 2024 19:41:50 +0300 Subject: [PATCH] Fix ESRGAN with CPU --- CHANGELOG.md | 1 + modules/postprocess/esrgan_model.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 2730719a5..5fc767179 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -55,6 +55,7 @@ NYE refresh release with quite a few optimizatios and bug fixes... - hires with refiner prompt and batch processing - processing with nested calls - ui networks initial sort + - esrgan on cpu devices ## Update for 2024-12-24 diff --git a/modules/postprocess/esrgan_model.py b/modules/postprocess/esrgan_model.py index 8ff75abab..41a3585bd 100644 --- a/modules/postprocess/esrgan_model.py +++ b/modules/postprocess/esrgan_model.py @@ -142,7 +142,7 @@ def load_model(self, path: str): if self.models.get(info.local_data_path, None) is not None: shared.log.debug(f"Upscaler cached: type={self.name} model={info.local_data_path}") return self.models[info.local_data_path] - state_dict = torch.load(info.local_data_path, map_location='cpu' if devices.device.type == 'mps' else None) + state_dict = torch.load(info.local_data_path, map_location='cpu' if devices.device.type in {'mps', 'cpu'} else None) shared.log.info(f"Upscaler loaded: type={self.name} model={info.local_data_path}") if "params_ema" in state_dict: