aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ui.py
diff options
context:
space:
mode:
authorFampai <>2022-10-10 21:10:29 +0000
committerFampai <>2022-10-10 21:10:29 +0000
commit2536ecbb1790da2af0d61b6a26f38732cba665cd (patch)
tree98952174ed80c0aa376c433d5cd6f4a500b4b18f /modules/ui.py
parentce37fdd30e9fc0fe0bc5805a068ce8b11b42b5a3 (diff)
downloadstable-diffusion-webui-gfx803-2536ecbb1790da2af0d61b6a26f38732cba665cd.tar.gz
stable-diffusion-webui-gfx803-2536ecbb1790da2af0d61b6a26f38732cba665cd.tar.bz2
stable-diffusion-webui-gfx803-2536ecbb1790da2af0d61b6a26f38732cba665cd.zip
Refactored learning rate code
Diffstat (limited to 'modules/ui.py')
-rw-r--r--modules/ui.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py
index 8c06ad7c..c9e8355b 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1047,7 +1047,7 @@ def create_ui(wrap_gradio_gpu_call):
with gr.Group():
gr.HTML(value="<p style='margin-bottom: 0.7em'>Train an embedding; must specify a directory with a set of 1:1 ratio 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)
+ learn_rate = gr.Textbox(label='Learning rate', placeholder="Learning rate", value = "5.0e-03")
dataset_directory = gr.Textbox(label='Dataset directory', placeholder="Path to directory with input images")
log_directory = gr.Textbox(label='Log directory', placeholder="Path to directory where to write outputs", value="textual_inversion")
template_file = gr.Textbox(label='Prompt template file', value=os.path.join(script_path, "textual_inversion_templates", "style_filewords.txt"))