From dcc32ed85602f4711c0167ff99730c8f541e28b7 Mon Sep 17 00:00:00 2001 From: Mo Li <82895469+DseidLi@users.noreply.github.com> Date: Thu, 18 Jan 2024 20:54:24 +0800 Subject: [PATCH] [Fix] Update yi 200k config (#815) --- configs/models/yi/hf_yi_6b_200k.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/configs/models/yi/hf_yi_6b_200k.py b/configs/models/yi/hf_yi_6b_200k.py index e08c84ae..ecaed66a 100644 --- a/configs/models/yi/hf_yi_6b_200k.py +++ b/configs/models/yi/hf_yi_6b_200k.py @@ -1,13 +1,6 @@ from opencompass.models import HuggingFace -_meta_template = dict( - round=[ - dict(role="HUMAN", end='\n\n'), - dict(role="BOT", begin="### Response:", end='', generate=True), - ], -) - models = [ dict( type=HuggingFace, @@ -23,11 +16,9 @@ models = [ truncation_side='left', trust_remote_code=True, ), - meta_template=_meta_template, max_out_len=100, max_seq_len=2048, batch_size=8, run_cfg=dict(num_gpus=1, num_procs=1), - end_str='', ) ]