Fix dtype in radnomly initialized head
# What does this PR do?

This PR fixes the following [issue](https://github.com/huggingface/accelerate/issues/757) reported in Accelerate: when a model is loaded with `device_map="auto"` and some `torch_dtype`, the randomly initialized head does not respect said dtype.
_The documentation is not available anymore as the PR was closed or merged._