there are no multiple models, so no need to check

This commit is contained in:
Alexey Shmelev
2024-08-13 11:09:13 -04:00
parent 94fce42ae1
commit 7d805e90ee
+1 -1
View File
@@ -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)