Merge pull request #24 from magic-research/dev

strip space in response
magic-research · May 7, 2024 · fd9194a · fd9194a
2 parents e2032aa + b178422
commit fd9194a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tasks/eval/model_utils.py b/tasks/eval/model_utils.py
@@ -167,7 +167,7 @@ def pllava_answer(conv: Conversation, model, processor, img_list, do_sample=True
         split_tag = conv.roles[-1]
     output_text = output_text.split(split_tag)[-1]
     ending = conv.sep if isinstance(conv.sep, str) else conv.sep[1]
-    output_text = output_text.removesuffix(ending)
+    output_text = output_text.removesuffix(ending).strip()
     conv.messages[-1][1] = output_text
     return output_text, conv