load audio has problem
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user