From 9536fd028ed8b88525268b2c44585cbee50be3e8 Mon Sep 17 00:00:00 2001 From: Acly Date: Sat, 8 Feb 2025 10:19:27 +0100 Subject: [PATCH] Fix upscale button remaining disabled after an error occured --- ai_diffusion/model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ai_diffusion/model.py b/ai_diffusion/model.py index d8584a087..888324b31 100644 --- 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):