github
/
minimind-v
mirror of https://github.com/jingyaogong/minimind-v.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix a bug

pull/45/head
gongjy 2025-02-23 18:59:35 +08:00
parent 694ce87e39
commit c1a069b532
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model/model_vlm.py
View File

@ -74,7 +74,7 @@ class MiniMindVLM(MiniMindLM):
if vision_tensors is not None and image_indices: if vision_tensors is not None and image_indices:
vision_proj = self.vision_proj(vision_tensors) vision_proj = self.vision_proj(vision_tensors)
if len(vision_proj.shape) == 3: if len(vision_proj.shape) == 3:
vision_proj = vision_proj.unsqueeze(1) vision_proj = vision_proj.unsqueeze(0)
new_h = [] new_h = []
for i in range(h.size(0)): for i in range(h.size(0)):
if i in image_indices: if i in image_indices: