Tuchuanhuhuhu commited on
Commit
7392813
·
1 Parent(s): 9e09956

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

Browse files
Files changed (1) hide show
  1. 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=MODEL_METADATA[model_name]["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,