(self, max_ngram_size: int = 2, num_pred_tokens: int = 10)
| 18 | """Based on https://github.com/apoorvumang/prompt-lookup-decoding""" |
| 19 | |
| 20 | def __init__(self, max_ngram_size: int = 2, num_pred_tokens: int = 10): |
| 21 | self.max_ngram_size = max_ngram_size |
| 22 | self.num_pred_tokens = num_pred_tokens |
| 23 | |
| 24 | @staticmethod |
| 25 | def find_candidate_pred_tokens( |
nothing calls this directly
no outgoing calls
no test coverage detected