Merge pull request #5 from rhymes-ai/fix_text_input

fix auto processor for text only input
rhymes-ai · Oct 1, 2024 · efd4c54 · efd4c54
2 parents 4d70b52 + 72a42e8
commit efd4c54
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/aria/model/processing_aria.py b/aria/model/processing_aria.py
@@ -166,6 +166,7 @@ def __call__(
 
         else:
             image_inputs = {}
+            prompt_strings = text
 
         text_inputs = self.tokenizer(
             prompt_strings,