[Fix] Fix moss template config (#897)

open-compass · Feb 21, 2024 · 6d04dec · 6d04dec
1 parent 2b7d376
commit 6d04dec
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/configs/models/moss/hf_moss_moon_003_sft.py b/configs/models/moss/hf_moss_moon_003_sft.py
@@ -1,5 +1,11 @@
 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 @@
             truncation_side='left',
             trust_remote_code=True,
         ),
+        meta_template=_meta_template,
         max_out_len=100,
         max_seq_len=2048,
         batch_size=8,