Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
·
7392813
1
Parent(s):
9e09956
bugfix: 修复使用finetune的GPT模型时出错的问题
Browse files- modules/models/OpenAI.py +3 -1
modules/models/OpenAI.py
CHANGED
@@ -25,8 +25,10 @@ class OpenAIClient(BaseLLMModel):
|
|
25 |
top_p=1.0,
|
26 |
user_name=""
|
27 |
) -> None:
|
|
|
|
|
28 |
super().__init__(
|
29 |
-
model_name=
|
30 |
temperature=temperature,
|
31 |
top_p=top_p,
|
32 |
system_prompt=system_prompt,
|
|
|
25 |
top_p=1.0,
|
26 |
user_name=""
|
27 |
) -> None:
|
28 |
+
if model_name in MODEL_METADATA:
|
29 |
+
model_name = MODEL_METADATA[model_name]["model_name"]
|
30 |
super().__init__(
|
31 |
+
model_name=model_name,
|
32 |
temperature=temperature,
|
33 |
top_p=top_p,
|
34 |
system_prompt=system_prompt,
|