xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 21:05:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] fix logit processor excceed vocab size issue (#6927)

Browse Source
This commit is contained in:
Fei 2024-07-31 01:16:01 -07:00 committed by GitHub
parent 533d1932d2
commit c0644cf9ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/entrypoints/openai/logits_processors.py
View File

@ -58,6 +58,12 @@ def get_logits_processors(
"Found token_id in logit_bias that is not " "Found token_id in logit_bias that is not "
"an integer or string representing an integer") from exc "an integer or string representing an integer") from exc
# Check if token_id is within the vocab size
for token_id, bias in clamped_logit_bias.items():
if token_id < 0 or token_id >= tokenizer.vocab_size:
raise ValueError("token_id in logit_bias contains "
"out-of-vocab token id")
def logit_bias_logits_processor(token_ids: List[int], def logit_bias_logits_processor(token_ids: List[int],
logits: torch.Tensor) -> torch.Tensor: logits: torch.Tensor) -> torch.Tensor:
for token_id, bias in clamped_logit_bias.items(): for token_id, bias in clamped_logit_bias.items():