Wan: Error while deserializing header

[tin2tin]

Describe the bug

Wan doc ex.: Error while deserializing header

Reproduction

The #2 example from the Wan docs will cause an error: https://huggingface.co/docs/diffusers/main/en/api/pipelines/wan

import torch
from diffusers import WanPipeline, AutoencoderKLWan
from diffusers.utils import export_to_video

model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"

vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)

# replace this with pipe.to("cuda") if you have sufficient VRAM
pipe.enable_model_cpu_offload()

prompt = "A cat and a dog baking a cake together in a kitchen. The cat is carefully measuring flour, while the dog is stirring the batter with a wooden spoon. The kitchen is cozy, with sunlight streaming through the window."
negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards"
num_frames = 33

frames = pipe(prompt=prompt, num_frames=num_frames).frames[0]
export_to_video(frames, "wan-t2v.mp4", fps=16)

Logs

Loading checkpoint shards: 0%| | 0/5 [00:00<?, ?it/s]
Loading pipeline components...: 0%| | 0/5 [00:00<?, ?it/s]
Error: Python: Traceback (most recent call last):
 File "\python\Lib\site-packages\transformers\modeling_utils.py", line 279, in _wrapper
 return func(*args, **kwargs)
 ^^^^^^^^^^^^^^^^^^^^^
 File "\python\Lib\site-packages\transformers\modeling_utils.py", line 4389, in from_pretrained
 ) = cls._load_pretrained_model(
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^
 File "\python\Lib\site-packages\transformers\modeling_utils.py", line 4812, in _load_pretrained_model
 state_dict = load_state_dict(
 ^^^^^^^^^^^^^^^^
 File "\python\Lib\site-packages\transformers\modeling_utils.py", line 508, in load_state_dict
 with safe_open(checkpoint_file, framework="pt") as f:
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
safetensors_rust.SafetensorError: Error while deserializing header: MetadataIncompleteBuffer

System Info

Diffuses 0.33.0

Who can help?

@DN6 @a-r-r-o-w

[asomoza]

Hi, that code works fine, I tested it multiple times, from the error it seems there's a problem with the connection, like the downloaded file was corrupted?

Maybe you can try clearing the hub cache and download it again?

[tin2tin]

@asomoza True. Thank you for the heads up! Closing.