diff --git a/modules/ui_models.py b/modules/ui_models.py index 3a5a29c3b..a216f3acf 100644 --- a/modules/ui_models.py +++ b/modules/ui_models.py @@ -179,7 +179,7 @@ def create_ui(): custom_name = gr.Textbox(label="New model name") with gr.Row(): merge_mode = gr.Dropdown(choices=merge_methods.__all__, value="weighted_sum", label="Interpolation Method") - merge_mode_docs = gr.HTML(value=(getattr(merge_methods, "weighted_sum", "").__doc__ or 'Error: Documentation missing').replace("\n", "
")) + merge_mode_docs = gr.HTML(value=merge_methods.weighted_sum.__doc__.replace("\n", "
")) # pylint: disable=no-member # pyright: ignore[reportOptionalMemberAccess] with gr.Row(): primary_model_name = gr.Dropdown(sd_model_choices(), label="Primary model", value="None") create_refresh_button(primary_model_name, sd_models.list_models, lambda: {"choices": sd_model_choices()}, "checkpoint_A_refresh") @@ -319,7 +319,11 @@ def create_ui(): return gr.Slider.update(value=None, visible=False) def show_help(mode): - doc = getattr(merge_methods, mode).__doc__.replace("\n", "
") + try: + doc = getattr(merge_methods, mode).__doc__.replace("\n", "
") + except AttributeError: + log.warning(f'Merge mode "{mode}" is missing documentation') + doc = "Error: Documentation missing" return gr.update(value=doc, visible=True) def show_unload(device):