update README

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

View File

@@ -283,10 +283,11 @@ class VoxCPMModel(nn.Module):
else:
break
else:
break
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
return decode_audio
return decode_audio
@torch.inference_mode()
def build_prompt_cache(