Skip to content

S
Stable Diffusion Webui
Commit 4d9f2c3e authored by w-e-w's avatar w-e-w
Browse files
Options

update p.seed and p.subseed

parent b00b4294
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
modules/txt2img.py
View file @ 4d9f2c3e
...@@ -67,13 +67,16 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g ...@@ -67,13 +67,16 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g
geninfo = json.loads(generation_info) geninfo = json.loads(generation_info)
all_seeds = geninfo["all_seeds"] all_seeds = geninfo["all_seeds"]
all_subseeds = geninfo["all_subseeds"]
image_info = gallery[gallery_index] if 0 <= gallery_index < len(gallery) else gallery[0] image_info = gallery[gallery_index] if 0 <= gallery_index < len(gallery) else gallery[0]
p.firstpass_image = infotext_utils.image_from_url_text(image_info) p.firstpass_image = infotext_utils.image_from_url_text(image_info)
gallery_index_from_end = len(gallery) - gallery_index gallery_index_from_end = len(gallery) - gallery_index
seed = all_seeds[-gallery_index_from_end if gallery_index_from_end < len(all_seeds) + 1 else 0] seed = all_seeds[-gallery_index_from_end if gallery_index_from_end < len(all_seeds) + 1 else 0]
p.script_args = modules.scripts.scripts_txt2img.set_named_arg(p.script_args, 'ScriptSeed', 'seed', seed) subseed = all_subseeds[-gallery_index_from_end if gallery_index_from_end < len(all_seeds) + 1 else 0]
p.seed = seed
p.subseed = subseed
with closing(p): with closing(p):
processed = modules.scripts.scripts_txt2img.run(p, *p.script_args) processed = modules.scripts.scripts_txt2img.run(p, *p.script_args)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment