Small bug fix in /v1/internal/model/load
This commit is contained in:
parent
050ff36bd6
commit
881e8a6e70
1 changed files with 1 additions and 1 deletions
|
@ -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…
Add table
Add a link
Reference in a new issue