Skip to content

Commit c28de15

Browse files
Merge pull request #4087 from ikasumi/feat/batch-img2img-improve
make save dir if save dir is not exists
2 parents d79b930 + 525c1ed commit c28de15

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modules/img2img.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def process_batch(p, input_dir, output_dir, args):
5555
filename = f"{left}-{n}{right}"
5656

5757
if not save_normally:
58+
os.makedirs(output_dir, exist_ok=True)
5859
processed_image.save(os.path.join(output_dir, filename))
5960

6061

0 commit comments

Comments
 (0)