Skip to content

Commit 4f28f43

Browse files
committed
fix format
Signed-off-by: Sungjae Lee <33976427+llsj14@users.noreply.github.com> Signed-off-by: Sungjae Lee <sung-jae.lee@navercorp.com>
1 parent 5d49100 commit 4f28f43

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/v1/sample/logits_processor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -585,7 +585,7 @@ def _init_state_entry(self, prompt_tok_ids: list[int],
585585
"in_think": in_think, # Currently in thinking mode
586586
"in_end": False, # Currently forcing end tokens
587587
"think_count": think_count, # Number of tokens in thinking section
588-
"end_count": 0, # Number of end tokens forced so far
588+
"end_count": 0, # Number of end tokens forced so far
589589
"prompt_tok_ids": prompt_tok_ids,
590590
"output_tok_ids": [],
591591
"thinking_token_budget": thinking_token_budget,
@@ -664,7 +664,7 @@ def apply(self, logits: torch.Tensor) -> torch.Tensor:
664664
return logits
665665

666666
mask = torch.zeros(batch_size, dtype=torch.bool, device=logits.device)
667-
force_token_ids = torch.full((batch_size,),
667+
force_token_ids = torch.full((batch_size, ),
668668
-1,
669669
dtype=torch.long,
670670
device=logits.device)

0 commit comments

Comments
 (0)