(self, min_tokens: int, token_eos: int)
| 2491 | |
| 2492 | class MinTokensLogitsProcessor(LogitsProcessor): |
| 2493 | def __init__(self, min_tokens: int, token_eos: int): |
| 2494 | self.min_tokens = min_tokens |
| 2495 | self.token_eos = token_eos |
| 2496 | self.prompt_tokens = None |
| 2497 | |
| 2498 | def __call__( |
| 2499 | self, input_ids: npt.NDArray[np.intc], scores: npt.NDArray[np.single] |
nothing calls this directly
no outgoing calls
no test coverage detected