nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-18 07:15:12 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

remove newlines from captions

Browse source
This commit is contained in:
Tom Young 2024-06-15 10:40:18 +00:00
parent 34252d40ef
commit 094f0770d7
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
tools/caption/pllava_dir/caption_pllava.py
View file

@ -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