diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-05 23:09:01 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-05 23:09:01 +0000 |
| commit | a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba (patch) | |
| tree | 8da54414ef8918317a6b7eab14f6ada3bed47d0e /scripts/prompt_matrix.py | |
| parent | b6763fb8847df5a5678f37137e7a702569e5c925 (diff) | |
| download | stable-diffusion-webui-gfx803-a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba.tar.gz stable-diffusion-webui-gfx803-a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba.tar.bz2 stable-diffusion-webui-gfx803-a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba.zip | |
added progressbar
added an option to disable progressbar
added interrupt support to DDIM/PLMS
Diffstat (limited to 'scripts/prompt_matrix.py')
| -rw-r--r-- | scripts/prompt_matrix.py | 3 |
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)
|