Skip to content

GitLab

Unverified Commit 6d04deca authored by Xu Song's avatar Xu Song Committed by GitHub
Browse files

[Fix] Fix moss template config (#897)

parent 2b7d376e
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
configs/models/moss/hf_moss_moon_003_sft.py
View file @ 6d04deca
from opencompass.models import HuggingFaceCausalLM
_meta_template = dict(
round=[
dict(role='HUMAN', begin='<|Human|>: ', end='<eoh>\n'),
dict(role='BOT', begin='<|MOSS|>: ', end='<eom>\n', generate=True),
],
)
models = [
dict(
......@@ -12,6 +18,7 @@ models = [
truncation_side='left',
trust_remote_code=True,
),
meta_template=_meta_template,
max_out_len=100,
max_seq_len=2048,
batch_size=8,
......
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