From 624fbadea2b742f2bf32d82efeb332f24695881c Mon Sep 17 00:00:00 2001 From: Kohya S Date: Thu, 15 Jun 2023 19:19:16 +0900 Subject: [PATCH] fix dynamic prompt with from_file --- gen_img_diffusers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gen_img_diffusers.py b/gen_img_diffusers.py index 01001646..7b5cee1f 100644 --- a/gen_img_diffusers.py +++ b/gen_img_diffusers.py @@ -3170,10 +3170,10 @@ def main(args): raw_prompts = handle_dynamic_prompt_variants(raw_prompt, args.images_per_prompt) # repeat prompt - for prompt_index in range(args.images_per_prompt if len(raw_prompts) == 1 else len(raw_prompts)): - raw_prompt = raw_prompts[prompt_index] if len(raw_prompts) > 1 else raw_prompts[0] + for pi in range(args.images_per_prompt if len(raw_prompts) == 1 else len(raw_prompts)): + raw_prompt = raw_prompts[pi] if len(raw_prompts) > 1 else raw_prompts[0] - if prompt_index == 0 or len(raw_prompts) > 1: + if pi == 0 or len(raw_prompts) > 1: # parse prompt: if prompt is not changed, skip parsing width = args.W height = args.H