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',
|