@@ -36,6 +36,7 @@ from collections import namedtuple
fromcontextlibimportnullcontext
importsignal
importtqdm
importre
importk_diffusion.sampling
fromldm.utilimportinstantiate_from_config
...
...
@@ -187,7 +188,14 @@ class Options:
data=None
data_labels={
"outdir":OptionInfo("","Output dictectory; if empty, defaults to 'outputs/*'"),
"outdir_samples":OptionInfo("","Output dictectory for images; if empty, defaults to two directories below"),
"outdir_txt2img_samples":OptionInfo("outputs/txt2img-images",'Output dictectory for txt2img images'),
"outdir_img2img_samples":OptionInfo("outputs/img2img-images",'Output dictectory for img2img images'),
"outdir_grids":OptionInfo("","Output dictectory for grids; if empty, defaults to two directories below"),
"outdir_txt2img_grids":OptionInfo("outputs/txt2img-grids",'Output dictectory for txt2img grids'),
"outdir_img2img_grids":OptionInfo("outputs/img2img-grids",'Output dictectory for img2img grids'),
"save_to_dirs":OptionInfo(False,"When writing images/grids, create a directory with name derived from the prompt"),
"save_to_dirs_prompt_len":OptionInfo(10,"When using above, how many words from prompt to put into directory name",gr.Slider,{"minimum":1,"maximum":32,"step":1}),