AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Check if app should display extensions ui

Browse Source
This commit is contained in:
Maya 2023-03-19 13:31:21 +00:00
parent 81c9d130f2
commit acdbd6b708
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/extensions.py
View File

@ -46,15 +46,18 @@ def create_extensions_block():
if _id in shared.settings: if _id in shared.settings:
extension.params[param] = shared.settings[_id] extension.params[param] = shared.settings[_id]
should_display_ui = False
# Running setup function # Running setup function
if not setup_called: if not setup_called:
for extension, name in iterator(): for extension, name in iterator():
if hasattr(extension, "setup"): if hasattr(extension, "setup"):
extension.setup() extension.setup()
if hasattr(extension, "ui"):
should_display_ui = True
setup_called = True setup_called = True
# Creating the extension ui elements # Creating the extension ui elements
if len(state) > 0: if should_display_ui:
with gr.Box(elem_id="extensions"): with gr.Box(elem_id="extensions"):
gr.Markdown("Extensions") gr.Markdown("Extensions")
for extension, name in iterator(): for extension, name in iterator():