【Bugfix】修复2.1分支上0.3B模型性能大幅下降 (#3624)

* 恢复异步方法。
【BugFix】completion接口echo回显支持 (#3245)

* wenxin-tools-511,修复v1/completion无法回显的问题。

* 支持多prompt的回显

* 支持多prompt情况下的流式回显

* 补充了 completion 接口支持 echo 的单元测试

* pre-commit

* 移除了多余的test文件

* 修复了completion接口echo支持的单测方法

* 补充了单元测试文件

* 补充单测

* unittest

* 补充单测

* 修复单测

* 删除不必要的assert.

* 重新提交

* 更新测试方法

* ut

* 验证是否是正确思路单测

* 验证是否是正确思路单测

* 验证是否是正确思路单测3

* 优化单测代码,有针对性地缩小单测范围。

* 优化单测代码2,有针对性地缩小单测范围。

* 优化单测代码3,有针对性地缩小单测范围。

* support 'echo' in chat/completion.

* update

* update

* update

* update

* update

* update

* 补充了关于tokenid的单元测试

* update

* 修正index错误

* 修正index错误

* [Bugfix] Significant performance degradation of 0.3B model on branch 2.1
This commit is contained in:
zhuzixuan
2025-08-27 15:29:01 +08:00
committed by GitHub
parent 96aed92e4a
commit 80db7fce05
2 changed files with 26 additions and 21 deletions

View File

@@ -240,7 +240,7 @@ class OpenAIServingCompletion:
dealer.close()
self.engine_client.semaphore.release()
def _echo_back_prompt(self, request, res, idx):
async def _echo_back_prompt(self, request, res, idx):
if res["outputs"].get("send_idx", -1) == 0 and request.echo:
if isinstance(request.prompt, list):
prompt_text = request.prompt[idx]
@@ -346,7 +346,7 @@ class OpenAIServingCompletion:
else:
arrival_time = res["metrics"]["arrival_time"] - inference_start_time[idx]
self._echo_back_prompt(request, res, idx)
await self._echo_back_prompt(request, res, idx)
output = res["outputs"]
output_top_logprobs = output["top_logprobs"]
logprobs_res: Optional[CompletionLogprobs] = None
@@ -471,7 +471,6 @@ class OpenAIServingCompletion:
else:
token_ids = output["token_ids"]
output_text = output["text"]
finish_reason = self.calc_finish_reason(request.max_tokens, final_res["output_token_ids"], output, False)
choice_data = CompletionResponseChoice(