From 8db164b052787f3cb84e1581316381ba40744dee Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Thu, 30 Jan 2025 02:58:03 +0800
Subject: [PATCH] Update gen_img.py

---
 gen_img.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/gen_img.py b/gen_img.py
index 60d9f862..a6c616e9 100644
--- 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: