From a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Tue, 6 Sep 2022 02:09:01 +0300
Subject: added progressbar added an option to disable progressbar added
 interrupt support to DDIM/PLMS

---
 scripts/poor_mans_outpainting.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'scripts/poor_mans_outpainting.py')

diff --git a/scripts/poor_mans_outpainting.py b/scripts/poor_mans_outpainting.py
index 98e1def0..149d8ac5 100644
--- a/scripts/poor_mans_outpainting.py
+++ b/scripts/poor_mans_outpainting.py
@@ -78,6 +78,8 @@ class Script(scripts.Script):
         batch_count = len(work)
         print(f"Poor man's outpainting will process a total of {len(work)} images tiled as {len(grid.tiles[0][2])}x{len(grid.tiles)}.")
 
+        state.job_count = batch_count
+
         for i in range(batch_count):
             p.init_images = [work[i]]
             p.image_mask = work_mask[i]
@@ -93,6 +95,8 @@ class Script(scripts.Script):
             p.seed = processed.seed + 1
             work_results += processed.images
 
+            state.nextjob()
+
         image_index = 0
         for y, h, row in grid.tiles:
             for tiledata in row:
-- 
cgit v1.2.1