Incompatible shape error in tensorflow when I try to run generate function of gpt-2?

Question

I have finetuned gpt-2 with my own dataset on colab and I managed to generate text but when I downloaded my model and tried to generate text in spyder I got this ValueError: Tr