UniDepth icon indicating copy to clipboard operation
UniDepth copied to clipboard

safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge

Open CharlotteHao opened this issue 1 year ago • 1 comments

Traceback (most recent call last): File "/home/shahao/my_study/PF3plat/UniDepth/UniDepth/./scripts/demo.py", line 46, in model = UniDepthV2.from_pretrained(f"lpiccinelli/{name}") ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/shahao/anaconda3/envs/unid/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^ File "/home/shahao/anaconda3/envs/unid/lib/python3.11/site-packages/huggingface_hub/hub_mixin.py", line 571, in from_pretrained instance = cls._from_pretrained( ^^^^^^^^^^^^^^^^^^^^^ File "/home/shahao/anaconda3/envs/unid/lib/python3.11/site-packages/huggingface_hub/hub_mixin.py", line 795, in _from_pretrained return cls._load_as_safetensor(model, model_file, map_location, strict) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/shahao/anaconda3/envs/unid/lib/python3.11/site-packages/huggingface_hub/hub_mixin.py", line 844, in _load_as_safetensor safetensors.torch.load_model(model, model_file, strict=strict, device=map_location) # type: ignore [arg-type] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/shahao/anaconda3/envs/unid/lib/python3.11/site-packages/safetensors/torch.py", line 202, in load_model state_dict = load_file(filename, device=device) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/shahao/anaconda3/envs/unid/lib/python3.11/site-packages/safetensors/torch.py", line 313, in load_file with safe_open(filename, framework="pt", device=device) as f: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

CharlotteHao avatar Nov 27 '24 15:11 CharlotteHao

I don't know what caused this error, how can I fix it? Thanks!!!

CharlotteHao avatar Nov 27 '24 15:11 CharlotteHao