Update vary_opt.py

fulotte · Mar 19, 2024 · cc25c58 · cc25c58
1 parent a8520b2
commit cc25c58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: