Skip to content

Commit db324d3

Browse files
make pre-commit happy again
Signed-off-by: Nathan Hoos <thwackyy.y@gmail.com>
1 parent 30464b5 commit db324d3

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

vllm/model_executor/guided_decoding/outlines_decoding.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -104,10 +104,13 @@ def _get_guide_and_mode(
104104

105105

106106
def _get_logits_processor(
107-
guide: str, tokenizer: PreTrainedTokenizerBase,
108-
mode: GuidedDecodingMode, whitespace_pattern: Union[str, None],
109-
reasoner: Optional[ReasoningParser],
110-
vocab_size: int,) -> Union[JSONLogitsProcessor, RegexLogitsProcessor]:
107+
guide: str,
108+
tokenizer: PreTrainedTokenizerBase,
109+
mode: GuidedDecodingMode,
110+
whitespace_pattern: Union[str, None],
111+
reasoner: Optional[ReasoningParser],
112+
vocab_size: int,
113+
) -> Union[JSONLogitsProcessor, RegexLogitsProcessor]:
111114
if mode == GuidedDecodingMode.JSON:
112115
return JSONLogitsProcessor(guide, tokenizer, whitespace_pattern,
113116
reasoner, vocab_size)

vllm/v1/structured_output/backend_outlines.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
import sre_constants
3333
import sre_parse
3434

35+
3536
@dataclass
3637
class OutlinesBackend(StructuredOutputBackend):
3738

@@ -125,7 +126,6 @@ def validate_tokens(self, tokens: list[int]) -> list[int]:
125126
accepted.pop()
126127
break
127128
return accepted
128-
129129

130130
def fill_bitmask(self, bitmask: torch.Tensor, idx: int) -> None:
131131
mask = bitmask[idx]

0 commit comments

Comments
 (0)