From 42e3d38a3f30db1a8d1929d5d51c644b60734e17 Mon Sep 17 00:00:00 2001
From: Jax922 <1322037892@qq.com>
Date: Wed, 21 May 2025 08:14:36 +0000
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8=E5=8F=98=E9=87=8F=E4=BB=A3?=
 =?UTF-8?q?=E6=9B=BF=E5=9B=BA=E5=AE=9A=E5=80=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 model/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model/model.py b/model/model.py
index 010c8e3..6235e4e 100644
--- a/model/model.py
+++ b/model/model.py
@@ -627,7 +627,7 @@ class MiniMindLM(PreTrainedModel):
         # Specific layers for v path
         self.downsample_v_specific = nn.Sequential(
             nn.Conv1d(128*8, 128, kernel_size=1, padding='same'),
-            nn.Conv1d(128, 8, kernel_size=1, padding='same')
+            nn.Conv1d(128, self.params.knowledge_length, kernel_size=1, padding='same')
         )
 
         # Specific layers for q path