(self, stops=[], encounters=1)
| 37 | class StoppingCriteriaSub(StoppingCriteria): |
| 38 | |
| 39 | def __init__(self, stops=[], encounters=1): |
| 40 | super().__init__() |
| 41 | self.stops = stops |
| 42 | |
| 43 | def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor): |
| 44 | for stop in self.stops: |
nothing calls this directly
no outgoing calls
no test coverage detected