From 5e20657146139557d82dc4d0054ceab619caa700 Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Tue, 6 Sep 2022 11:51:34 +0300 Subject: [PATCH] fix grid overwriting bug --- .gitignore | 2 ++ modules/images.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index dc87da46c..0758b0eb2 100644 --- a/.gitignore +++ b/.gitignore @@ -6,3 +6,5 @@ __pycache__ /model.ckpt /GFPGANv1.3.pth /ui-config.json +/outputs +/config.json diff --git a/modules/images.py b/modules/images.py index 4226db00c..3399887d2 100644 --- a/modules/images.py +++ b/modules/images.py @@ -267,8 +267,8 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i filecount = len([x for x in os.listdir(path) if os.path.splitext(x)[1] == '.' + extension]) fullfn = "a.png" fullfn_without_extension = "a" - for i in range(100): - fn = f"{filecount:05}" if basename == '' else f"{basename}-{filecount:04}" + for i in range(500): + fn = f"{filecount+i:05}" if basename == '' else f"{basename}-{filecount+i:04}" fullfn = os.path.join(path, f"{fn}{file_decoration}.{extension}") fullfn_without_extension = os.path.join(path, f"{fn}{file_decoration}") if not os.path.exists(fullfn):