gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix：修复LPMM学习问题

Browse Source
This commit is contained in:
墨梓柒
2025-07-16 19:58:19 +08:00
parent c0cde24529
commit 2229f98993
7 changed files with 313 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/llm_models/utils_model.py
View File

@@ -255,12 +255,11 @@ class LLMRequest:
if self.temp != 0.7:
payload["temperature"] = self.temp
# 添加enable_thinking参数（如果不是默认值False）
if not self.enable_thinking:
payload["enable_thinking"] = False
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
# 添加enable_thinking参数（仅在启用时添加）
if self.enable_thinking:
payload["enable_thinking"] = True
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
if self.max_tokens:
payload["max_tokens"] = self.max_tokens
@@ -670,12 +669,11 @@ class LLMRequest:
if self.temp != 0.7:
payload["temperature"] = self.temp
# 添加enable_thinking参数（如果不是默认值False）
if not self.enable_thinking:
payload["enable_thinking"] = False
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
# 添加enable_thinking参数（仅在启用时添加）
if self.enable_thinking:
payload["enable_thinking"] = True
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
if self.max_tokens:
payload["max_tokens"] = self.max_tokens