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

fix bug by xinyanghuang7

pull/20/head
gongjy 2024-11-02 21:29:01 +08:00
parent 9271e16256
commit c10fdf1d8d
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.py
View File

@ -387,7 +387,7 @@ class Transformer(PreTrainedModel):
# 如果此时有图像编码 # 如果此时有图像编码
if image_encoders is not None: if image_encoders is not None:
vision_proj = self.vision_proj(image_encoders) vision_proj = self.vision_proj(image_encoders)
vision_proj = vision_proj.unsqueeze(0) if len(vision_proj.shape) == 3 else vision_proj vision_proj = vision_proj.unsqueeze(1) if len(vision_proj.shape) == 3 else vision_proj
if image_indices is not None: if image_indices is not None:
# 创建一个新的张量来存储拼接后的结果 # 创建一个新的张量来存储拼接后的结果
new_h = [] new_h = []