Wan-AI
/

Wan2.1-T2V-1.3B

video-generation

Model card Files Files and versions

a-r-r-o-w commited on Feb 28, 2025

Commit

8f26733

·

verified ·

1 Parent(s): f691456

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -171,7 +171,7 @@ from diffusers import AutoencoderKLWan, WanPipeline
 from diffusers.utils import export_to_video
 # Available models: Wan-AI/Wan2.1-T2V-14B, Wan-AI/Wan2.1-T2V-1.3B
-model_id = "Wan-AI/Wan2.1-T2V-14B"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 pipe.to("cuda")

 from diffusers.utils import export_to_video
 # Available models: Wan-AI/Wan2.1-T2V-14B, Wan-AI/Wan2.1-T2V-1.3B
+model_id = "Wan-AI/Wan2.1-T2V-1.3B"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 pipe.to("cuda")