diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-11 21:50:30 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-11 21:50:30 +0300 |
| commit | 12f4f4761bc6114e6cce6972409488e26c4c8ab4 (patch) | |
| tree | 131ea9501cefd29b7e5cb4b3e1ea36d76c64beb6 /modules/ui.py | |
| parent | d7474a5185df2af84a93a12bc7e140d24e0fc516 (diff) | |
| parent | 419e539fe33e5bd1f7c7f55b477f955f6bdfc822 (diff) | |
Merge pull request #1795 from MarkovInequality/learnschedule
Added learning_rate scheduling for TI
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py index 14b87b92..2b688e32 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1070,7 +1070,7 @@ def create_ui(wrap_gradio_gpu_call): 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()))
train_hypernetwork_name = gr.Dropdown(label='Hypernetwork', choices=[x for x in shared.hypernetworks.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"))
|