fix bug by xinyanghuang7

2024-11-02 21:29:01 +08:00 · 2024-11-02 21:29:01 +08:00 · c10fdf1d8d
parent 9271e16256
commit c10fdf1d8d
1 changed files with 1 additions and 1 deletions
--- a/model/model.py
+++ b/model/model.py
@ -387,7 +387,7 @@ class Transformer(PreTrainedModel):
        # 如果此时有图像编码
        if image_encoders is not None:
            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:
                # 创建一个新的张量来存储拼接后的结果
                new_h = []