mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
Download patch file Download diff file Expand all files Collapse all files

12
server.py
View file

@ -72,7 +72,6 @@ if args.settings is not None and Path(args.settings).exists():
with open(Path(args.settings), 'r') as f:
new_settings = json.load(f)
for item in new_settings:
if item in settings:
settings[item] = new_settings[item]
def load_model(model_name):
@ -608,17 +607,21 @@ if args.chat or args.cai_chat:
if args.extensions is not None:
extensions_ui_elements = []
default_values = []
gr.Markdown('## Extensions parameters')
for ext in sorted(extension_state, key=lambda x : extension_state[x][1]):
if extension_state[ext][0] == True:
params = eval(f"extensions.{ext}.script.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:
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]:
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:
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):
i = 0
@ -630,6 +633,7 @@ if args.chat or args.cai_chat:
params[param] = eval(f"kwargs[{i}]")
i += 1
update_extensions_parameters(*default_values)
btn_extensions = gr.Button("Apply")
btn_extensions.click(update_extensions_parameters, [*extensions_ui_elements], [])