Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-01-11 21:10:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

Load extensions at startup

Browse Source
This commit is contained in:
oobabooga 2023-01-27 10:53:05 -03:00
parent 70e034589f
commit 1d1f931757
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.py
View File

@ -231,7 +231,6 @@ def apply_extensions(text, typ):
for ext in sorted(extension_state, key=lambda x : extension_state[x][1]):
if extension_state[ext][0] == True:
ext_string = f"extensions.{ext}.script"
exec(f"import {ext_string}")
if typ == "input":
text = eval(f"{ext_string}.input_modifier(text)")
else:
@ -258,8 +257,11 @@ extension_state = {}
if args.extensions is not None:
for i,ext in enumerate(args.extensions.split(',')):
if ext in available_extensions:
print(f'The extension "{ext}" is enabled.')
print(f'Loading the extension "{ext}"... ', end='')
ext_string = f"extensions.{ext}.script"
exec(f"import {ext_string}")
extension_state[ext] = [True, i]
print(f'Ok.')
# Choosing the default model
if args.model is not None: