Model.generate() -- IndexError: too many indices for tensor of dimension 2

model_checkpoint = "distilgpt2" (removed when I edited the original question)

Thought gpt2 was autoregressive and suitable for text-generation. Will try another model :slight_smile:

Thank you Niels!