bugfix: 修复使用finetune的GPT模型时出错的问题

GaiZhenbiao · Nov 17, 2023 · 7392813 · 7392813
1 parent 9e09956
commit 7392813
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/models/OpenAI.py b/modules/models/OpenAI.py
@@ -25,8 +25,10 @@ def __init__(
         top_p=1.0,
         user_name=""
     ) -> None:
+        if model_name in MODEL_METADATA:
+            model_name = MODEL_METADATA[model_name]["model_name"]
         super().__init__(
-            model_name=MODEL_METADATA[model_name]["model_name"],
+            model_name=model_name,
             temperature=temperature,
             top_p=top_p,
             system_prompt=system_prompt,