Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-01-27 20:43:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

update reading from / writing to model config files for layered dirs

Browse Source
This commit is contained in:
Googolplexed0 2025-01-22 21:37:53 -05:00
parent 37cf3ae194
commit e1b8524381
No known key found for this signature in database
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/models_settings.py
View File

@ -28,7 +28,7 @@ def get_model_metadata(model):
# Get settings from models/config.yaml and models/config-user.yaml # Get settings from models/config.yaml and models/config-user.yaml
settings = shared.model_config settings = shared.model_config
for pat in settings: for pat in settings:
if re.match(pat.lower(), model.lower()): if re.match(pat.lower(), Path(model).name.lower()):
for k in settings[pat]: for k in settings[pat]:
model_settings[k] = settings[pat][k] model_settings[k] = settings[pat][k]
@ -140,7 +140,7 @@ def get_model_metadata(model):
# Apply user settings from models/config-user.yaml # Apply user settings from models/config-user.yaml
settings = shared.user_config settings = shared.user_config
for pat in settings: for pat in settings:
if re.match(pat.lower(), model.lower()): if re.match(pat.lower(), Path(model).name.lower()):
for k in settings[pat]: for k in settings[pat]:
model_settings[k] = settings[pat][k] model_settings[k] = settings[pat][k]
@ -242,7 +242,7 @@ def save_model_settings(model, state):
return return
user_config = shared.load_user_config() user_config = shared.load_user_config()
model_regex = model + '$' # For exact matches model_regex = Path(model).name + '$' # For exact matches
if model_regex not in user_config: if model_regex not in user_config:
user_config[model_regex] = {} user_config[model_regex] = {}
@ -269,7 +269,7 @@ def save_instruction_template(model, template):
return return
user_config = shared.load_user_config() user_config = shared.load_user_config()
model_regex = model + '$' # For exact matches model_regex = Path(model).name + '$' # For exact matches
if model_regex not in user_config: if model_regex not in user_config:
user_config[model_regex] = {} user_config[model_regex] = {}