Update processing_granite_vision_embedding.py (#12)
Browse files- Update processing_granite_vision_embedding.py (5aa116f7d0fc85dc53e3d0135f0480e3366d22a3)
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 |
|