update README

This commit is contained in:
周逸轩
2025-09-18 14:53:37 +08:00
parent 5257ec3dc5
commit 1a46c5d1ad

View File

@@ -283,6 +283,7 @@ class VoxCPMModel(nn.Module):
else: else:
break break
else: else:
break
decode_audio = self.audio_vae.decode(latent_pred.to(torch.float32)).squeeze(1).cpu() decode_audio = self.audio_vae.decode(latent_pred.to(torch.float32)).squeeze(1).cpu()
decode_audio = decode_audio[..., 640:-640] # trick: trim the start and end of the audio decode_audio = decode_audio[..., 640:-640] # trick: trim the start and end of the audio