safetensors.torch bug

[lawrence-cj]

Describe the bug

When using
import safetensors:

diffusers/src/diffusers/models/modeling_utils.py

Line 25 in f72b28c

import safetensors

then safetensors.torch:

diffusers/src/diffusers/models/modeling_utils.py

Line 111 in f72b28c

return safetensors.torch.load_file(checkpoint_file, device="cpu")

Bug reported AttributeError: module 'safetensors' has no attribute 'torch'.

Maybe we can change
from import safetensors
to import safetensors.torch as safetensors_torch

Reproduction

import safetensors
safetensors.torch

Logs

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'safetensors' has no attribute 'torch'

System Info

diffusers==0.23.0
safetensors==0.4.1

Who can help?

No response

[sayakpaul]

What's the code you're using that's causing this problem?

Cc: @patrickvonplaten here.

[lawrence-cj]

When loading .safetensor type ckpt. BTW, .bin ckpt won't cause the error.

[sayakpaul]

I mean could you provide a full code snippet for us to see?

[lawrence-cj]

Frist we run the below code:

from diffusers.models import AutoencoderKL
vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae")

However, we found the loading is normal now. It seems we made a mistake. Sry for disturbing...