michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Reset init img in loopback at start of each batch (#2214)

Browse source 
Before a new batch would use the last image from the previous batch. Now
each batch will use the original image for the init image at the start of the
batch.
This commit is contained in:
Rory Grieve 2022-10-11 11:23:47 +01:00 committed by GitHub
parent 255be75d30
commit 4b460fcb1a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/loopback.py
View file

@ -38,6 +38,7 @@ class Script(scripts.Script):
grids = [] grids = []
all_images = [] all_images = []
original_init_image = p.init_images
state.job_count = loops * batch_count state.job_count = loops * batch_count
initial_color_corrections = [processing.setup_color_correction(p.init_images[0])] 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): for n in range(batch_count):
history = [] history = []
# Reset to original init image at the start of each batch
p.init_images = original_init_image
for i in range(loops): for i in range(loops):
p.n_iter = 1 p.n_iter = 1
p.batch_size = 1 p.batch_size = 1