fix(mistral_chat_transformation.py): add missing comma (#9606)
This commit is contained in:
parent
79175ddb53
commit
340a63bace
@ -80,6 +80,7 @@ class MistralConfig(OpenAIGPTConfig):
|
||||
"temperature",
|
||||
"top_p",
|
||||
"max_tokens",
|
||||
"max_completion_tokens",
|
||||
"tools",
|
||||
"tool_choice",
|
||||
"seed",
|
||||
@ -105,6 +106,10 @@ class MistralConfig(OpenAIGPTConfig):
|
||||
for param, value in non_default_params.items():
|
||||
if param == "max_tokens":
|
||||
optional_params["max_tokens"] = value
|
||||
if (
|
||||
param == "max_completion_tokens"
|
||||
): # max_completion_tokens should take priority
|
||||
optional_params["max_tokens"] = value
|
||||
if param == "tools":
|
||||
optional_params["tools"] = value
|
||||
if param == "stream" and value is True:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user