Transform Logits to probabilities doesn't work

[sgugger]

The predictions are a NumPy array, so you need to convert them to a torch Tensor if you want to use PyTorch functions on them.