Skip to content

GitLab

Unverified Commit 17735f0c authored by liushz's avatar liushz Committed by GitHub
Browse files

Fix Llama-3 meta template (#1079) 


Co-authored-by: default avatarliuhongwei <liuhongwei@pjlab.org.cn>
parent 81d0e4d7
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
configs/models/hf_llama/hf_llama3_70b_instruct.py
View file @ 17735f0c
...@@ -2,8 +2,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -2,8 +2,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin="<|begin_of_text|>user<|end_header_id|>\n\n", end="<|eot_id|>"), dict(role="HUMAN", begin="<|start_header_id|>user<|end_header_id|>\n\n", end="<|eot_id|>"),
dict(role="BOT", begin="<|begin_of_text|>assistant<|end_header_id|>\n\n", end="<|eot_id|>", generate=True), dict(role="BOT", begin="<|start_header_id|>assistant<|end_header_id|>\n\n", end="<|eot_id|>", generate=True),
], ],
) )
......
configs/models/hf_llama/hf_llama3_8b_instruct.py
View file @ 17735f0c
...@@ -2,8 +2,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -2,8 +2,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin="<|begin_of_text|>user<|end_header_id|>\n\n", end="<|eot_id|>"), dict(role="HUMAN", begin="<|start_header_id|>user<|end_header_id|>\n\n", end="<|eot_id|>"),
dict(role="BOT", begin="<|begin_of_text|>assistant<|end_header_id|>\n\n", end="<|eot_id|>", generate=True), dict(role="BOT", begin="<|start_header_id|>assistant<|end_header_id|>\n\n", end="<|eot_id|>", generate=True),
], ],
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment