Unverified Commit cd8673bd authored by DepFA's avatar DepFA Committed by GitHub

add embed embedding to ui

parent 5841990b
...@@ -1001,7 +1001,8 @@ def create_ui(wrap_gradio_gpu_call): ...@@ -1001,7 +1001,8 @@ def create_ui(wrap_gradio_gpu_call):
steps = gr.Number(label='Max steps', value=100000, precision=0) 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) 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_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.Row():
with gr.Column(scale=2): with gr.Column(scale=2):
gr.HTML(value="") gr.HTML(value="")
...@@ -1063,6 +1064,7 @@ def create_ui(wrap_gradio_gpu_call): ...@@ -1063,6 +1064,7 @@ def create_ui(wrap_gradio_gpu_call):
create_image_every, create_image_every,
save_embedding_every, save_embedding_every,
template_file, template_file,
save_image_with_stored_embedding,
], ],
outputs=[ outputs=[
ti_output, ti_output,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment