aria-dev commited on
Commit
28b3978
1 Parent(s): 9710af0

fix auto processor for text only input

Browse files
Files changed (3) hide show
  1. .gitignore +1 -0
  2. processing_aria.py +1 -0
  3. 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
+ )