Update modeling_gpt2vision.py
Browse files- modeling_gpt2vision.py +1 -1
modeling_gpt2vision.py
CHANGED
@@ -84,7 +84,7 @@ class GPT2Vision(PreTrainedModel):
|
|
84 |
}
|
85 |
print("batch",batch)
|
86 |
print("pixel_values",batch['pixel_values'].size())
|
87 |
-
print("input_ids",batch['input_ids'
|
88 |
print("attention_mask",batch['attention_mask'].size())
|
89 |
|
90 |
inputs_embeds, attention_mask, input_ids = self.preprocess_inputs(batch)
|
|
|
84 |
}
|
85 |
print("batch",batch)
|
86 |
print("pixel_values",batch['pixel_values'].size())
|
87 |
+
print("input_ids",batch['input_ids'].size())
|
88 |
print("attention_mask",batch['attention_mask'].size())
|
89 |
|
90 |
inputs_embeds, attention_mask, input_ids = self.preprocess_inputs(batch)
|