images history improvement

1 files changed, 9 insertions, 2 deletions

diff --git a/scripts/xy_grid.py b/scripts/xy_grid.py
index 771eb8e4..16918c99 100644
--- a/scripts/xy_grid.py
+++ b/scripts/xy_grid.py
@@ -10,7 +10,8 @@ import numpy as np
 import modules.scripts as scripts

 import gradio as gr

 

-from modules import images, hypernetwork

+from modules import images

+from modules.hypernetwork import hypernetwork

 from modules.processing import process_images, Processed, get_correct_sampler

 from modules.shared import opts, cmd_opts, state

 import modules.shared as shared

@@ -27,6 +28,9 @@ def apply_field(field):
 

 

 def apply_prompt(p, x, xs):

+    if xs[0] not in p.prompt and xs[0] not in p.negative_prompt:

+        raise RuntimeError(f"Prompt S/R did not find {xs[0]} in prompt or negative prompt.")

+

     p.prompt = p.prompt.replace(xs[0], x)

     p.negative_prompt = p.negative_prompt.replace(xs[0], x)

 

@@ -205,7 +209,10 @@ class Script(scripts.Script):
         if not no_fixed_seeds:

             modules.processing.fix_seed(p)

 

-        p.batch_size = 1

+        if not opts.return_grid:

+            p.batch_size = 1

+

+

         CLIP_stop_at_last_layers = opts.CLIP_stop_at_last_layers

 

         def process_axis(opt, vals):