Skip to content

GitLab

"...composable_kernel-1.git" did not exist on "37f4e2b6d8dc45f151f7de81b4d0e35c2804b77f"
Unverified Commit 2b7d376e authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

[Fix] Fix chatglm2 config (#893)

parent 9119e2ac
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
configs/models/chatglm/hf_chatglm2_6b.py
View file @ 2b7d376e
from opencompass.models import HuggingFace from opencompass.models import HuggingFace
_meta_template = dict(
round=[
dict(role='HUMAN', begin='[Round 1]\n\n问：', end='\n\n'),
dict(role='BOT', begin='答：', end='\n\n', generate=True),
],
)
models = [ models = [
dict( dict(
...@@ -16,6 +22,7 @@ models = [ ...@@ -16,6 +22,7 @@ models = [
truncation_side='left', truncation_side='left',
trust_remote_code=True, trust_remote_code=True,
), ),
meta_template=_meta_template,
max_out_len=100, max_out_len=100,
max_seq_len=4096, max_seq_len=4096,
batch_size=8, 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