Adirazgold commited on
Commit
5aa116f
·
verified ·
1 Parent(s): 4b8a24b

Update processing_granite_vision_embedding.py

Browse files

Contextualized query, removing query length limitations

processing_granite_vision_embedding.py CHANGED
@@ -320,10 +320,7 @@ class GraniteVisionEmbProcessor(LlavaNextProcessor):
320
 
321
  processed = []
322
  for q in queries:
323
- q = self.query_start + self.query_prefix + q
324
- # truncate before it eats actual query content
325
- if len(q) + len(suffix) > max_length:
326
- q = q[: max_length - len(suffix) - 1]
327
  q += suffix + "\n"
328
  processed.append(q)
329
 
 
320
 
321
  processed = []
322
  for q in queries:
323
+ q = self.query_start + self.query_prefix + q + ' ' + q
 
 
 
324
  q += suffix + "\n"
325
  processed.append(q)
326