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

Update accel_sdxl_gen_img.py

Browse Source
This commit is contained in:
DKnight54
2025-02-01 05:16:12 +08:00
committed by GitHub
parent 5308a0bfc1
commit 086e90dd4f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
accel_sdxl_gen_img.py
View File

@@ -2382,7 +2382,8 @@ def main(args):
# save image
highres_prefix = ("0" if highres_1st else "1") if highres_fix else ""
ts_str = time.strftime("%Y%m%d%H%M%S", time.localtime())
ds_str = time.strftime("%Y%m%d", time.localtime())
ts_str = time.strftime("%H%M%S", time.localtime())
for i, (image, globalcount, prompt, negative_prompts, seed, clip_prompt, raw_prompt) in enumerate(
zip(images, global_counter, prompts, negative_prompts, seeds, clip_prompts, raw_prompts)
):
@@ -2417,7 +2418,7 @@ def main(args):
elif args.sequential_file_name:
fln = f"im_{globalcount:02d}_{highres_prefix}{step_first + i + 1:06d}.png"
else:
fln = f"im_{globalcount:02d}_{ts_str}_{highres_prefix}{i:03d}_{seed}.png"
fln = f"im_{ds_str}_{globalcount:02d}_{ts_str}_{highres_prefix}{i:03d}_{seed}.png"
logger.info(f"Saving Image: {fln}:\nPrompt: {prompt}")
if negative_prompt is not None: