text: Title: Exploding gradients/weights/logits in a fine-tuning notebook
  Body: Hello, after I successfully fine-tuned, I run to the last step and when generating text, the following error occurs. 
  May I ask what is the reason?
  
  ![image](https://user-images.githubusercontent.com/7599454/224216132-416b7044-dee1-4401-a7c7-c2f76ea37660.png)
  ![image](https://user-images.githubusercontent.com/7599454/224216173-81b75b67-79d0-4b6f-a654-2cf500ea07d5.png)
  