fix auto processor for text only input
Browse files- .gitignore +1 -0
- processing_aria.py +1 -0
- vision_processor.py +1 -1
.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
.ipynb_checkpoints
|
processing_aria.py
CHANGED
@@ -166,6 +166,7 @@ class AriaProcessor(ProcessorMixin):
|
|
166 |
|
167 |
else:
|
168 |
image_inputs = {}
|
|
|
169 |
|
170 |
text_inputs = self.tokenizer(
|
171 |
prompt_strings,
|
|
|
166 |
|
167 |
else:
|
168 |
image_inputs = {}
|
169 |
+
prompt_strings = text
|
170 |
|
171 |
text_inputs = self.tokenizer(
|
172 |
prompt_strings,
|
vision_processor.py
CHANGED
@@ -296,4 +296,4 @@ class AriaVisionProcessor(BaseImageProcessor):
|
|
296 |
return_tensors=return_tensors,
|
297 |
split_image=split_image,
|
298 |
split_ratio=split_ratio,
|
299 |
-
)
|
|
|
296 |
return_tensors=return_tensors,
|
297 |
split_image=split_image,
|
298 |
split_ratio=split_ratio,
|
299 |
+
)
|