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/prompt_matrix.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'scripts/prompt_matrix.py')

diff --git a/scripts/prompt_matrix.py b/scripts/prompt_matrix.py
index ed31dd36..720d7583 100644
--- a/scripts/prompt_matrix.py
+++ b/scripts/prompt_matrix.py
@@ -20,6 +20,8 @@ def draw_xy_grid(xs, ys, x_label, y_label, cell):
 
     first_pocessed = None
 
+    state.job_count = len(xs) * len(ys)
+
     for iy, y in enumerate(ys):
         for ix, x in enumerate(xs):
             state.job = f"{ix + iy * len(xs) + 1} out of {len(xs) * len(ys)}"
@@ -29,6 +31,7 @@ def draw_xy_grid(xs, ys, x_label, y_label, cell):
                 first_pocessed = processed
 
             res.append(processed.images[0])
+            state.nextjob()
 
     grid = images.image_grid(res, rows=len(ys))
     grid = images.draw_grid_annotations(grid, res[0].width, res[0].height, hor_texts, ver_texts)
-- 
cgit v1.2.3