Update modeling_internlm_xcomposer2.py
Browse files
modeling_internlm_xcomposer2.py
CHANGED
|
@@ -119,6 +119,7 @@ class InternLMXComposer2ForCausalLM(InternLM2PreTrainedModel):
|
|
| 119 |
return img_embeds
|
| 120 |
|
| 121 |
def img2emb(self, image):
|
|
|
|
| 122 |
img_embeds = self.vision_proj(self.vit(image.to(self.device)))
|
| 123 |
atts_img = torch.ones(
|
| 124 |
img_embeds.size()[:-1], dtype=torch.long).to(img_embeds.device)
|
|
|
|
| 119 |
return img_embeds
|
| 120 |
|
| 121 |
def img2emb(self, image):
|
| 122 |
+
print('self.device = ', self.device)
|
| 123 |
img_embeds = self.vision_proj(self.vit(image.to(self.device)))
|
| 124 |
atts_img = torch.ones(
|
| 125 |
img_embeds.size()[:-1], dtype=torch.long).to(img_embeds.device)
|