src/transformers/generation/logits_process.py/SuppressTokensAtBeginLogitsProcessor/__call__
class SuppressTokensAtBeginLogitsProcessor:    def __call__(self, input_ids, scores):
        if input_ids.shape[1] == self.begin_index:
            scores[:, self.begin_suppress_tokens] = -float("inf")

        return scores