Skip to content

Commit

Permalink
Fix gibberish token problem for prefill also (nod-ai#723)
Browse files Browse the repository at this point in the history
missed a line in nod-ai#665
  • Loading branch information
renxida authored Dec 20, 2024
1 parent fc9576b commit 2776186
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion shortfin/python/shortfin_apps/llm/components/service.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,7 +383,7 @@ async def run(self):
if self.phase == InferencePhase.PREFILL:
seq_lens_host = seq_lens.for_transfer()
with seq_lens_host.map(discard=True) as m:
m.fill(0)
m.fill(1)
m.items = [len(req.input_token_ids) for req in self.exec_requests]
seq_lens_host.copy_to(seq_lens)

Expand Down

0 comments on commit 2776186

Please sign in to comment.