| 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.") | |