From cc25c58603a2b052aa044d92f75b8223cfeead51 Mon Sep 17 00:00:00 2001 From: WeiHaoran <50487563+Ucas-HaoranWei@users.noreply.github.com> Date: Tue, 19 Mar 2024 18:23:33 +0800 Subject: [PATCH] Update vary_opt.py --- Vary-master/vary/model/vary_opt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Vary-master/vary/model/vary_opt.py b/Vary-master/vary/model/vary_opt.py index d4314c0..646e45f 100644 --- a/Vary-master/vary/model/vary_opt.py +++ b/Vary-master/vary/model/vary_opt.py @@ -49,7 +49,7 @@ def __init__(self, config: OPTConfig): super(varyOPTModel, self).__init__(config) - self.vision_tower = build_sam_vit_b() + self.vision_tower_2 = build_sam_vit_b() self.mm_projector = nn.Linear(1024, 768) @@ -113,7 +113,7 @@ def forward( # inputs_embeds = self.wte(input_ids) - vision_tower = getattr(self, 'vision_tower', None) + vision_tower = getattr(self, 'vision_tower_2', None) if vision_tower is not None and (input_ids.shape[1] != 1 or self.training) and images is not None: