From 362eebfabcd017e10ac1955df7483c42a809c16d Mon Sep 17 00:00:00 2001 From: Vladimir Atanasov Date: Tue, 4 Oct 2022 23:12:38 +0100 Subject: [PATCH 1/2] write metadata to a txt file --- optimizedSD/txt2img_gradio.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/optimizedSD/txt2img_gradio.py b/optimizedSD/txt2img_gradio.py index a909351f1..ab103888e 100644 --- a/optimizedSD/txt2img_gradio.py +++ b/optimizedSD/txt2img_gradio.py @@ -222,6 +222,27 @@ def generate( + "\nSeeds used = " + seeds[:-1] ) + + # write the metadata to a file + with open(os.path.join(sample_path, "description" + "_" + f"{base_count:05}.txt"), 'w') as f: + f.write(prompt) + f.write("\n--------------") + f.write("\nddim_steps: "+str(ddim_steps)) + f.write("\nn_iter: "+str(n_iter)) + f.write("\nbatch_size: "+str(batch_size)) + f.write("\nHeight: "+str(Height)) + f.write("\nWidth: "+str(Width)) + f.write("\nscale: "+str(scale)) + f.write("\nddim_eta: "+str(ddim_eta)) + f.write("\nunet_bs: "+str(unet_bs)) + f.write("\ndevice: "+str(device)) + f.write("\nseed: "+str(seed)) + f.write("\nturbo: "+str(turbo)) + f.write("\nfull_precision: "+str(full_precision)) + f.write("\nsampler: "+str(sampler)) + f.write("\n--------------") + f.write("\nOutput: \n"+txt) + return Image.fromarray(grid.astype(np.uint8)), txt From 0220e79cd0360b590b59183af9d6cc4b344c5686 Mon Sep 17 00:00:00 2001 From: Vladimir Atanasov Date: Tue, 4 Oct 2022 23:41:54 +0100 Subject: [PATCH 2/2] fixed description filename count --- optimizedSD/txt2img_gradio.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/optimizedSD/txt2img_gradio.py b/optimizedSD/txt2img_gradio.py index ab103888e..eae43a865 100644 --- a/optimizedSD/txt2img_gradio.py +++ b/optimizedSD/txt2img_gradio.py @@ -24,6 +24,7 @@ from transformers import logging logging.set_verbosity_error() import mimetypes +import glob mimetypes.init() mimetypes.add_type("application/javascript", ".js") @@ -222,9 +223,10 @@ def generate( + "\nSeeds used = " + seeds[:-1] ) - + # write the metadata to a file - with open(os.path.join(sample_path, "description" + "_" + f"{base_count:05}.txt"), 'w') as f: + txt_count = len(glob.glob(sample_path+"/*.txt")) + 1 + with open(os.path.join(sample_path, "description" + "_" + f"{txt_count:05}.txt"), 'w') as f: f.write(prompt) f.write("\n--------------") f.write("\nddim_steps: "+str(ddim_steps)) @@ -236,13 +238,13 @@ def generate( f.write("\nddim_eta: "+str(ddim_eta)) f.write("\nunet_bs: "+str(unet_bs)) f.write("\ndevice: "+str(device)) - f.write("\nseed: "+str(seed)) + f.write("\nseeds: "+seeds[:-1]) f.write("\nturbo: "+str(turbo)) f.write("\nfull_precision: "+str(full_precision)) f.write("\nsampler: "+str(sampler)) f.write("\n--------------") f.write("\nOutput: \n"+txt) - + return Image.fromarray(grid.astype(np.uint8)), txt