From a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 4 Jan 2023 22:06:05 +0100 Subject: Update --- infer.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'infer.py') diff --git a/infer.py b/infer.py index c4d1e0d..b29b136 100644 --- a/infer.py +++ b/infer.py @@ -295,11 +295,10 @@ def generate(output_dir, pipeline, args): for j, image in enumerate(images): image_dir = output_dir if use_subdirs: - idx = j % len(args.prompt) - image_dir = image_dir.joinpath(slugify(args.prompt[idx])[:100]) + image_dir = image_dir.joinpath(slugify(args.prompt[j % len(args.prompt)])[:100]) image_dir.mkdir(parents=True, exist_ok=True) - image.save(image_dir.joinpath(f"{seed}_{j}.png")) - image.save(image_dir.joinpath(f"{seed}_{j}.jpg"), quality=85) + image.save(image_dir.joinpath(f"{seed}_{j // len(args.prompt)}.png")) + image.save(image_dir.joinpath(f"{seed}_{j // len(args.prompt)}.jpg"), quality=85) if torch.cuda.is_available(): torch.cuda.empty_cache() -- cgit v1.2.3-70-g09d2