GPT NEO X LLM cannot perform inference

Beginners
1

I am trying to perform inference from GPT NEO X, but i am getting an error

logits[:, -1].view(batch_size, -1).contiguous() TypeError: tuple indices must be integers or slices, not tuple

Can anyone tell me how can i resolve this?

More information can be found here

Thanks