src/transformers/generation/logits_process.py/ForceTokensLogitsProcessor/__call__
class ForceTokensLogitsProcessor:    def __call__(self, input_ids, scores):
        generation_idx = input_ids.shape[-1]
        current_token = self.force_token_map.get(generation_idx, None)
        if current_token is not None:
            scores[:, :] = -float("inf")
            scores[:, current_token] = 0
        return scores