Add help messages to the SDXL finetune UI.

invoke-ai · May 21, 2024 · 654bba1 · 654bba1
1 parent 399c130
commit 654bba1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/invoke_training/ui/config_groups/sdxl_finetune_config_group.py b/src/invoke_training/ui/config_groups/sdxl_finetune_config_group.py
@@ -49,11 +49,15 @@ def __init__(self):
                     self.base_pipeline_config_group = BasePipelineConfigGroup()
                     self.save_checkpoint_format = gr.Dropdown(
                         label="Checkpoint Format",
+                        info="The save format for the checkpoints. `full_diffusers` saves the full model in diffusers "
+                        "format. `trained_only_diffusers` saves only the parts of the model that were finetuned "
+                        "(i.e. the UNet).",
                         choices=get_typing_literal_options(SdxlFinetuneConfig, "save_checkpoint_format"),
                         interactive=True,
                     )
                     self.save_dtype = gr.Dropdown(
                         label="Save Dtype",
+                        info="The dtype to use when saving the model.",
                         choices=get_typing_literal_options(SdxlFinetuneConfig, "save_dtype"),
                         interactive=True,
                     )