[BUG]: loading sharded model does not handle `unexpected_keys`

[flymin]

Is there an existing issue for this bug?

• [X] I have searched the existing issues

🐛 Describe the bug

loading sharded model does not raise an error on unexpected_keys. I will open a pr for more details.

Environment

No response