From 7d805e90ee2d6078ccf9960e6e3cd9f99c0c91b4 Mon Sep 17 00:00:00 2001
From: Alexey Shmelev <aznamir@gmail.com>
Date: Tue, 13 Aug 2024 11:09:13 -0400
Subject: [PATCH] there are no multiple models, so no need to check

---
 modules/postprocess/aurasr_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)