zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Attempt at fixing preset loading

Browse Source
This commit is contained in:
oobabooga
2023-04-05 23:23:18 -03:00
parent 119726d986
commit c58fd41f46
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
server.py
View File

@@ -85,7 +85,7 @@ def load_lora_wrapper(selected_lora):
add_lora_to_model(selected_lora)
return selected_lora
def load_preset_values(preset_menu):
def load_preset_values(preset_menu, state):
generate_params = {
'do_sample': True,
'temperature': 1,
@@ -107,10 +107,10 @@ def load_preset_values(preset_menu):
i = i.rstrip(',').strip().split('=')
if len(i) == 2 and i[0].strip() != 'tokens':
generate_params[i[0].strip()] = eval(i[1].strip())
generate_params['temperature'] = min(1.99, generate_params['temperature'])
return generate_params
state.update(generate_params)
return generate_params, [gr.update(value=state[k]) for k in generate_params]
def upload_soft_prompt(file):
with zipfile.ZipFile(io.BytesIO(file)) as zf:
@@ -167,7 +167,7 @@ def create_prompt_menus():
shared.gradio['save_prompt'].click(save_prompt, [shared.gradio['textbox']], [shared.gradio['status']], show_progress=False)
def create_settings_menus(default_preset):
generate_params = load_preset_values(default_preset if not shared.args.flexgen else 'Naive')
generate_params, _ = load_preset_values(default_preset if not shared.args.flexgen else 'Naive', {})
for k in ['max_new_tokens', 'seed', 'stop_at_newline', 'chat_prompt_size', 'chat_generation_attempts']:
generate_params[k] = shared.settings[k]
shared.gradio['generation_state'] = gr.State(generate_params)
@@ -222,7 +222,7 @@ def create_settings_menus(default_preset):
shared.gradio['upload_softprompt'] = gr.File(type='binary', file_types=['.zip'])
shared.gradio['model_menu'].change(load_model_wrapper, shared.gradio['model_menu'], shared.gradio['model_menu'], show_progress=True)
shared.gradio['preset_menu'].change(load_preset_values, shared.gradio['preset_menu'], shared.gradio['generation_state'])
shared.gradio['preset_menu'].change(load_preset_values, shared.gradio['preset_menu'], [shared.gradio[k] for k in ['generation_state'] + [k for k in shared.gradio['generation_state']][0:10]])
shared.gradio['lora_menu'].change(load_lora_wrapper, shared.gradio['lora_menu'], shared.gradio['lora_menu'], show_progress=True)
shared.gradio['softprompts_menu'].change(load_soft_prompt, shared.gradio['softprompts_menu'], shared.gradio['softprompts_menu'], show_progress=True)
shared.gradio['upload_softprompt'].upload(upload_soft_prompt, shared.gradio['upload_softprompt'], shared.gradio['softprompts_menu'])