diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 19:41:21 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 19:41:21 +0000 |
| commit | c7543d4940da672d970124ae8f2fec9de7bdc1da (patch) | |
| tree | 6fd8f544b7ff201b959e563c3b0832eb50e130a3 /modules/ui.py | |
| parent | c4445225f79f1c57afe52358ff4b205864eb7aac (diff) | |
| download | stable-diffusion-webui-gfx803-c7543d4940da672d970124ae8f2fec9de7bdc1da.tar.gz stable-diffusion-webui-gfx803-c7543d4940da672d970124ae8f2fec9de7bdc1da.tar.bz2 stable-diffusion-webui-gfx803-c7543d4940da672d970124ae8f2fec9de7bdc1da.zip | |
preprocessing for textual inversion added
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/modules/ui.py b/modules/ui.py index 8912deff..e7bde53b 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -961,6 +961,8 @@ def create_ui(wrap_gradio_gpu_call): with gr.Row().style(equal_height=False):
with gr.Column():
with gr.Group():
+ gr.HTML(value="<p style='margin-bottom: 0.7em'>See <b><a href=\"https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Textual-Inversion\">wiki</a></b> for detailed explanation.</p>")
+
gr.HTML(value="<p style='margin-bottom: 0.7em'>Create a new embedding</p>")
new_embedding_name = gr.Textbox(label="Name")
@@ -975,6 +977,24 @@ def create_ui(wrap_gradio_gpu_call): create_embedding = gr.Button(value="Create", variant='primary')
with gr.Group():
+ gr.HTML(value="<p style='margin-bottom: 0.7em'>Preprocess images</p>")
+
+ process_src = gr.Textbox(label='Source directory')
+ process_dst = gr.Textbox(label='Destination directory')
+
+ with gr.Row():
+ process_flip = gr.Checkbox(label='Flip')
+ process_split = gr.Checkbox(label='Split into two')
+ process_caption = gr.Checkbox(label='Add caption')
+
+ with gr.Row():
+ with gr.Column(scale=3):
+ gr.HTML(value="")
+
+ with gr.Column():
+ run_preprocess = gr.Button(value="Preprocess", variant='primary')
+
+ with gr.Group():
gr.HTML(value="<p style='margin-bottom: 0.7em'>Train an embedding; must specify a directory with a set of 512x512 images</p>")
train_embedding_name = gr.Dropdown(label='Embedding', choices=sorted(sd_hijack.model_hijack.embedding_db.word_embeddings.keys()))
learn_rate = gr.Number(label='Learning rate', value=5.0e-03)
@@ -1018,6 +1038,22 @@ def create_ui(wrap_gradio_gpu_call): ]
)
+ run_preprocess.click(
+ fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.preprocess, extra_outputs=[gr.update()]),
+ _js="start_training_textual_inversion",
+ inputs=[
+ process_src,
+ process_dst,
+ process_flip,
+ process_split,
+ process_caption,
+ ],
+ outputs=[
+ ti_output,
+ ti_outcome,
+ ],
+ )
+
train_embedding.click(
fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.train_embedding, extra_outputs=[gr.update()]),
_js="start_training_textual_inversion",
|