Fix upscale button remaining disabled after an error occured

Acly · Feb 8, 2025 · 9536fd0 · 9536fd0
1 parent 92187d8
commit 9536fd0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ai_diffusion/model.py b/ai_diffusion/model.py
@@ -296,10 +296,11 @@ def upscale_image(self):
             return
 
         self.clear_error()
+        self.upscale.set_in_progress(True)
+
         eventloop.run(_report_errors(self, self._enqueue_job(job, inputs)))
 
         self._doc.resize(job.params.bounds.extent)
-        self.upscale.set_in_progress(True)
         self.upscale.target_extent_changed.emit(self.upscale.target_extent)
 
     def estimate_cost(self, kind=JobKind.diffusion):
@@ -866,7 +867,7 @@ def _set_factor(self, value: float):
             self._update_can_generate()
 
     def _update_can_generate(self):
-        self.can_generate = not self._in_progress and (self.factor > 1.0 or self.use_diffusion)
+        self.can_generate = not self._in_progress
 
     @property
     def target_extent(self):