apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

support logprob in scheduler v1 (#3249)
Some checks failed
Deploy GitHub Pages / deploy (push) Has been cancelled
Details

Browse Source 
Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
This commit is contained in:
chenjian
2025-08-07 20:14:01 +08:00
committed by GitHub
parent 6037dd5d9c
commit 5f0b30f6d0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/output/token_processor.py
View File

@@ -383,6 +383,7 @@ class TokenProcessor:
self.tokens_counter[task_id] += 1 self.tokens_counter[task_id] += 1
if token_id != RECOVERY_STOP_SIGNAL: if token_id != RECOVERY_STOP_SIGNAL:
result.outputs.token_ids.append(token_id) result.outputs.token_ids.append(token_id)
task.output_token_ids.append(token_id)
result.outputs.logprob = float(scores[i, 0]) result.outputs.logprob = float(scores[i, 0])
# Construct top_logprobs # Construct top_logprobs
topk_token_ids = tokens[i, :].tolist() topk_token_ids = tokens[i, :].tolist()