robust reload logic

AUTOMATIC1111 · May 23, 2023 · 17b75e9 · 17b75e9
1 parent 9ad38b3
commit 17b75e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/stablesr.py b/scripts/stablesr.py
@@ -206,7 +206,8 @@ def run(self, p: StableDiffusionProcessingImg2Img, model: str, scale_factor:floa
         if self.last_path != self.model_list[model]:
             # load the model
             self.stablesr_model = None
-            # get the type and the device of the unet model's first parameter
+
+        if self.stablesr_model is None:
             self.stablesr_model = StableSR(self.model_list[model], dtype=first_param.dtype, device=first_param.device)
             self.last_path = self.model_list[model]