Skip to content

GitLab

"tools/vscode:/vscode.git/clone" did not exist on "24757ba6bca0ec68b08321a322eefffae0c28345"
Commit 642516a3 authored by BazettFraga's avatar BazettFraga Committed by comfyanonymous
Browse files

create output dir if none is present

parent 773cdabf
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
nodes.py
View file @ 642516a3
...@@ -493,6 +493,9 @@ class SaveImage: ...@@ -493,6 +493,9 @@ class SaveImage:
counter = max(filter(lambda a: a[1][:-1] == filename_prefix and a[1][-1] == "_", map(map_filename, os.listdir(self.output_dir))))[0] + 1 counter = max(filter(lambda a: a[1][:-1] == filename_prefix and a[1][-1] == "_", map(map_filename, os.listdir(self.output_dir))))[0] + 1
except ValueError: except ValueError:
counter = 1 counter = 1
except FileNotFoundError:
os.mkdir(self.output_dir)
counter = 1
for image in images: for image in images:
i = 255. * image.cpu().numpy() i = 255. * image.cpu().numpy()
img = Image.fromarray(i.astype(np.uint8)) img = Image.fromarray(i.astype(np.uint8))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment