naicoi commited on
Commit
bacbf2d
Β·
1 Parent(s): d6c476b
Files changed (1) hide show
  1. musetalk/pipeline.py +2 -2
musetalk/pipeline.py CHANGED
@@ -109,8 +109,8 @@ def apply_musetalk_lipsync(
109
  crop_frame = crop_frame.to(vae.device, dtype=vae.dtype)
110
 
111
  with torch.no_grad():
112
- masked_latents = vae.encode(crop_frame).sample
113
- latents = vae.encode(crop_frame).sample
114
  latent_model_input = torch.cat([masked_latents, latents], dim=1)
115
  input_latent_list.append(latent_model_input)
116
  print(f"βœ“ Encoded {len(input_latent_list)} frames")
 
109
  crop_frame = crop_frame.to(vae.device, dtype=vae.dtype)
110
 
111
  with torch.no_grad():
112
+ masked_latents = vae.encode(crop_frame).latent_dist.sample()
113
+ latents = vae.encode(crop_frame).latent_dist.sample()
114
  latent_model_input = torch.cat([masked_latents, latents], dim=1)
115
  input_latent_list.append(latent_model_input)
116
  print(f"βœ“ Encoded {len(input_latent_list)} frames")