diff --git a/modules/extensions.py b/modules/extensions.py index fe8cb7be..2b7af020 100644 --- a/modules/extensions.py +++ b/modules/extensions.py @@ -173,5 +173,7 @@ def create_extensions_block(): with gr.Column(elem_id="extensions"): for extension, name in iterator(): if hasattr(extension, "ui"): - gr.Markdown(f"\n### {name}") + gr.Markdown(f"\n### {extension.params['display_name']}" + if hasattr(extension, 'params') and extension.params.__contains__('display_name') + else f"\n### {name}") extension.ui()