Merge pull request basujindal#168 from blessedcoolant/bug-fixes

Fixed grid image not saving
kylewlacy · Aug 29, 2022 · b605310 · b605310
2 parents 373a2d9 + f0e2fa7
commit b605310
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ldm/dream/pngwriter.py b/ldm/dream/pngwriter.py
@@ -37,7 +37,7 @@ def write_image(self, image, seed, upscaled=False):
         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 @@ def unique_filename(self, seed, upscaled, previouspath=None):
             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