diff --git a/optimizedSD/txt2img_gradio.py b/optimizedSD/txt2img_gradio.py index a909351f1..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,6 +223,28 @@ def generate( + "\nSeeds used = " + seeds[:-1] ) + + # write the metadata to a file + 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)) + 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("\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