add embed embedding to ui

This commit is contained in:
DepFA 2022-10-09 05:40:57 +01:00 committed by GitHub
parent 5841990b0d
commit cd8673bd9b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions

View File

@ -1001,6 +1001,7 @@ def create_ui(wrap_gradio_gpu_call):
steps = gr.Number(label='Max steps', value=100000, 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):
@ -1063,6 +1064,7 @@ def create_ui(wrap_gradio_gpu_call):
create_image_every,
save_embedding_every,
template_file,
save_image_with_stored_embedding,
],
outputs=[
ti_output,