fix mask bug

2025-10-11 13:14:51 +08:00
parent dcdafaf8ac
commit c8cfda7ac4
7 changed files with 9 additions and 9 deletions
@@ -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(),
                )?)
            }