Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/extensions.py
View File

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