Update processing_granite_vision_embedding.py
Browse filesContextualized 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 |
|