aboutsummaryrefslogtreecommitdiffstats
path: root/scripts/loopback.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-10-12 07:35:42 +0000
committerGitHub <noreply@github.com>2022-10-12 07:35:42 +0000
commitdc1432e0dd2b826bbf5aee3e87d8270c151e4912 (patch)
tree276c27910e8ec38a1a054df02c67f73a84cb35df /scripts/loopback.py
parent1d64976dbc5a0f3124567b91fadd5014a9d93c5f (diff)
parentca5efc316b9431746ff886d259275310f63f95fb (diff)
downloadstable-diffusion-webui-gfx803-dc1432e0dd2b826bbf5aee3e87d8270c151e4912.tar.gz
stable-diffusion-webui-gfx803-dc1432e0dd2b826bbf5aee3e87d8270c151e4912.tar.bz2
stable-diffusion-webui-gfx803-dc1432e0dd2b826bbf5aee3e87d8270c151e4912.zip
Merge branch 'master' into feature/scale_to
Diffstat (limited to 'scripts/loopback.py')
-rw-r--r--scripts/loopback.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/scripts/loopback.py b/scripts/loopback.py
index e90b58d4..d8c68af8 100644
--- a/scripts/loopback.py
+++ b/scripts/loopback.py
@@ -38,6 +38,7 @@ class Script(scripts.Script):
grids = []
all_images = []
+ original_init_image = p.init_images
state.job_count = loops * batch_count
initial_color_corrections = [processing.setup_color_correction(p.init_images[0])]
@@ -45,6 +46,9 @@ class Script(scripts.Script):
for n in range(batch_count):
history = []
+ # Reset to original init image at the start of each batch
+ p.init_images = original_init_image
+
for i in range(loops):
p.n_iter = 1
p.batch_size = 1