add skip button (#616)

2026-08-16 13:13:16 +02:00 · 2023-10-09 20:02:03 -07:00
parent 0723c2fa52
commit c275434c59
5 changed files with 26 additions and 8 deletions
@@ -542,8 +542,12 @@ def worker():

                results += imgs
            except comfy.model_management.InterruptProcessingException as e:
-                print('User stopped')
-                break
+                if shared.last_stop == 'skip':
+                    print('User skipped')
+                    continue
+                else:
+                    print('User stopped')
+                    break

            execution_time = time.perf_counter() - execution_start_time
            print(f'Generating and saving time: {execution_time:.2f} seconds')