split learn rates

3 years ago · 632e8d6602
parent 0087079c2d
commit 632e8d6602
1 changed files with 2 additions and 2 deletions
--- a/modules/ui.py
+++ b/modules/ui.py
@ -1342,7 +1342,7 @@ def create_ui(wrap_gradio_gpu_call):
            _js="start_training_textual_inversion",
            inputs=[
                train_embedding_name,
-                learn_rate,
+                embedding_learn_rate,
                batch_size,
                dataset_directory,
                log_directory,
@ -1367,7 +1367,7 @@ def create_ui(wrap_gradio_gpu_call):
            _js="start_training_textual_inversion",
            inputs=[
                train_hypernetwork_name,
-                learn_rate,
+                hypernetwork_learn_rate,
                batch_size,
                dataset_directory,
                log_directory,