mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-01-02 23:00:12 +01:00
Add backward compatibility with saved n_gpu_layers values
This commit is contained in:
parent
1c549d176b
commit
d99fb0a22a
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue