From 094f0770d78656a8339ea19eb878cb266bb6196e Mon Sep 17 00:00:00 2001 From: Tom Young Date: Sat, 15 Jun 2024 10:40:18 +0000 Subject: [PATCH] remove newlines from captions --- tools/caption/pllava_dir/caption_pllava.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/caption/pllava_dir/caption_pllava.py b/tools/caption/pllava_dir/caption_pllava.py index 9e02a0a..6f65152 100644 --- a/tools/caption/pllava_dir/caption_pllava.py +++ b/tools/caption/pllava_dir/caption_pllava.py @@ -60,6 +60,7 @@ def pllava_answer(conv: Conversation, model, processor, video_list, do_sample=Tr output_texts[i] = output_texts[i].split(split_tag)[-1] ending = conv.sep if isinstance(conv.sep, str) else conv.sep[1] output_texts[i] = output_texts[i].removesuffix(ending).strip() + output_texts[i] = output_texts[i].replace('\n', ' ') conv.messages[-1][1] = output_texts[i] return output_texts, conv