diff --git a/kohya_gui/class_advanced_training.py b/kohya_gui/class_advanced_training.py index ffed0202e..b18ec7bfb 100644 --- a/kohya_gui/class_advanced_training.py +++ b/kohya_gui/class_advanced_training.py @@ -454,11 +454,13 @@ def full_options_update(full_fp16, full_bf16): self.save_state = gr.Checkbox( label="Save training state", value=self.config.get("advanced.save_state", False), + info="Save training state (including optimizer states etc.) when saving models" ) self.save_state_on_train_end = gr.Checkbox( label="Save training state at end of training", value=self.config.get("advanced.save_state_on_train_end", False), + info="Save training state (including optimizer states etc.) on train end" ) def list_state_dirs(path): @@ -472,6 +474,7 @@ def list_state_dirs(path): value=self.config.get("advanced.state_dir", ""), interactive=True, allow_custom_value=True, + info="Saved state to resume training from" ) create_refresh_button( self.resume, diff --git a/kohya_gui/class_basic_training.py b/kohya_gui/class_basic_training.py index 15497ea5d..4a7eb54bb 100644 --- a/kohya_gui/class_basic_training.py +++ b/kohya_gui/class_basic_training.py @@ -117,7 +117,7 @@ def init_training_controls(self) -> None: # Initialize the caption extension input self.caption_extension = gr.Dropdown( label="Caption file extension", - choices=[".cap", ".caption", ".txt"], + choices=["", ".cap", ".caption", ".txt"], value=".txt", interactive=True, ) diff --git a/kohya_gui/convert_model_gui.py b/kohya_gui/convert_model_gui.py index 55639a91b..14cb71eb4 100644 --- a/kohya_gui/convert_model_gui.py +++ b/kohya_gui/convert_model_gui.py @@ -199,6 +199,7 @@ def list_target_folder(path): "runwayml/stable-diffusion-v1-5", "CompVis/stable-diffusion-v1-4", ], + allow_custom_value=True, ) with gr.Group(), gr.Row(): with gr.Column(), gr.Row():