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

Don't reset LoRA menu when loading a model

Browse source
This commit is contained in:
oobabooga 2023-07-17 12:50:25 -07:00
parent 4ce766414b
commit f83fdb9270
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
server.py
View file

@ -277,20 +277,17 @@ def create_model_menus():
load.click(
ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(
update_model_parameters, gradio('interface_state'), None).then(
partial(load_model_wrapper, autoload=True), gradio('model_menu', 'loader'), gradio('model_status'), show_progress=False).then(
lambda: shared.lora_names, None, gradio('lora_menu'))
partial(load_model_wrapper, autoload=True), gradio('model_menu', 'loader'), gradio('model_status'), show_progress=False)
unload.click(
unload_model, None, None).then(
lambda: "Model unloaded", None, gradio('model_status')).then(
lambda: shared.lora_names, None, gradio('lora_menu'))
lambda: "Model unloaded", None, gradio('model_status'))
reload.click(
unload_model, None, None).then(
ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(
update_model_parameters, gradio('interface_state'), None).then(
partial(load_model_wrapper, autoload=True), gradio('model_menu', 'loader'), gradio('model_status'), show_progress=False).then(
lambda: shared.lora_names, None, gradio('lora_menu'))
partial(load_model_wrapper, autoload=True), gradio('model_menu', 'loader'), gradio('model_status'), show_progress=False)
save_settings.click(
ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(