src/transformers/generation/logits_process.py/LogitNormalization/__call__
class LogitNormalization:    def __call__(self, input_ids: torch.Tensor, scores: torch.Tensor) -> torch.Tensor:
        scores = scores.log_softmax(dim=-1)
        return scores