a-r-r-o-w HF Staff commited on
Commit
9000314
·
verified ·
1 Parent(s): 0980267

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = "/raid/aryan/diffusers-wan-vace-1.3b/"
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