added progressbar

added an option to disable progressbar
added interrupt support to DDIM/PLMS

1 files changed, 3 insertions, 0 deletions

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)