Torch >= 2.2.0 bug on MPS

[davmacario]

When running inference from a pretrained model (huggingface) using MPS and a torch version above 2.2.0, I only get '!' at output (token 0).