fix mask bug
This commit is contained in:
@@ -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(
|
||||
|
||||
@@ -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(),
|
||||
)?)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user