fix mask bug

2025-10-11 13:14:51 +08:00
parent dcdafaf8ac
commit c8cfda7ac4
7 changed files with 9 additions and 9 deletions
@@ -233,7 +233,7 @@ impl MiniCPMModel {
                Some(&prepare_causal_attention_mask(
                    bs,
                    seq_len,
-                    position_id,
+                    0,
                    input_ids.device(),
                )?)
            }
@@ -267,7 +267,7 @@ impl MiniCPMModel {
                Some(&prepare_causal_attention_mask(
                    bs,
                    seq_len,
-                    position_id,
+                    0,
                    input_ids.device(),
                )?)
            }