vladmandic / automatic

SD.Next: Advanced Implementation of Stable Diffusion and other Diffusion-based generative image models
https://github.com/vladmandic/automatic
GNU Affero General Public License v3.0
5.61k stars 411 forks source link

[Issue]: Aura SR 4x upscaler blows up at the end of upscaling when "Unload upscaler after processing" option is checked #3367

Closed AznamirWoW closed 2 months ago

AznamirWoW commented 2 months ago

Issue Description

unlike other upscaler scripts aurasr_model.py does not populate self.models, so it breaks on this line if shared.opts.upscaler_unload and selected_model in self.models:

should replace with

if shared.opts.upscaler_unload:

Version Platform Description

dev

Relevant log output

No response

Backend

Diffusers

UI

ModernUI

Branch

Dev

Model

StableDiffusion XL

Acknowledgements

vladmandic commented 2 months ago

since you already found the cause and solution, why not do a pr?

AznamirWoW commented 2 months ago

since you already found the cause and solution, why not do a pr?

done