update messy

This commit is contained in:
jhqxxx
2026-04-25 19:08:48 +08:00
parent 0d408c6d06
commit 5df1f30f83
+6 -6
View File
@@ -477,16 +477,16 @@ impl VoxCPMModelRefact {
let decode_start = decode_audio.dim(D::Minus1)? - self.decode_patch_len; let decode_start = decode_audio.dim(D::Minus1)? - self.decode_patch_len;
decode_audio = decode_audio.narrow(D::Minus1, decode_start, self.decode_patch_len)?; decode_audio = decode_audio.narrow(D::Minus1, decode_start, self.decode_patch_len)?;
if i > min_len && stop_flag == 1 { if i > min_len && stop_flag == 1 {
// 最后一段去除噪音 // // 最后一段容易有噪音,暂时不返回
let decode_audio_len = decode_audio.dim(D::Minus1)? - 640; // let decode_audio_len = decode_audio.dim(D::Minus1)? - 640;
decode_audio = decode_audio.narrow(D::Minus1, 0, decode_audio_len)?; // decode_audio = decode_audio.narrow(D::Minus1, 0, decode_audio_len)?;
yield Ok(decode_audio); // yield Ok(decode_audio);
break; break;
} }
if first_flag { if first_flag {
// 去除初始噪音 // 去除初始噪音
let decode_audio_len = decode_audio.dim(D::Minus1)? - 640; let decode_audio_len = decode_audio.dim(D::Minus1)? - 1280;
decode_audio = decode_audio.narrow(D::Minus1, 640, decode_audio_len)?; decode_audio = decode_audio.narrow(D::Minus1, 1280, decode_audio_len)?;
first_flag = false; first_flag = false;
} }
yield Ok(decode_audio); yield Ok(decode_audio);