mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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:
user_config = {}
model_regex = model + '$' # For exact matches
if model not in user_config:
user_config[model] = {}
user_config[model_regex] = {}
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:
f.write(yaml.dump(user_config))