a-r-r-o-w HF Staff commited on
Commit
6b78f01
·
verified ·
1 Parent(s): 4c175c4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -158,7 +158,7 @@ import torch
158
  from diffusers import AutoencoderKLWan, WanImageToVideoPipeline
159
  from diffusers.utils import export_to_video, load_image
160
 
161
- # Available models: Wan-AI/Wan2.1-I2V-14B-480P-Diffusers, Wan-AI/Wan2.1-I2V-1.3B-720P-Diffusers
162
  model_id = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
163
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
164
  pipe = WanImageToVideoPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 
158
  from diffusers import AutoencoderKLWan, WanImageToVideoPipeline
159
  from diffusers.utils import export_to_video, load_image
160
 
161
+ # Available models: Wan-AI/Wan2.1-I2V-14B-480P-Diffusers, Wan-AI/Wan2.1-I2V-14B-720P-Diffusers
162
  model_id = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
163
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
164
  pipe = WanImageToVideoPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)