From 0640c495c362a46fe09e1c22366daccd1638e5dc Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Tue, 13 Jun 2023 11:50:14 -0500 Subject: [PATCH] Use sortable image filenames for SD outputs. --- apps/stable_diffusion/src/utils/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/apps/stable_diffusion/src/utils/utils.py b/apps/stable_diffusion/src/utils/utils.py index ddf444e788..4583fa00f4 100644 --- a/apps/stable_diffusion/src/utils/utils.py +++ b/apps/stable_diffusion/src/utils/utils.py @@ -724,9 +724,7 @@ def save_output_img(output_img, img_seed, extra_info={}): csv_path = Path(generated_imgs_path, "imgs_details.csv") prompt_slice = re.sub("[^a-zA-Z0-9]", "_", args.prompts[0][:15]) - out_img_name = ( - f"{prompt_slice}_{img_seed}_{dt.now().strftime('%y%m%d_%H%M%S')}" - ) + out_img_name = f"{dt.now().strftime('%H%M%S')}_{prompt_slice}_{img_seed}" img_model = args.hf_model_id if args.ckpt_loc: