Indicate that live generaiton has stopped after an error

Acly · Oct 18, 2024 · 5be4e08 · 5be4e08
1 parent da39321
commit 5be4e08
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ai_diffusion/model.py b/ai_diffusion/model.py
@@ -437,6 +437,7 @@ def cancel(self, active=False, queued=False):
     def report_error(self, message: str):
         self.error = message
         self.live.is_active = False
+        self.custom.is_live = False
 
     def clear_error(self):
         if self.error != "":