load audio has problem

This commit is contained in:
jhqxxx
2025-12-31 17:39:25 +08:00
parent 3ba4de8a7d
commit 981457347a
11 changed files with 225 additions and 21 deletions
+2 -2
View File
@@ -513,7 +513,7 @@ impl VoxCPMModel {
let text_token = Tensor::cat(&[text_token, audio_start], D::Minus1)?;
let text_length = text_token.dim(0)?;
let mut audio =
load_audio_with_resample(&path, self.device.clone(), Some(self.sample_rate))?;
load_audio_with_resample(&path, &self.device, Some(self.sample_rate))?;
let patch_len = self.patch_size * self.chunk_size;
if audio.dim(1)? % patch_len != 0 {
audio = audio.pad_with_zeros(
@@ -730,7 +730,7 @@ impl VoxCPMModel {
let text_token = Tensor::from_slice(&text_token, text_token.len(), &self.device)?;
let mut audio = load_audio_with_resample(
&prompt_wav_path,
self.device.clone(),
&self.device,
Some(self.sample_rate),
)?;
let patch_len = self.patch_size * self.chunk_size;