Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add setup method to extensions

Browse Source
This commit is contained in:
Maya 2023-03-19 13:22:24 +00:00
parent bd27353a08
commit 099d7a844b
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
extensions/api/script.py
View File

@ -8,8 +8,6 @@ params = {
'port': 5000, 'port': 5000,
} }
server = None
class Handler(BaseHTTPRequestHandler): class Handler(BaseHTTPRequestHandler):
def do_GET(self): def do_GET(self):
if self.path == '/api/v1/model': if self.path == '/api/v1/model':
@ -75,7 +73,6 @@ class Handler(BaseHTTPRequestHandler):
def run_server(): def run_server():
global server
server_addr = ('0.0.0.0' if shared.args.listen else '127.0.0.1', params['port']) server_addr = ('0.0.0.0' if shared.args.listen else '127.0.0.1', params['port'])
server = ThreadingHTTPServer(server_addr, Handler) server = ThreadingHTTPServer(server_addr, Handler)
if shared.args.share: if shared.args.share:
@ -89,6 +86,5 @@ def run_server():
print(f'Starting KoboldAI compatible api at http://{server_addr[0]}:{server_addr[1]}/api') print(f'Starting KoboldAI compatible api at http://{server_addr[0]}:{server_addr[1]}/api')
server.serve_forever() server.serve_forever()
def ui(): def setup():
if server is None: Thread(target=run_server, daemon=True).start()
Thread(target=run_server, daemon=True).start()

8
modules/extensions.py
View File

@ -5,6 +5,7 @@ import modules.shared as shared
state = {} state = {}
available_extensions = [] available_extensions = []
setup_called = False
def load_extensions(): def load_extensions():
global state global state
@ -44,6 +45,13 @@ 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]
# Running setup function
if not setup_called:
for extension, name in iterator():
if hasattr(extension, "setup"):
extension.setup()
setup_called = True
# Creating the extension ui elements # Creating the extension ui elements
if len(state) > 0: if len(state) > 0:
with gr.Box(elem_id="extensions"): with gr.Box(elem_id="extensions"):