admin/VoxCPM
1
0
Fork 0
mirror of https://github.com/OpenBMB/VoxCPM synced 2025-12-12 03:48:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

update README

Browse Source
This commit is contained in:
周逸轩
2025-09-18 14:53:37 +08:00
parent 5257ec3dc5
commit 1a46c5d1ad
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/voxcpm/model/voxcpm.py
View File

@@ -283,10 +283,11 @@ 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
return decode_audio return decode_audio
@torch.inference_mode() @torch.inference_mode()
def build_prompt_cache( def build_prompt_cache(