fix grid overwriting bug

This commit is contained in:
AUTOMATIC 2022-09-06 11:51:34 +03:00
parent 3f55d9a932
commit 5e20657146
2 changed files with 4 additions and 2 deletions

2
.gitignore vendored
View file

@ -6,3 +6,5 @@ __pycache__
/model.ckpt
/GFPGANv1.3.pth
/ui-config.json
/outputs
/config.json

View file

@ -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):