Skip to content

Commit d41f906

Browse files
Fix issues with flake8
1 parent 905e201 commit d41f906

3 files changed

Lines changed: 5 additions & 3 deletions

File tree

logits_processor_zoo/vllm/generation_length.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ def __init__(self, tokenizer: PreTrainedTokenizer, boost_factor: float,
4949
self.tokenizer = tokenizer
5050

5151
def clone(self):
52-
return GenLengthLogitsProcessor(self.tokenizer, self.boost_factor, self.p, self.complete_sentences, self.boost_token_str)
52+
return GenLengthLogitsProcessor(self.tokenizer, self.boost_factor, self.p,
53+
self.complete_sentences, self.boost_token_str)
5354

5455
def __call__(self, prompt_tokens_ids: List[int], past_token_ids: List[int], scores: torch.Tensor) -> torch.Tensor:
5556
gen_length = len(past_token_ids)

logits_processor_zoo/vllm/multiple_choice.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(self, tokenizer: PreTrainedTokenizer, choices: List[str] = None,
4545
delimiter: str = ".", boost_first_words: float = 0.0):
4646
self.tokenizer = tokenizer
4747
self.choices = choices
48-
self.delimeter = delimeter
48+
self.delimiter = delimiter
4949
if choices is None:
5050
choices = ["1", "2", "3", "4"]
5151

logits_processor_zoo/vllm/trigger_phrase.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,8 @@ def __init__(self, phrase: str, trigger_token_phrase: str, tokenizer: PreTrained
4747
self._reset()
4848

4949
def clone(self):
50-
return TriggerPhraseLogitsProcessor(self.phrase, self.trigger_token_phrase, self.tokenizer, self.trigger_count, self.trigger_after)
50+
return TriggerPhraseLogitsProcessor(self.phrase, self.trigger_token_phrase, self.tokenizer,
51+
self.trigger_count, self.trigger_after)
5152

5253
def _reset(self):
5354
self.index = -1

0 commit comments

Comments
 (0)