Update README.md
Browse files
README.md
CHANGED
@@ -556,7 +556,7 @@ def prepare_video_and_mask(first_img: PIL.Image.Image, last_img: PIL.Image.Image
|
|
556 |
return frames, mask
|
557 |
|
558 |
|
559 |
-
model_id = "/
|
560 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
561 |
pipe = WanVACEPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
|
562 |
flow_shift = 5.0 # 5.0 for 720P, 3.0 for 480P
|
|
|
556 |
return frames, mask
|
557 |
|
558 |
|
559 |
+
model_id = "Wan-AI/Wan2.1-VACE-1.3B-diffusers"
|
560 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
561 |
pipe = WanVACEPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
|
562 |
flow_shift = 5.0 # 5.0 for 720P, 3.0 for 480P
|