From 6a7cd01ebf8021a8ee6da094643f09da41516ccd Mon Sep 17 00:00:00 2001
From: GuizzyQC <86683381+GuizzyQC@users.noreply.github.com>
Date: Thu, 9 Nov 2023 22:16:38 -0500
Subject: [PATCH] Fix bug with /internal/model/load (#4549)

Update shared.model_name after loading model through API call
---
 extensions/openai/models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/extensions/openai/models.py b/extensions/openai/models.py
index a737f0c6..178ede90 100644
--- a/extensions/openai/models.py
+++ b/extensions/openai/models.py
@@ -55,7 +55,8 @@ 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:
         for k in settings: