Update gen_img.py

kohya-ss · Jan 29, 2025 · 8db164b · 8db164b
1 parent 1a1496c
commit 8db164b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gen_img.py b/gen_img.py
@@ -2811,7 +2811,8 @@ def scale_and_round(x):
 
                 # 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: