diff options
| author | yfszzx <yfszzx@gmail.com> | 2022-10-11 12:23:41 +0000 |
|---|---|---|
| committer | yfszzx <yfszzx@gmail.com> | 2022-10-11 12:23:41 +0000 |
| commit | 594ab4ba53a80a0a3606565404e6d1fbb60f6629 (patch) | |
| tree | 2821eef661e6568b65d8d29c268b657a17dcd423 /scripts/loopback.py | |
| parent | 7b1db45e1fda8603d4617affd976066be5e5b821 (diff) | |
| parent | dce7fc902ae2c6d9ffa71db67471abdcda72f82c (diff) | |
| download | stable-diffusion-webui-gfx803-594ab4ba53a80a0a3606565404e6d1fbb60f6629.tar.gz stable-diffusion-webui-gfx803-594ab4ba53a80a0a3606565404e6d1fbb60f6629.tar.bz2 stable-diffusion-webui-gfx803-594ab4ba53a80a0a3606565404e6d1fbb60f6629.zip | |
images history improvement
Diffstat (limited to 'scripts/loopback.py')
| -rw-r--r-- | scripts/loopback.py | 4 |
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
|