safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge
Loading pipeline components...: 0%| | 0/7 [00:00<?, ?it/s]
Traceback (most recent call last):
File "test.py", line 4, in
pipe = FluxPipeline.from_pretrained("./FLUX.1-dev", torch_dtype=torch.bfloat16)
File "/home/jtc/.local/lib/python3.8/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
File "/home/jtc/.local/lib/python3.8/site-packages/diffusers/pipelines/pipeline_utils.py", line 876, in from_pretrained
loaded_sub_model = load_sub_model(
File "/home/jtc/.local/lib/python3.8/site-packages/diffusers/pipelines/pipeline_loading_utils.py", line 700, in load_sub_model
loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs)
File "/home/jtc/.local/lib/python3.8/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
File "/home/jtc/.local/lib/python3.8/site-packages/diffusers/models/modeling_utils.py", line 774, in from_pretrained
accelerate.load_checkpoint_and_dispatch(
File "/home/jtc/.local/lib/python3.8/site-packages/accelerate/big_modeling.py", line 613, in load_checkpoint_and_dispatch
load_checkpoint_in_model(
File "/home/jtc/.local/lib/python3.8/site-packages/accelerate/utils/modeling.py", line 1758, in load_checkpoint_in_model
loaded_checkpoint = load_state_dict(checkpoint_file, device_map=device_map)
File "/home/jtc/.local/lib/python3.8/site-packages/accelerate/utils/modeling.py", line 1452, in load_state_dict
with safe_open(checkpoint_file, framework="pt") as f:
safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge