mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-01 01:26:03 -04:00
Small bug fix in /v1/internal/model/load
This commit is contained in:
parent
050ff36bd6
commit
881e8a6e70
@ -51,7 +51,7 @@ def _load_model(data):
|
||||
# Update shared.args with custom model loading settings
|
||||
if args:
|
||||
for k in args:
|
||||
if k in shared.args:
|
||||
if hasattr(shared.args, k):
|
||||
setattr(shared.args, k, args[k])
|
||||
|
||||
shared.model, shared.tokenizer = load_model(model_name)
|
||||
|
Loading…
Reference in New Issue
Block a user