naicoi commited on
Commit
5d6431c
Β·
2 Parent(s): fd03f564d32dce

Merge branch 'opencode/playful-orchid'

Browse files
Files changed (1) hide show
  1. musetalk/pipeline.py +1 -1
musetalk/pipeline.py CHANGED
@@ -148,7 +148,7 @@ def apply_musetalk_lipsync(
148
  image = vae.decode(pred_latents).sample
149
 
150
  image = (image / 2 + 0.5).clamp(0, 1)
151
- image = image.cpu().permute(0, 2, 3, 1).numpy()
152
  image = (image * 255).round().astype("uint8")
153
  image = image[..., ::-1]
154
 
 
148
  image = vae.decode(pred_latents).sample
149
 
150
  image = (image / 2 + 0.5).clamp(0, 1)
151
+ image = image.cpu().permute(0, 2, 3, 1).detach().numpy()
152
  image = (image * 255).round().astype("uint8")
153
  image = image[..., ::-1]
154