getting error in loading model flux dev model from local_path

[DevilANANDGupta]

hi @sayakpaul while using the local path i am getting the error

from diffusers import FluxInpaintPipeline
pipe = FluxInpaintPipeline.from_pretrained(
    "/mnt/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safe_tensors=True
).to("cuda")

error is pipe = FluxInpaintPipeline.from_pretrained( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/anaconda3/envs/flux/lib/python3.12/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^ File "/root/anaconda3/envs/flux/lib/python3.12/site-packages/diffusers/pipelines/pipeline_utils.py", line 896, in from_pretrained loaded_sub_model = load_sub_model( ^^^^^^^^^^^^^^^ File "/root/anaconda3/envs/flux/lib/python3.12/site-packages/diffusers/pipelines/pipeline_loading_utils.py", line 704, in load_sub_model loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/anaconda3/envs/flux/lib/python3.12/site-packages/transformers/modeling_utils.py", line 4224, in from_pretrained ) = cls._load_pretrained_model( ^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/anaconda3/envs/flux/lib/python3.12/site-packages/transformers/modeling_utils.py", line 4705, in _load_pretrained_model state_dict = load_state_dict( ^^^^^^^^^^^^^^^^ File "/root/anaconda3/envs/flux/lib/python3.12/site-packages/transformers/modeling_utils.py", line 555, in load_state_dict with safe_open(checkpoint_file, framework="pt") as f: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge

[DevilANANDGupta]

can anyone suggest me? @community @sayakpaul

[sayakpaul]

Seems like your local download isn't correct:

in load_state_dict with safe_open(checkpoint_file, framework="pt") as f: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge