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

Read extensions parameters from settings file

Browse Source
This commit is contained in:
oobabooga 2023-01-28 23:21:40 -03:00
parent 2239be2351
commit e349b52256
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server.py
View File

@ -72,8 +72,7 @@ if args.settings is not None and Path(args.settings).exists():
with open(Path(args.settings), 'r') as f: with open(Path(args.settings), 'r') as f:
new_settings = json.load(f) new_settings = json.load(f)
for item in new_settings: for item in new_settings:
if item in settings: settings[item] = new_settings[item]
settings[item] = new_settings[item]
def load_model(model_name): def load_model(model_name):
print(f"Loading {model_name}...") print(f"Loading {model_name}...")
@ -608,17 +607,21 @@ if args.chat or args.cai_chat:
if args.extensions is not None: if args.extensions is not None:
extensions_ui_elements = [] extensions_ui_elements = []
default_values = []
gr.Markdown('## Extensions parameters') gr.Markdown('## Extensions parameters')
for ext in sorted(extension_state, key=lambda x : extension_state[x][1]): for ext in sorted(extension_state, key=lambda x : extension_state[x][1]):
if extension_state[ext][0] == True: if extension_state[ext][0] == True:
params = eval(f"extensions.{ext}.script.params") params = eval(f"extensions.{ext}.script.params")
for param in params: for param in params:
_id = f"{ext}-{param}"
default_value = settings[_id] if _id in settings else params[param]
default_values.append(default_value)
if type(params[param]) == str: if type(params[param]) == str:
extensions_ui_elements.append(gr.Textbox(value=params[param], label=param)) extensions_ui_elements.append(gr.Textbox(value=default_value, label=f"{ext}-{param}"))
elif type(params[param]) in [int, float]: elif type(params[param]) in [int, float]:
extensions_ui_elements.append(gr.Number(value=params[param], label=param)) extensions_ui_elements.append(gr.Number(value=default_value, label=f"{ext}-{param}"))
elif type(params[param]) == bool: elif type(params[param]) == bool:
extensions_ui_elements.append(gr.Checkbox(value=params[param], label=param)) extensions_ui_elements.append(gr.Checkbox(value=default_value, label=f"{ext}-{param}"))
def update_extensions_parameters(*kwargs): def update_extensions_parameters(*kwargs):
i = 0 i = 0
@ -630,6 +633,7 @@ if args.chat or args.cai_chat:
params[param] = eval(f"kwargs[{i}]") params[param] = eval(f"kwargs[{i}]")
i += 1 i += 1
update_extensions_parameters(*default_values)
btn_extensions = gr.Button("Apply") btn_extensions = gr.Button("Apply")
btn_extensions.click(update_extensions_parameters, [*extensions_ui_elements], []) btn_extensions.click(update_extensions_parameters, [*extensions_ui_elements], [])