Wan-AI
/

Wan2.2-I2V-A14B-Diffusers

WanImageToVideoPipeline

Model card Files Files and versions

YiYiXu commited on Jul 28

Commit

6df09af

·

verified ·

1 Parent(s): 1e0e98c

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -7,11 +7,11 @@ import numpy as np
 from diffusers import WanImageToVideoPipeline
 from diffusers.utils import export_to_video, load_image
-new_model_folder = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
 dtype = torch.bfloat16
 device = "cuda"
-pipe = WanImageToVideoPipeline.from_pretrained(new_model_folder, torch_dtype=dtype)
 pipe.to(device)

 from diffusers import WanImageToVideoPipeline
 from diffusers.utils import export_to_video, load_image
+model_id = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
 dtype = torch.bfloat16
 device = "cuda"
+pipe = WanImageToVideoPipeline.from_pretrained(model_id, torch_dtype=dtype)
 pipe.to(device)