Upload model_vlm.py
Browse files- 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(
|
| 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:
|