423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

add llava-next/llava-next-video/video-llava

Browse Source
This commit is contained in:
BUAADreamer
2024-09-28 00:57:03 +08:00
parent 900631755b
commit 6642cd501d
5 changed files with 134 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/llamafactory/data/template.py
View File

@@ -686,16 +686,6 @@ _register_template(
)
_register_template(
name="idefics2",
format_user=StringFormatter(slots=["User:{{content}}<end_of_utterance>\nAssistant:"]),
format_separator=EmptyFormatter(slots=["\n"]),
stop_words=["<end_of_utterance>"],
replace_eos=True,
mm_plugin=get_mm_plugin(name="idefics2", image_token="<image>"),
)
_register_template(
name="intern",
format_user=StringFormatter(slots=["<|User|>:{{content}}\n<|Bot|>:"]),