Update ui_models.py

This commit is contained in:
awsr
2026-01-13 15:33:18 -08:00
parent 1dffdc17a9
commit de52f6bfda
+6 -2
View File
@@ -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", "<br>"))
merge_mode_docs = gr.HTML(value=merge_methods.weighted_sum.__doc__.replace("\n", "<br>")) # 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", "<br>")
try:
doc = getattr(merge_methods, mode).__doc__.replace("\n", "<br>")
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):