diff --git a/modules/postprocess/aurasr_model.py b/modules/postprocess/aurasr_model.py index d1e69455e..ab5844d1a 100644 --- a/modules/postprocess/aurasr_model.py +++ b/modules/postprocess/aurasr_model.py @@ -30,7 +30,7 @@ class UpscalerAuraSR(Upscaler): image = self.model.upscale_4x(img) self.model.upsampler.to(devices.cpu) - if shared.opts.upscaler_unload and selected_model in self.models: + if shared.opts.upscaler_unload: self.model = None shared.log.debug(f"Upscaler unloaded: type={self.name} model={selected_model}") devices.torch_gc(force=True)