Hugging Face's logo

jingyaogong
/
MiniMind2-V

PyTorch
minimind-v
custom_code
Model card Files
xet
Community
4
jingyaogong commited on
Commit
1c1c92f
·
verified ·
1 Parent(s): afe6a87

Upload model_vlm.py

Browse files
Files changed (1) hide show
  1. model_vlm.py +1 -1
model_vlm.py CHANGED
@@ -74,7 +74,7 @@ class MiniMindVLM(MiniMindLM):
74
  if vision_tensors is not None and image_indices:
75
  vision_proj = self.vision_proj(vision_tensors)
76
  if len(vision_proj.shape) == 3:
77
- vision_proj = vision_proj.unsqueeze(1)
78
  new_h = []
79
  for i in range(h.size(0)):
80
  if i in image_indices:
 
74
  if vision_tensors is not None and image_indices:
75
  vision_proj = self.vision_proj(vision_tensors)
76
  if len(vision_proj.shape) == 3:
77
+ vision_proj = vision_proj.unsqueeze(0)
78
  new_h = []
79
  for i in range(h.size(0)):
80
  if i in image_indices: