aboutsummaryrefslogtreecommitdiffstats
path: root/scripts/poor_mans_outpainting.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-09-05 23:09:01 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-09-05 23:09:01 +0000
commita243bc7859b7ab92a28d28c11b0ed5525fa0d6ba (patch)
tree8da54414ef8918317a6b7eab14f6ada3bed47d0e /scripts/poor_mans_outpainting.py
parentb6763fb8847df5a5678f37137e7a702569e5c925 (diff)
downloadstable-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/poor_mans_outpainting.py')
-rw-r--r--scripts/poor_mans_outpainting.py4
1 files changed, 4 insertions, 0 deletions
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: