Skip to content

Commit c83d01b

Browse files
committed
fix hang during GFPGAN processing due to bug introduced by recent removal of batch_size arg from pngwriter
1 parent 3be1cee commit c83d01b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ldm/dream/pngwriter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,8 @@ def unique_filename(self, seed, upscaled=False, previouspath=None):
6565
series += 1
6666
filename = f'{basecount:06}.{seed}.png'
6767
path = os.path.join(self.outdir, filename)
68+
if os.path.exists(path) and upscaled:
69+
break
6870
finished = not os.path.exists(path)
6971
return os.path.join(self.outdir, filename)
7072

0 commit comments

Comments
 (0)