Skip to content

Commit 78b6c11

Browse files
Use initial_trigger_phrase instead of trigger_phrase
1 parent 92e310d commit 78b6c11

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

logits_processor_zoo/vllm/trigger_phrase.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def __init__(self, phrase: str, trigger_token_phrase: str, tokenizer: PreTrained
4848

4949
def clone(self):
5050
return TriggerPhraseLogitsProcessor(self.phrase, self.trigger_token_phrase, self.tokenizer,
51-
self.trigger_count, self.trigger_after)
51+
self.initial_trigger_count, self.trigger_after)
5252

5353
def _reset(self):
5454
self.index = -1

0 commit comments

Comments
 (0)