Update config code

Author: bmaltais

.gitignore

@@ -50,3 +50,4 @@ dataset/**
 !dataset/**/.gitkeep
 models
 data
+config.toml

kohya_gui/class_advanced_training.py

@@ -4,7 +4,6 @@
 from .common_gui import (
     get_folder_path,
     get_any_file_path,
-    scriptdir,
     list_files,
     list_dirs,
     create_refresh_button,
@@ -30,7 +29,7 @@ def __init__(
         headless: bool = False,
         finetuning: bool = False,
         training_type: str = "",
-        config:dict = {},
+        config: dict = {},
     ) -> None:
         """
         Initializes the AdvancedTraining class with given settings.
@@ -50,7 +49,9 @@ def __init__(
         # Determine the current directories for VAE and output, falling back to defaults if not specified.
         self.current_vae_dir = self.config.get("vae_dir", "./models/vae")
         self.current_state_dir = self.config.get("state_dir", "./outputs")
-        self.current_log_tracker_config_dir = self.config.get("log_tracker_config_dir", "./logs")
+        self.current_log_tracker_config_dir = self.config.get(
+            "log_tracker_config_dir", "./logs"
+        )
 
         # Define the behavior for changing noise offset type.
         def noise_offset_type_change(
@@ -406,10 +407,11 @@ def list_state_dirs(path):
                 info="The name of the specific wandb session",
             )
         with gr.Group(), gr.Row():
+
             def list_log_tracker_config_files(path):
                 self.current_log_tracker_config_dir = path if not path == "" else "."
                 return list(list_files(path, exts=[".json"], all=True))
-    
+
             self.log_tracker_name = gr.Textbox(
                 label="Log tracker name",
                 value="",
@@ -418,7 +420,8 @@ def list_log_tracker_config_files(path):
             )
             self.log_tracker_config = gr.Dropdown(
                 label="Log tracker config",
-                choices=[""] + list_log_tracker_config_files(self.current_log_tracker_config_dir),
+                choices=[""]
+                + list_log_tracker_config_files(self.current_log_tracker_config_dir),
                 value="",
                 info="Path to tracker config file to use for logging",
                 interactive=True,
@@ -427,7 +430,10 @@ def list_log_tracker_config_files(path):
             create_refresh_button(
                 self.log_tracker_config,
                 lambda: None,
-                lambda: {"choices": [""] + list_log_tracker_config_files(self.current_log_tracker_config_dir)},
+                lambda: {
+                    "choices": [""]
+                    + list_log_tracker_config_files(self.current_log_tracker_config_dir)
+                },
                 "open_folder_small",
             )
             self.log_tracker_config_button = gr.Button(
@@ -439,7 +445,9 @@ def list_log_tracker_config_files(path):
                 show_progress=False,
             )
             self.log_tracker_config.change(
-                fn=lambda path: gr.Dropdown(choices=[""] + list_log_tracker_config_files(path)),
+                fn=lambda path: gr.Dropdown(
+                    choices=[""] + list_log_tracker_config_files(path)
+                ),
                 inputs=self.log_tracker_config,
                 outputs=self.log_tracker_config,
                 show_progress=False,

kohya_gui/class_gui_config.py

@@ -43,6 +43,22 @@ def get(self, key, default=None):
         Returns:
         The value associated with the key, or the default value if the key is not found.
         """
-        if key not in self.config:
-            log.debug(f"Key '{key}' not found in configuration. Returning default value.")
-        return self.config.get(key, default)
+        # Split the key into a list of keys if it contains a dot (.)
+        keys = key.split(".")
+        # Initialize `data` with the entire configuration data
+        data = self.config
+
+        # Iterate over the keys to access nested values
+        for k in keys:
+            log.debug(k)
+            # If the key is not found in the current data, return the default value
+            if k not in data:
+                log.debug(f"Key '{key}' not found in configuration. Returning default value.")
+                return default
+
+            # Update `data` to the value associated with the current key
+            data = data.get(k)
+
+        # Return the final value
+        log.debug(f"Returned {data}")
+        return data