From 5316db3fa75aa32c67718c7bf9cc0b1068c987c8 Mon Sep 17 00:00:00 2001
From: bmaltais <bernard@ducourier.com>
Date: Sun, 14 Apr 2024 22:03:36 -0400
Subject: [PATCH] Add huggingface parms to config file

---
 config example.toml               | 10 ++++++++++
 kohya_gui.py                      |  2 +-
 kohya_gui/class_basic_training.py |  2 +-
 kohya_gui/class_huggingface.py    |  2 +-
 4 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/config example.toml b/config example.toml
index 0141d5a..4565228 100644
--- a/config example.toml	
+++ b/config example.toml	
@@ -126,6 +126,16 @@ reg_images_repeat = 1                                   # Regularisation images
 util_regularization_images_repeat_input = 1             # Regularisation images repeat input
 util_training_images_repeat_input = 40                  # Training images repeat input
 
+[huggingface]
+async_upload = false              # Async upload
+huggingface_path_in_repo = ""     # Huggingface path in repo
+huggingface_repo_id = ""          # Huggingface repo id
+huggingface_repo_type = ""        # Huggingface repo type
+huggingface_repo_visibility = ""  # Huggingface repo visibility
+huggingface_token = ""            # Huggingface token
+resume_from_huggingface = ""      # Resume from huggingface (ex: {repo_id}/{path_in_repo}:{revision}:{repo_type})
+save_state_to_huggingface = false # Save state to huggingface
+
 [samples]
 sample_every_n_steps = 0   # Sample every n steps
 sample_every_n_epochs = 0  # Sample every n epochs
diff --git a/kohya_gui.py b/kohya_gui.py
index ed9d195..21c4f3e 100644
--- a/kohya_gui.py
+++ b/kohya_gui.py
@@ -40,7 +40,7 @@ def UI(**kwargs):
         css=css, title=f"Kohya_ss GUI {release}", theme=gr.themes.Default()
     )
     
-    config = KohyaSSGUIConfig(config_file_path=kwargs.get("config_file_path"))
+    config = KohyaSSGUIConfig(config_file_path=kwargs.get("config"))
 
     with interface:
         with gr.Tab("Dreambooth"):
diff --git a/kohya_gui/class_basic_training.py b/kohya_gui/class_basic_training.py
index 8599a1a..70270d8 100644
--- a/kohya_gui/class_basic_training.py
+++ b/kohya_gui/class_basic_training.py
@@ -302,7 +302,7 @@ class BasicTraining:
             )
             # Initialize the stop text encoder training slider
             self.stop_text_encoder_training = gr.Slider(
-                minimum=0,
+                minimum=-1,
                 maximum=100,
                 value=self.config.get("basic.stop_text_encoder_training", 0),
                 step=1,
diff --git a/kohya_gui/class_huggingface.py b/kohya_gui/class_huggingface.py
index 93001fc..e5baf57 100644
--- a/kohya_gui/class_huggingface.py
+++ b/kohya_gui/class_huggingface.py
@@ -5,7 +5,7 @@ from .class_gui_config import KohyaSSGUIConfig
 class HuggingFace:
     def __init__(
         self,
-        config: KohyaSSGUIConfig = {},
+        config: KohyaSSGUIConfig,
     ) -> None:
         self.config = config