mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-28 10:03:54 +08:00
参数补充
This commit is contained in:
@@ -24,6 +24,7 @@ class LLMProvider(LLMProviderBase):
|
|||||||
"max_tokens": int,
|
"max_tokens": int,
|
||||||
"temperature": lambda x: round(float(x), 1),
|
"temperature": lambda x: round(float(x), 1),
|
||||||
"top_p": lambda x: round(float(x), 1),
|
"top_p": lambda x: round(float(x), 1),
|
||||||
|
"top_k": int,
|
||||||
"frequency_penalty": lambda x: round(float(x), 1),
|
"frequency_penalty": lambda x: round(float(x), 1),
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -39,7 +40,7 @@ class LLMProvider(LLMProviderBase):
|
|||||||
setattr(self, param, None)
|
setattr(self, param, None)
|
||||||
|
|
||||||
logger.debug(
|
logger.debug(
|
||||||
f"意图识别参数初始化: {self.temperature}, {self.max_tokens}, {self.top_p}, {self.frequency_penalty}"
|
f"意图识别参数初始化: {self.temperature}, {self.max_tokens}, {self.top_p}, {self.top_k}, {self.frequency_penalty}"
|
||||||
)
|
)
|
||||||
|
|
||||||
model_key_msg = check_model_key("LLM", self.api_key)
|
model_key_msg = check_model_key("LLM", self.api_key)
|
||||||
@@ -70,6 +71,7 @@ class LLMProvider(LLMProviderBase):
|
|||||||
"max_tokens": kwargs.get("max_tokens", self.max_tokens),
|
"max_tokens": kwargs.get("max_tokens", self.max_tokens),
|
||||||
"temperature": kwargs.get("temperature", self.temperature),
|
"temperature": kwargs.get("temperature", self.temperature),
|
||||||
"top_p": kwargs.get("top_p", self.top_p),
|
"top_p": kwargs.get("top_p", self.top_p),
|
||||||
|
"top_k": kwargs.get("top_k", self.top_k),
|
||||||
"frequency_penalty": kwargs.get("frequency_penalty", self.frequency_penalty),
|
"frequency_penalty": kwargs.get("frequency_penalty", self.frequency_penalty),
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -114,6 +116,7 @@ class LLMProvider(LLMProviderBase):
|
|||||||
"max_tokens": kwargs.get("max_tokens", self.max_tokens),
|
"max_tokens": kwargs.get("max_tokens", self.max_tokens),
|
||||||
"temperature": kwargs.get("temperature", self.temperature),
|
"temperature": kwargs.get("temperature", self.temperature),
|
||||||
"top_p": kwargs.get("top_p", self.top_p),
|
"top_p": kwargs.get("top_p", self.top_p),
|
||||||
|
"top_k": kwargs.get("top_k", self.top_k),
|
||||||
"frequency_penalty": kwargs.get("frequency_penalty", self.frequency_penalty),
|
"frequency_penalty": kwargs.get("frequency_penalty", self.frequency_penalty),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user