Re-adding support for u16, u32, u64.

[Narsil]

What does this PR do?

Fixes # (issue) or description of the problem this PR solves.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[BenjaminBossan]

@Narsil What are the supported PyTorch versions? In v2.2.0, which is still supported by transformers, there is no torch.uint64, resulting in a failure to import safetensors.torch.

[surojit-ts]

@Narsil What are the supported PyTorch versions? In v2.2.0, which is still supported by transformers, there is no torch.uint64, resulting in a failure to import safetensors.torch.

Second this, I am also facing the same issue.

[zhichao-aws]

In v2.2.0, which is still supported by transformers, there is no torch.uint64, resulting in a failure to import safetensors.torch

Get the same exception when import PretrainedModel from transformers. I'm using transformers 4.44.1 and torch 2.0