VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-15 08:36:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update accel_sdxl_gen_img.py

Browse Source
This commit is contained in:
DKnight54
2025-01-28 06:08:17 +08:00
committed by GitHub
parent 009d07a988
commit 54ca0420e3
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
accel_sdxl_gen_img.py
View File

@@ -2467,8 +2467,12 @@ def main(args):
# repeat prompt
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 pi == 0: # or len(raw_prompts) > 1:
prompt_args = raw_prompt.strip().split(" --")
prompt = prompt_args[0]
if pi == 0 or len(raw_prompts) > 1:
logger.info(f"prompt {prompt_index+1}/{len(prompt_list)}: {prompt}")
if pi == 0:
# parse prompt: if prompt is not changed, skip parsing
width = args.W
height = args.H
@@ -2503,10 +2507,6 @@ def main(args):
gl_s_noise = args.gradual_latent_s_noise
gl_unsharp_params = args.gradual_latent_unsharp_params
prompt_args = raw_prompt.strip().split(" --")
prompt = prompt_args[0]
logger.info(f"prompt {prompt_index+1}/{len(prompt_list)}: {prompt}")
for parg in prompt_args[1:]:
try: