| 2471 | |
| 2472 | |
| 2473 | class LogitsProcessorList(List[LogitsProcessor]): |
| 2474 | def __call__( |
| 2475 | self, input_ids: npt.NDArray[np.intc], scores: npt.NDArray[np.single] |
| 2476 | ) -> npt.NDArray[np.single]: |
| 2477 | for processor in self: |
| 2478 | scores = processor(input_ids, scores) |
| 2479 | return scores |
| 2480 | |
| 2481 | |
| 2482 | StoppingCriteria = Callable[[npt.NDArray[np.intc], npt.NDArray[np.single]], bool] |
no outgoing calls
no test coverage detected
searching dependent graphs…