[bugfix]VoxCPM2 audio encoder adapt other than CUDA (#4374)

Signed-off-by: tanhaoan333 <tanhaoan@huawei.com>
This commit is contained in:
tanhaoan333 authored and GitHub committed 2026-06-12 08:54:29 +00:00
1 parent f44f3ef9f7
commit aa59dc50c7
1 file changed
+2 -1
@@ -283,7 +283,8 @@ def _encode_raw_audio(
pad = (padding_size, 0) if padding_mode == "left" else (0, padding_size)
audio = torch.nn.functional.pad(audio, pad)
feat = tts.audio_vae.encode(audio.to(tts.device), encode_sr).cpu()
vae_device = next(tts.audio_vae.parameters()).device
feat = tts.audio_vae.encode(audio.to(vae_device), encode_sr).cpu()
return feat.view(tts.audio_vae.latent_dim, -1, tts.patch_size).permute(1, 2, 0)