Update modeling_gpt2vision.py
Browse files- modeling_gpt2vision.py +1 -1
modeling_gpt2vision.py
CHANGED
|
@@ -86,7 +86,7 @@ class GPT2Vision(PreTrainedModel):
|
|
| 86 |
attention_mask = batch['attention_mask'].to(self.device)
|
| 87 |
|
| 88 |
img_embs = self.vision_encoder(images, device=self.device)
|
| 89 |
-
print("img_embs",img_embs
|
| 90 |
img_embs = self.mlp(img_embs)
|
| 91 |
|
| 92 |
tok_embs = self.language_model.get_input_embeddings()(input_ids)
|
|
|
|
| 86 |
attention_mask = batch['attention_mask'].to(self.device)
|
| 87 |
|
| 88 |
img_embs = self.vision_encoder(images, device=self.device)
|
| 89 |
+
print("img_embs",img_embs)
|
| 90 |
img_embs = self.mlp(img_embs)
|
| 91 |
|
| 92 |
tok_embs = self.language_model.get_input_embeddings()(input_ids)
|