fix a bug

2025-02-23 18:59:35 +08:00 · 2025-02-23 18:59:35 +08:00 · c1a069b532
parent 694ce87e39
commit c1a069b532
1 changed files with 1 additions and 1 deletions
--- 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: