Update modeling_internlm_xcomposer2.py

modeling_internlm_xcomposer2.py

@@ -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)))
+        # 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)