From c1a069b5325d9e9f69056fc618d2984d3909a563 Mon Sep 17 00:00:00 2001
From: gongjy <2474590974@qq.com>
Date: Sun, 23 Feb 2025 18:59:35 +0800
Subject: [PATCH] fix a bug

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

diff --git a/model/model_vlm.py b/model/model_vlm.py
index 9a5f8f3..f9b10d7 100644
--- a/model/model_vlm.py
+++ b/model/model_vlm.py
@@ -74,7 +74,7 @@ class MiniMindVLM(MiniMindLM):
         if vision_tensors is not None and image_indices:
             vision_proj = self.vision_proj(vision_tensors)
             if len(vision_proj.shape) == 3:
-                vision_proj = vision_proj.unsqueeze(1)
+                vision_proj = vision_proj.unsqueeze(0)
             new_h = []
             for i in range(h.size(0)):
                 if i in image_indices: