Spaces:

naicoi
/

lipsync-docker

Runtime error

naicoi commited on Jan 29

Commit

002abe4

1 Parent(s): 0c6c615

update

Files changed (1) hide show

musetalk/pipeline.py CHANGED Viewed

@@ -106,7 +106,7 @@ def apply_musetalk_lipsync(
         crop_frame = crop_frame.astype(np.float32) / 255.0
         crop_frame = torch.from_numpy(crop_frame).permute(2, 0, 1).unsqueeze(0)
-        crop_frame = crop_frame.to(vae.device)
         with torch.no_grad():
             masked_latents = vae.encode(crop_frame)[0]

         crop_frame = crop_frame.astype(np.float32) / 255.0
         crop_frame = torch.from_numpy(crop_frame).permute(2, 0, 1).unsqueeze(0)
+        crop_frame = crop_frame.to(vae.device, dtype=vae.dtype)
         with torch.no_grad():
             masked_latents = vae.encode(crop_frame)[0]