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):