split learn rates

632e8d66 · DepFA · GitHub · 0087079c · 632e8d66
Commit 632e8d66 authored Oct 20, 2022 by DepFA Committed by GitHub Oct 20, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

modules/ui.py modules/ui.py +2 -2

No files found.
--- 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,