internlm
/

internlm-xcomposer2d5-7b

Visual Question Answering

feature-extraction

Model card Files Files and versions Community

Update build_mlp.py

#20

by cardcounter - opened Aug 19

base: refs/heads/main

←

from: refs/pr/20

Discussion Files changed

Files changed (1) hide show

build_mlp.py +1 -1

build_mlp.py CHANGED Viewed

@@ -59,7 +59,7 @@ class CLIPVisionTower(nn.Module):
     def load_model(self):
         self.vision_tower = CLIPVisionModel.from_pretrained(self.vision_tower_name)
         self.vision_tower.requires_grad_(False)
         self.is_loaded = True
     def resize_pos(self):

     def load_model(self):
         self.vision_tower = CLIPVisionModel.from_pretrained(self.vision_tower_name)
         self.vision_tower.requires_grad_(False)
+        self.vision_tower.gradient_checkpointing_enable({"use_reentrant": "True"})
         self.is_loaded = True
     def resize_pos(self):