1 files changed, 7 insertions, 7 deletions

diff --git a/infer.py b/infer.py
index 42b4e2d..aa75ee5 100644
--- a/infer.py
+++ b/infer.py
@@ -264,16 +264,16 @@ def generate(output_dir: Path, pipeline, args):
 
     if len(args.prompt) != 1:
         if len(args.project) != 0:
-            output_dir = output_dir.joinpath(f"{now}_{slugify(args.project)}")
+            output_dir = output_dir / f"{now}_{slugify(args.project)}"
         else:
-            output_dir = output_dir.joinpath(now)
+            output_dir = output_dir / now
 
         for prompt in args.prompt:
-            dir = output_dir.joinpath(slugify(prompt)[:100])
+            dir = output_dir / slugify(prompt)[:100]
             dir.mkdir(parents=True, exist_ok=True)
             image_dir.append(dir)
     else:
-        output_dir = output_dir.joinpath(f"{now}_{slugify(args.prompt[0])[:100]}")
+        output_dir = output_dir / f"{now}_{slugify(args.prompt[0])[:100]}"
         output_dir.mkdir(parents=True, exist_ok=True)
         image_dir.append(output_dir)
 
@@ -332,9 +332,9 @@ def generate(output_dir: Path, pipeline, args):
             basename = f"{seed}_{j // len(args.prompt)}"
             dir = image_dir[j % len(args.prompt)]
 
-            image.save(dir.joinpath(f"{basename}.png"))
-            image.save(dir.joinpath(f"{basename}.jpg"), quality=85)
-            with open(dir.joinpath(f"{basename}.txt"), 'w') as f:
+            image.save(dir / f"{basename}.png")
+            image.save(dir / f"{basename}.jpg", quality=85)
+            with open(dir / f"{basename}.txt", 'w') as f:
                 f.write(prompt[j % len(args.prompt)])
 
     if torch.cuda.is_available():