mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
Update ui_models.py
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user