mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Support for number/bool extension parameters

Browse source
This commit is contained in:
oobabooga 2023-01-28 23:08:28 -03:00
parent 6da94e358c
commit 2239be2351
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
server.py
View file

@ -613,8 +613,12 @@ if args.chat or args.cai_chat:
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:
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=params[param], label=param))
elif type(params[param]) in [int, float]:
extensions_ui_elements.append(gr.Number(value=params[param], label=param))
elif type(params[param]) == bool:
extensions_ui_elements.append(gr.Checkbox(value=params[param], label=param))
def update_extensions_parameters(*kwargs): def update_extensions_parameters(*kwargs):
i = 0 i = 0