Update README.md
Browse files
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-
|
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)
|