diff --git a/scripts/xyz_grid_shared.py b/scripts/xyz_grid_shared.py index 672bd0ff7..8f3019f73 100644 --- a/scripts/xyz_grid_shared.py +++ b/scripts/xyz_grid_shared.py @@ -52,11 +52,13 @@ def apply_prompt(p, x, xs): p.prompt = p.prompt.replace(xs[0], x) if p.all_prompts is not None: for i in range(len(p.all_prompts)): - p.all_prompts[i] = p.all_prompts[i].replace(xs[0], x) + for j in range(len(xs)): + p.all_prompts[i] = p.all_prompts[i].replace(xs[j], x) p.negative_prompt = p.negative_prompt.replace(xs[0], x) if p.all_negative_prompts is not None: for i in range(len(p.all_negative_prompts)): - p.all_negative_prompts[i] = p.all_negative_prompts[i].replace(xs[0], x) + for j in range(len(xs)): + p.all_negative_prompts[i] = p.all_negative_prompts[i].replace(xs[j], x) shared.log.debug(f'XYZ grid apply prompt: "{xs[0]}"="{x}"')