1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix: Correct config key access for sectioned values 

- Modified `class_folders.py` to use `folders.*` prefix for `output_dir`, `logging_dir`, and `reg_data_dir`.
- Modified `class_advanced_training.py` to use `advanced.log_tracker_config_dir` for its UI component.
- Modified `dreambooth_folder_creation_gui.py` to use `model.train_data_dir` for the 'Destination training directory' field.

These changes ensure that the GUI correctly loads default values from their respective sections in the `config.toml` file, resolving INFO messages about keys not being found when they were defined within sections.
fix/config-key-loading
google-labs-jules[bot] 2025-06-25 02:25:04 +00:00
parent 26a79fe7d4
commit d131355c0a
3 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kohya_gui/class_advanced_training.py
View File

@ -599,9 +599,9 @@ class AdvancedTraining:
)
self.log_tracker_config = gr.Dropdown(
label="Log tracker config",
choices=[self.config.get("log_tracker_config_dir", "")]
choices=[self.config.get("advanced.log_tracker_config_dir", "")]
+ list_log_tracker_config_files(self.current_log_tracker_config_dir),
value=self.config.get("log_tracker_config_dir", ""),
value=self.config.get("advanced.log_tracker_config_dir", ""),
info="Path to tracker config file to use for logging",
interactive=True,
allow_custom_value=True,
@ -610,7 +610,7 @@ class AdvancedTraining:
self.log_tracker_config,
lambda: None,
lambda: {
"choices": [self.config.get("log_tracker_config_dir", "")]
"choices": [self.config.get("advanced.log_tracker_config_dir", "")]
+ list_log_tracker_config_files(self.current_log_tracker_config_dir)
},
"open_folder_small",
@ -625,7 +625,7 @@ class AdvancedTraining:
)
self.log_tracker_config.change(
fn=lambda path: gr.Dropdown(
choices=[self.config.get("log_tracker_config_dir", "")]
choices=[self.config.get("advanced.log_tracker_config_dir", "")]
+ list_log_tracker_config_files(path)
),
inputs=self.log_tracker_config,

6
kohya_gui/class_folders.py
View File

@ -26,13 +26,13 @@ class Folders:
# Set default directories if not provided
self.current_output_dir = self.config.get(
"output_dir", os.path.join(scriptdir, "outputs")
"folders.output_dir", os.path.join(scriptdir, "outputs")
)
self.current_logging_dir = self.config.get(
"logging_dir", os.path.join(scriptdir, "logs")
"folders.logging_dir", os.path.join(scriptdir, "logs")
)
self.current_reg_data_dir = self.config.get(
"reg_data_dir", os.path.join(scriptdir, "reg")
"folders.reg_data_dir", os.path.join(scriptdir, "reg")
)
# Create directories if they don't exist

6
kohya_gui/dreambooth_folder_creation_gui.py
View File

@ -250,9 +250,9 @@ def gradio_dreambooth_folder_creation_tab(
util_training_dir_output = gr.Dropdown(
label="Destination training directory (where formatted training and regularisation folders will be placed)",
interactive=True,
choices=[config.get(key="train_data_dir", default="")]
choices=[config.get(key="model.train_data_dir", default="")]
+ list_train_output_dirs(current_train_output_dir),
value=config.get(key="train_data_dir", default=""),
value=config.get(key="model.train_data_dir", default=""),
allow_custom_value=True,
)
create_refresh_button(
@ -272,7 +272,7 @@ def gradio_dreambooth_folder_creation_tab(
)
util_training_dir_output.change(
fn=lambda path: gr.Dropdown(
choices=[config.get(key="train_data_dir", default="")] + list_train_output_dirs(path)
choices=[config.get(key="model.train_data_dir", default="")] + list_train_output_dirs(path)
),
inputs=util_training_dir_output,
outputs=util_training_dir_output,