AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update server.py - clear LORA after reload (#2952)

Browse Source
This commit is contained in:
FartyPants 2023-07-03 23:13:38 -04:00 committed by GitHub
parent 7e8340b14d
commit eb6112d5a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server.py
View File

@ -273,17 +273,20 @@ 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)
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'))
unload.click(
unload_model, None, None).then(
lambda: "Model unloaded", None, gradio('model_status'))
lambda: "Model unloaded", None, gradio('model_status')).then(
lambda: shared.lora_names, None, gradio('lora_menu'))
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)
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'))
save_settings.click(
ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(