diff options
author | AUTOMATIC <16777216c@gmail.com> | 2022-09-10 13:16:18 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-10 13:16:18 +0000 |
commit | b3311a50dc7a9447fec980fc7de19a71f27cd527 (patch) | |
tree | 794038648e4a6f4a27daca4fa9fb11c252736aa2 /modules/processing.py | |
parent | df58e0bfff2fd9e4881f5734a022a02100c035c6 (diff) | |
download | stable-diffusion-webui-gfx803-b3311a50dc7a9447fec980fc7de19a71f27cd527.tar.gz stable-diffusion-webui-gfx803-b3311a50dc7a9447fec980fc7de19a71f27cd527.tar.bz2 stable-diffusion-webui-gfx803-b3311a50dc7a9447fec980fc7de19a71f27cd527.zip |
fix repeating subseeds for every batch #221
Diffstat (limited to 'modules/processing.py')
-rw-r--r-- | modules/processing.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py index 816fdd18..df45d877 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -229,6 +229,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed: prompts = all_prompts[n * p.batch_size:(n + 1) * p.batch_size]
seeds = all_seeds[n * p.batch_size:(n + 1) * p.batch_size]
+ subseeds = all_subseeds[n * p.batch_size:(n + 1) * p.batch_size]
uc = p.sd_model.get_learned_conditioning(len(prompts) * [p.negative_prompt])
c = p.sd_model.get_learned_conditioning(prompts)
@@ -237,7 +238,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed: comments += model_hijack.comments
# we manually generate all input noises because each one should have a specific seed
- x = create_random_tensors([opt_C, p.height // opt_f, p.width // opt_f], seeds=seeds, subseeds=all_subseeds, subseed_strength=p.subseed_strength, seed_resize_from_h=p.seed_resize_from_h, seed_resize_from_w=p.seed_resize_from_w)
+ x = create_random_tensors([opt_C, p.height // opt_f, p.width // opt_f], seeds=seeds, subseeds=subseeds, subseed_strength=p.subseed_strength, seed_resize_from_h=p.seed_resize_from_h, seed_resize_from_w=p.seed_resize_from_w)
if p.n_iter > 1:
shared.state.job = f"Batch {n+1} out of {p.n_iter}"
|