KohakuBlueleaf / HyperKohaku

A diffusers based implementation of HyperDreamBooth
Apache License 2.0
127 stars 7 forks source link

safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge #8

Open sasso-effe opened 1 year ago

sasso-effe commented 1 year ago

I followed all the steps:

Traceback (most recent call last):
  File "/home/wizard/bendai/research/dawnai/hyper_dreambooth/inference_test.py", line 19, in <module>
    pipe.unet.load_attn_procs(model_path)
  File "/home/wizard/mambaforge/envs/hyper/lib/python3.9/site-packages/diffusers/loaders.py", line 297, in load_attn_procs
    state_dict = safetensors.torch.load_file(model_file, device="cpu")
  File "/home/wizard/mambaforge/envs/hyper/lib/python3.9/site-packages/safetensors/torch.py", line 259, in load_file
    with safe_open(filename, framework="pt", device=device) as f:
safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge

Any idea? Thank you!