Update gen_img.py

2026-04-15 08:36:41 +00:00 · 2025-01-30 02:58:03 +08:00
parent 1a1496ca93
commit 8db164b052
1 changed files with 2 additions and 1 deletions
--- a/gen_img.py
+++ b/gen_img.py
@@ -2811,7 +2811,8 @@ def main(args):

                # override filename to add index number if more than one image per prompt
                if filename is not None and (args.images_per_prompt >  1 or len(raw_prompts) > 1):
-                    filename = filename + "_%s" % pi
+                    fileext = os.path.splitext(filename)
+                    filename = fileext[0] + "_%s" + fileext[1] % pi
                    
                # override Deep Shrink    
                if ds_depth_1 is not None: