import torch
from diffusers import AutoencoderKLWan, WanPipeline

model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
print("Pre-downloading model weights on CPU...")
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
_ = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
print("Model weights pre-downloaded.")