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

FIX save_model_settings function to also update shared.model_config (#2282)

Browse Source
This commit is contained in:
Gabriel Terrien 2023-05-24 15:01:07 +02:00 committed by GitHub
parent d37a28730d
commit fc116711b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@ -289,11 +289,16 @@ def save_model_settings(model, state):
user_config = {} user_config = {}
model_regex = model + '$' # For exact matches model_regex = model + '$' # For exact matches
for _dict in [user_config, shared.model_config]:
if model_regex not in _dict:
_dict[model_regex] = {}
if model_regex not in user_config: if model_regex not in user_config:
user_config[model_regex] = {} user_config[model_regex] = {}
for k in ui.list_model_elements(): for k in ui.list_model_elements():
user_config[model_regex][k] = state[k] user_config[model_regex][k] = state[k]
shared.model_config[model_regex][k] = state[k]
with open(p, 'w') as f: with open(p, 'w') as f:
f.write(yaml.dump(user_config)) f.write(yaml.dump(user_config))