stanrom
/

internlm-xcomposer2-7b

Text Generation

internlmxcomposer2

feature-extraction

Model card Files Files and versions Community

stanrom commited on May 30, 2024

Commit

c46a3ad

·

verified ·

1 Parent(s): b0d7156

Update modeling_internlm_xcomposer2.py

Files changed (1) hide show

modeling_internlm_xcomposer2.py +2 -1

modeling_internlm_xcomposer2.py CHANGED Viewed

@@ -120,7 +120,8 @@ class InternLMXComposer2ForCausalLM(InternLM2PreTrainedModel):
     def img2emb(self, image):
         print('self.device = ', self.device)
-        img_embeds = self.vision_proj(self.vit(image.to(self.device)))
         atts_img = torch.ones(
             img_embeds.size()[:-1], dtype=torch.long).to(img_embeds.device)

     def img2emb(self, image):
         print('self.device = ', self.device)
+        # img_embeds = self.vision_proj(self.vit(image.to(self.device)))
+        img_embeds = self.vision_proj(self.vit(image))
         atts_img = torch.ones(
             img_embeds.size()[:-1], dtype=torch.long).to(img_embeds.device)