apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[Feature] add mm token usage (#4570)

Browse Source 
* add mm token usage

* fix unit test

* fix unit test

* fix unit test

* fix model path

* fix unit test

* fix unit test

* fix unit test

* remove uncomment

* change var name

* fix code style

* fix code style

* fix code style

* fix code style

* fix unit test
This commit is contained in:
ApplEOFDiscord
2025-10-29 14:37:12 +08:00
committed by GitHub
parent fc5cd1adb1
commit 14f8cddaf1
9 changed files with 70 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/output/token_processor.py
+6
View File
@@ -289,6 +289,9 @@ class TokenProcessor:
if task.messages is not None:
result.prompt = task.messages
result.num_cached_tokens = task.num_cached_tokens
if task.get("multimodal_inputs", None):
result.num_input_image_tokens = task.multimodal_inputs.get("num_input_image_tokens", 0)
result.num_input_video_tokens = task.multimodal_inputs.get("num_input_video_tokens", 0)
is_prefill = task.disaggregate_info is not None and task.disaggregate_info["role"] == "prefill"
result = self._process_per_token(task, i, token_ids, result, is_prefill)
@@ -655,6 +658,9 @@ class TokenProcessor:
if task.messages is not None:
result.prompt = task.messages
result.num_cached_tokens = task.num_cached_tokens
if task.get("multimodal_inputs", None):
result.num_input_image_tokens = task.multimodal_inputs.get("num_input_image_tokens", 0)
result.num_input_video_tokens = task.multimodal_inputs.get("num_input_video_tokens", 0)
is_prefill = task.disaggregate_info is not None and task.disaggregate_info["role"] == "prefill"