zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix loading extensions from within the interface

Browse Source
This commit is contained in:
oobabooga
2023-03-28 23:27:02 -03:00
committed by GitHub
parent 304f812c63
commit 1edfb96778
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
modules/extensions.py
View File

@@ -7,7 +7,7 @@ import modules.shared as shared
state = {} state = {}
available_extensions = [] available_extensions = []
setup_called = False setup_called = set()
def load_extensions(): def load_extensions():
global state global state
@@ -53,13 +53,12 @@ def create_extensions_block():
should_display_ui = False should_display_ui = False
# Running setup function # Running setup function
if not setup_called: for extension, name in iterator():
for extension, name in iterator(): if hasattr(extension, "ui"):
if hasattr(extension, "setup"): should_display_ui = True
extension.setup() if extension not in setup_called and hasattr(extension, "setup"):
if hasattr(extension, "ui"): setup_called.add(extension)
should_display_ui = True extension.setup()
setup_called = True
# Creating the extension ui elements # Creating the extension ui elements
if should_display_ui: if should_display_ui: