From 94b676d91382267e7429bd68362019868affd9d1 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 13 Feb 2023 17:19:18 +0100 Subject: Update --- infer.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'infer.py') 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(): -- cgit v1.2.3-54-g00ecf