jpg grid, named grid

Grid file named with seed and prompt, output as jpg
3 years ago · 29d0d65504
parent 852baf422a
commit 29d0d65504
1 changed files with 2 additions and 1 deletions
--- a/webui.py
+++ b/webui.py
@ -401,7 +401,8 @@ def process_images(outpath, func_init, func_sample, prompt, seed, sampler_name,

                output_images.insert(0, grid)

-            grid.save(os.path.join(outpath, f'grid-{grid_count:04}.png'))
+            grid_file = f"grid-{grid_count:05}-{seed}_{prompts[i].replace(' ', '_').translate({ord(x): '' for x in invalid_filename_chars})[:128]}.jpg"
+            grid.save(os.path.join(outpath, grid_file), 'jpeg', quality=80, optimize=True)
            grid_count += 1

    info = f"""