Skip to content

fix: image token output #4399

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Oct 16, 2025
Merged

fix: image token output #4399

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
4711f7f
fix: image token output
zhuangzhuang12 Oct 14, 2025
719b0fe
fix: code style
zhuangzhuang12 Oct 14, 2025
edfb729
fix: CompletionOutput.decode_type
zhuangzhuang12 Oct 15, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 8 additions & 3 deletions fastdeploy/engine/common_engine.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -705,9 +705,14 @@ def _zmq_send_generated_tokens(self):
for request_id, contents in results.items():
new_contents = []
for content in contents:
delta_text, token_ids = self._decode_token(
token_ids=content.outputs.token_ids, req_id=request_id, is_end=content.finished
)
decode_type = content.outputs.decode_type
delta_text = ""
if decode_type == 0:
delta_text, token_ids = self._decode_token(
token_ids=content.outputs.token_ids, req_id=request_id, is_end=content.finished
)
else:
token_ids = content.outputs.token_ids
if len(token_ids):
content.outputs.token_ids = token_ids
content.outputs.text = delta_text
Expand Down
2 changes: 2 additions & 0 deletions fastdeploy/engine/request.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -285,6 +285,7 @@ class CompletionOutput:
index: int
send_idx: int
token_ids: list[int]
decode_type: int = 0
logprob: Optional[float] = None
top_logprobs: Optional[LogprobsLists] = None
logprobs: Optional[SampleLogprobs] = None
Expand All @@ -302,6 +303,7 @@ def to_dict(self):
"send_idx": self.send_idx,
"token_ids": self.token_ids,
"logprob": self.logprob,
"decode_type": self.decode_type,
"top_logprobs": self.top_logprobs,
"logprobs": self.logprobs,
"draft_token_ids": self.draft_token_ids,
Expand Down
Loading