Wan-AI
/

Wan2.2-TI2V-5B-Diffusers

Model card Files Files and versions

YiYiXu HF Staff commited on Jul 28, 2025

Commit

bfbd008

·

verified ·

1 Parent(s): 6cf84ea

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -10,8 +10,9 @@ from diffusers.utils import export_to_video, load_image
 dtype = torch.bfloat16
 device = "cuda"
-vae = AutoencoderKLWan.from_pretrained("Wan-AI/Wan2.2-TI2V-5B-Diffusers", subfolder="vae", torch_dtype=torch.float32)
-pipe = WanPipeline.from_pretrained(new_model_folder, vae=vae, torch_dtype=dtype)
 pipe.to(device)
 height = 704

 dtype = torch.bfloat16
 device = "cuda"
+model_id = "Wan-AI/Wan2.2-TI2V-5B-Diffusers"
+vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
+pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=dtype)
 pipe.to(device)
 height = 704