Add backward compatibility with saved n_gpu_layers values

This commit is contained in:
oobabooga 2025-05-16 17:29:18 -07:00
parent 1c549d176b
commit d99fb0a22a

View file

@ -154,6 +154,9 @@ def get_model_metadata(model):
for pat in settings:
if re.match(pat.lower(), Path(model).name.lower()):
for k in settings[pat]:
if k == 'n_gpu_layers':
k = 'gpu_layers'
model_settings[k] = settings[pat][k]
# Load instruction template if defined by name rather than by value