aboutsummaryrefslogtreecommitdiff
path: root/modules/ui.py
diff options
context:
space:
mode:
Diffstat (limited to 'modules/ui.py')
-rw-r--r--modules/ui.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py
index 8c06ad7c..0f6427a6 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1057,7 +1057,8 @@ def create_ui(wrap_gradio_gpu_call):
num_repeats = gr.Number(label='Number of repeats for a single input image per epoch', value=100, precision=0)
create_image_every = gr.Number(label='Save an image to log directory every N steps, 0 to disable', value=500, precision=0)
save_embedding_every = gr.Number(label='Save a copy of embedding to log directory every N steps, 0 to disable', value=500, precision=0)
-
+ save_image_with_stored_embedding = gr.Checkbox(label='Save images with embedding in PNG chunks', value=True)
+
with gr.Row():
with gr.Column(scale=2):
gr.HTML(value="")
@@ -1124,6 +1125,7 @@ def create_ui(wrap_gradio_gpu_call):
create_image_every,
save_embedding_every,
template_file,
+ save_image_with_stored_embedding,
],
outputs=[
ti_output,