Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Make the regex for "Save settings for this model" exact

Browse Source
This commit is contained in:
oobabooga 2023-05-12 00:43:13 -03:00
parent f98fd01dcd
commit a77965e801
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@ -264,11 +264,12 @@ def save_model_settings(model, state):
else: else:
user_config = {} user_config = {}
model_regex = model + '$' # For exact matches
if model not in user_config: if model not in user_config:
user_config[model] = {} user_config[model_regex] = {}
for k in ui.list_model_elements(): for k in ui.list_model_elements():
user_config[model][k] = state[k] user_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))