Update modeling_gpt2vision.py
Browse files- modeling_gpt2vision.py +2 -0
modeling_gpt2vision.py
CHANGED
|
@@ -72,7 +72,9 @@ class GPT2Vision(PreTrainedModel):
|
|
| 72 |
}
|
| 73 |
|
| 74 |
with torch.no_grad():
|
|
|
|
| 75 |
inputs_embeds = self.input_embeds(prompt, image_embeds, tokenizer)
|
|
|
|
| 76 |
output_ids = self.text_model.generate(
|
| 77 |
inputs_embeds=inputs_embeds, **generate_config
|
| 78 |
)
|
|
|
|
| 72 |
}
|
| 73 |
|
| 74 |
with torch.no_grad():
|
| 75 |
+
|
| 76 |
inputs_embeds = self.input_embeds(prompt, image_embeds, tokenizer)
|
| 77 |
+
print("inputs_embeds",inputs_embeds.size())
|
| 78 |
output_ids = self.text_model.generate(
|
| 79 |
inputs_embeds=inputs_embeds, **generate_config
|
| 80 |
)
|