fix mask bug

This commit is contained in:
jhqxxx
2025-10-11 13:14:51 +08:00
parent dcdafaf8ac
commit c8cfda7ac4
7 changed files with 9 additions and 9 deletions
+1 -1
View File
@@ -130,7 +130,7 @@ impl VoxCPMGenerate {
Ok(audio)
}
pub fn generate_simple(&mut self, target_text: String) -> Result<Tensor> {
let audio = self.generate(target_text, None, None, 2, 1000, 10, 2.0, false, 6.0)?;
let audio = self.generate(target_text, None, None, 2, 100, 10, 2.0, false, 6.0)?;
Ok(audio)
}
pub fn generate(
+2 -2
View File
@@ -266,7 +266,7 @@ impl MiniCPMModel {
Some(&prepare_causal_attention_mask(
bs,
seq_len,
position_id,
0,
input_embeds.device(),
)?)
}
@@ -294,7 +294,7 @@ impl MiniCPMModel {
Some(&prepare_causal_attention_mask(
bs,
seq_len,
position_id,
0,
input_embeds.device(),
)?)
}