(
self,
input_ids: torch.LongTensor,
next_scores: torch.FloatTensor,
next_tokens: torch.LongTensor,
next_indices: torch.LongTensor,
**kwargs
)
| 128 | |
| 129 | @abstractmethod |
| 130 | def finalize( |
| 131 | self, |
| 132 | input_ids: torch.LongTensor, |
| 133 | next_scores: torch.FloatTensor, |
| 134 | next_tokens: torch.LongTensor, |
| 135 | next_indices: torch.LongTensor, |
| 136 | **kwargs |
| 137 | ) -> torch.LongTensor: |
| 138 | raise NotImplementedError("This is an abstract method.") |
| 139 | |
| 140 | |
| 141 | class BeamSearchScorer(BeamScorer): |
nothing calls this directly
no outgoing calls
no test coverage detected