naicoi commited on
Commit
e6d42b9
Β·
2 Parent(s): 3d44971bacbf2d

Merge branch 'opencode/playful-orchid'

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