Update modeling_gpt2vision.py
Browse files- modeling_gpt2vision.py +1 -1
modeling_gpt2vision.py
CHANGED
|
@@ -100,7 +100,7 @@ class GPT2Vision(PreTrainedModel):
|
|
| 100 |
if isinstance(text, str):
|
| 101 |
text = [text]
|
| 102 |
|
| 103 |
-
input_texts = [f"{IMAGE_TOKEN}{
|
| 104 |
text_inputs = self.tokenizer(
|
| 105 |
input_texts,
|
| 106 |
padding='max_length',
|
|
|
|
| 100 |
if isinstance(text, str):
|
| 101 |
text = [text]
|
| 102 |
|
| 103 |
+
input_texts = [f"{IMAGE_TOKEN}{t}" for t in text]
|
| 104 |
text_inputs = self.tokenizer(
|
| 105 |
input_texts,
|
| 106 |
padding='max_length',
|