From f0e2fa781f3437023f9caa35322be855b49ce580 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Mon, 29 Aug 2022 10:11:11 +1200
Subject: [PATCH] Grid image not saving after recent changes has been fixed.

---
 ldm/dream/pngwriter.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ldm/dream/pngwriter.py b/ldm/dream/pngwriter.py
index dbd634e5f5..cba612528a 100644
--- a/ldm/dream/pngwriter.py
+++ b/ldm/dream/pngwriter.py
@@ -37,7 +37,7 @@ class PngWriter:
         if not upscaled:
             self.files_written.append([self.filepath, seed])
 
-    def unique_filename(self, seed, upscaled, previouspath=None):
+    def unique_filename(self, seed, upscaled=False, previouspath=None):
         revision = 1
 
         if previouspath is None:
@@ -60,7 +60,7 @@ class PngWriter:
             basename = os.path.basename(previouspath)
             x = re.match('^(\d+)\..*\.png', basename)
             if not x:
-                return self.unique_filename(seed, previouspath)
+                return self.unique_filename(seed, upscaled, previouspath)
 
             basecount = int(x.groups()[0])
             series = 0