Spaces:
Runtime error
Runtime error
Merge branch 'opencode/playful-orchid'
Browse files- 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")
|