update voxcpm model init

2025-10-11 14:16:49 +08:00
parent c8cfda7ac4
commit 2bac7672e1
3 changed files with 6 additions and 7 deletions
@@ -585,7 +585,7 @@ impl VoxCPMModel {
    ) -> Result<Tensor> {
        let text_token = text_token.unsqueeze(0)?;
        let text_mask = text_mask.unsqueeze(0)?;
-        let audio_feat = audio_feat.unsqueeze(0)?.to_dtype(DType::BF16)?;
+        let audio_feat = audio_feat.unsqueeze(0)?.to_dtype(self.dtype)?;
        let audio_mask = audio_mask.unsqueeze(0)?;

        let latent_pred = self.inference(