Error importing huggingface_hub.hub_mixin: module 'torch' has no attribute 'uint64'

[murlock1000]

Describe the bug

Started getting Error importing huggingface_hub.hub_mixin: module 'torch' has no attribute 'uint64' Error upon importing from huggingface_hub import PyTorchModelHubMixin.

Using:
pytorch version 2.2.0+cu121
huggingface-hub 0.34.3

Reproduction

torch version 2.2.0+cu121
huggingface-hub 0.34.3
`
from huggingface_hub import PyTorchModelHubMixin

Error importing huggingface_hub.hub_mixin: module 'torch' has no attribute 'uint64'
Error importing huggingface_hub.hub_mixin: module 'torch' has no attribute 'uint64'
Traceback (most recent call last):
File "", line 1, in
ImportError: cannot import name 'PyTorchModelHubMixin' from 'huggingface_hub' (/usr/local/lib/python3.10/dist-packages/huggingface_hub/init.py)
`

Logs

System info

- huggingface_hub version: 0.34.3
- Platform: Linux-6.8.0-52-generic-x86_64-with-glibc2.35
- Python version: 3.10.12
- Running in iPython ?: No
- Running in notebook ?: No
- Running in Google Colab ?: No
- Running in Google Colab Enterprise ?: No
- Token path ?: /root/.cache/huggingface/token
- Has saved token ?: False
- Configured git credential helpers: 
- FastAI: N/A
- Tensorflow: N/A
- Torch: 2.2.0
- Jinja2: 3.1.3
- Graphviz: N/A
- keras: N/A
- Pydot: N/A
- Pillow: 10.2.0
- hf_transfer: N/A
- gradio: 5.17.1
- tensorboard: N/A
- numpy: 1.26.3
- pydantic: 2.10.6
- aiohttp: N/A
- hf_xet: 1.1.7
- ENDPOINT: https://huggingface.co
- HF_HUB_CACHE: /root/.cache/huggingface/hub
- HF_ASSETS_CACHE: /root/.cache/huggingface/assets
- HF_TOKEN_PATH: /root/.cache/huggingface/token
- HF_STORED_TOKENS_PATH: /root/.cache/huggingface/stored_tokens
- HF_HUB_OFFLINE: False
- HF_HUB_DISABLE_TELEMETRY: False
- HF_HUB_DISABLE_PROGRESS_BARS: None
- HF_HUB_DISABLE_SYMLINKS_WARNING: False
- HF_HUB_DISABLE_EXPERIMENTAL_WARNING: False
- HF_HUB_DISABLE_IMPLICIT_TOKEN: False
- HF_HUB_ENABLE_HF_TRANSFER: False
- HF_HUB_ETAG_TIMEOUT: 10
- HF_HUB_DOWNLOAD_TIMEOUT: 10

[victkk]

I had the same issue...
I setup my environment 1 weeks ago and everything is fine. I setup the environment today again then I see this error.

[hanouticelina]

Hi @murlock1000 and @victkk,
the error comes from safetensors library, there is no reference of torch.uint64 in huggingface_hub. also, i don't have the error when downgrading safetensors: pip install "safetensors<0.6.1".

pinging @Narsil in case you have any insight on this

[KoichiYasuoka]

pip install safetensors==0.6.2 now fixes the bug.
huggingface/safetensors#643

[cMakawita]

ggml-org/llama.cpp#15134
this also worked

[hanouticelina]

closing this as the issue has been fixed in safetensors: https://github.com/huggingface/safetensors/releases/tag/v0.6.2
you just need to upgrade the package to the latest version:

pip install -U safetensors