aboutsummaryrefslogtreecommitdiffstats
path: root/modules/processing.py
diff options
context:
space:
mode:
authorspace-nuko <24979496+space-nuko@users.noreply.github.com>2023-02-22 09:52:53 +0000
committerspace-nuko <24979496+space-nuko@users.noreply.github.com>2023-02-22 09:52:53 +0000
commita2d635ad135241a0a40f67f7e1638c9c8a4ded04 (patch)
tree6ade834a27f4e12b9d5cd291f335620d74376683 /modules/processing.py
parent0cc0ee1bcb4c24a8c9715f66cede06601bfc00c8 (diff)
downloadstable-diffusion-webui-gfx803-a2d635ad135241a0a40f67f7e1638c9c8a4ded04.tar.gz
stable-diffusion-webui-gfx803-a2d635ad135241a0a40f67f7e1638c9c8a4ded04.tar.bz2
stable-diffusion-webui-gfx803-a2d635ad135241a0a40f67f7e1638c9c8a4ded04.zip
Add before_process_batch script callback
Diffstat (limited to 'modules/processing.py')
-rw-r--r--modules/processing.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py
index 2009d3bf..187e98fd 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -597,6 +597,9 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
seeds = p.all_seeds[n * p.batch_size:(n + 1) * p.batch_size]
subseeds = p.all_subseeds[n * p.batch_size:(n + 1) * p.batch_size]
+ if p.scripts is not None:
+ p.scripts.before_process_batch(p, batch_number=n, prompts=prompts, seeds=seeds, subseeds=subseeds)
+
if len(prompts) == 0:
break