Fix annotations
# What does this PR do?

This PR is minor. Just fixed wrong annotations.
All models were tested and confirmed based on Pytorch model.

I think all the codes I modified are made based on the template.
Please check if it is okay to modify the annotation in the template code.

## Before submitting
- [x] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
- [x] Did you read the [contributor guideline](https://github.com/huggingface/transformers/blob/main/CONTRIBUTING.md#start-contributing-pull-requests),
      Pull Request section?
_The documentation is not available anymore as the PR was closed or merged._