Fix bug with /internal/model/load (#4549)

Update shared.model_name after loading model through API call
2023-11-09 22:16:38 -05:00 · 2023-11-09 22:16:38 -05:00 · 6a7cd01ebf
commit 6a7cd01ebf
parent 2af7e382b1
1 changed files with 2 additions and 1 deletions
--- a/extensions/openai/models.py
+++ b/extensions/openai/models.py
@ -55,6 +55,7 @@ def _load_model(data):
                setattr(shared.args, k, args[k])

    shared.model, shared.tokenizer = load_model(model_name)
+    shared.model_name = model_name
    
    # Update shared.settings with custom generation defaults
    if settings: