Unverified Commit aa2dd2b5 authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

[Format] Add config lints (#892)

parent 3dbba119
...@@ -12,10 +12,10 @@ from opencompass.models import Llama2 ...@@ -12,10 +12,10 @@ from opencompass.models import Llama2
models = [ models = [
dict( dict(
abbr="llama-7b", abbr='llama-7b',
type=Llama2, type=Llama2,
path="./models/llama/7B/", path='./models/llama/7B/',
tokenizer_path="./models/llama/tokenizer.model", tokenizer_path='./models/llama/tokenizer.model',
max_out_len=100, max_out_len=100,
max_seq_len=2048, max_seq_len=2048,
batch_size=16, batch_size=16,
......
...@@ -6,14 +6,14 @@ from opencompass.models import Mixtral ...@@ -6,14 +6,14 @@ from opencompass.models import Mixtral
models = [ models = [
dict( dict(
abbr="mixtral-8x7b-32k", abbr='mixtral-8x7b-32k',
type=Mixtral, type=Mixtral,
path="./models/mixtral/mixtral-8x7b-32kseqlen", path='./models/mixtral/mixtral-8x7b-32kseqlen',
tokenizer_path="./models/mixtral/mixtral-8x7b-32kseqlen/tokenizer.model", tokenizer_path='./models/mixtral/mixtral-8x7b-32kseqlen/tokenizer.model',
max_out_len=100, max_out_len=100,
max_seq_len=2048, max_seq_len=2048,
batch_size=8, batch_size=8,
num_gpus=2, num_gpus=2,
run_cfg=dict(num_gpus=2, num_procs=1), run_cfg=dict(num_gpus=2, num_procs=1),
), ),
] ]
\ No newline at end of file
...@@ -2,10 +2,10 @@ from opencompass.models import VLLM ...@@ -2,10 +2,10 @@ from opencompass.models import VLLM
_meta_template = dict( _meta_template = dict(
begin="<s>", begin='<s>',
round=[ round=[
dict(role="HUMAN", begin='[INST]', end='[/INST]'), dict(role='HUMAN', begin='[INST]', end='[/INST]'),
dict(role="BOT", begin="", end='</s>', generate=True), dict(role='BOT', begin='', end='</s>', generate=True),
], ],
) )
......
...@@ -2,10 +2,10 @@ from opencompass.models import VLLM ...@@ -2,10 +2,10 @@ from opencompass.models import VLLM
_meta_template = dict( _meta_template = dict(
begin="<s>", begin='<s>',
round=[ round=[
dict(role="HUMAN", begin='[INST]', end='[/INST]'), dict(role='HUMAN', begin='[INST]', end='[/INST]'),
dict(role="BOT", begin="", end='</s>', generate=True), dict(role='BOT', begin='', end='</s>', generate=True),
], ],
) )
......
...@@ -2,10 +2,10 @@ from opencompass.models import VLLM ...@@ -2,10 +2,10 @@ from opencompass.models import VLLM
_meta_template = dict( _meta_template = dict(
begin="<s>", begin='<s>',
round=[ round=[
dict(role="HUMAN", begin='[INST]', end='[/INST]'), dict(role='HUMAN', begin='[INST]', end='[/INST]'),
dict(role="BOT", begin="", end='</s>', generate=True), dict(role='BOT', begin='', end='</s>', generate=True),
], ],
) )
......
...@@ -5,8 +5,8 @@ models = [ ...@@ -5,8 +5,8 @@ models = [
dict( dict(
type=HuggingFaceCausalLM, type=HuggingFaceCausalLM,
abbr='mpt-instruct-7b-hf', abbr='mpt-instruct-7b-hf',
path="mosaicml/mpt-7b-instruct", path='mosaicml/mpt-7b-instruct',
tokenizer_path="mosaicml/mpt-7b-instruct", tokenizer_path='mosaicml/mpt-7b-instruct',
tokenizer_kwargs=dict( tokenizer_kwargs=dict(
padding_side='left', padding_side='left',
truncation_side='left', truncation_side='left',
......
...@@ -11,7 +11,7 @@ models = [ ...@@ -11,7 +11,7 @@ models = [
dict( dict(
type=HuggingFaceCausalLM, type=HuggingFaceCausalLM,
abbr='nanbeige-16b-chat-hf', abbr='nanbeige-16b-chat-hf',
path="Nanbeige/Nanbeige-16B-Chat", path='Nanbeige/Nanbeige-16B-Chat',
tokenizer_path='Nanbeige/Nanbeige-16B-Chat', tokenizer_path='Nanbeige/Nanbeige-16B-Chat',
model_kwargs=dict( model_kwargs=dict(
device_map='auto', device_map='auto',
......
...@@ -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='Question:\n', end='\n'), dict(role='HUMAN', begin='Question:\n', end='\n'),
dict(role="BOT", begin="Answer:\n", end='\n', generate=True), dict(role='BOT', begin='Answer:\n', end='\n', generate=True),
], ],
) )
......
...@@ -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='Question:\n', end='\n'), dict(role='HUMAN', begin='Question:\n', end='\n'),
dict(role="BOT", begin="Answer:\n", end='\n', generate=True), dict(role='BOT', begin='Answer:\n', end='\n', generate=True),
], ],
) )
......
...@@ -4,8 +4,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -4,8 +4,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
begin='', begin='',
round=[ round=[
dict(role="HUMAN", begin='Question: ', end='\n\n'), dict(role='HUMAN', begin='Question: ', end='\n\n'),
dict(role="BOT", begin="Answer: ", end='\n\n', generate=True), dict(role='BOT', begin='Answer: ', end='\n\n', generate=True),
], ],
) )
......
...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin='<|im_start|>user\n', end='<|im_end|>\n'), dict(role='HUMAN', begin='<|im_start|>user\n', end='<|im_end|>\n'),
dict(role="BOT", begin="<|im_start|>assistant\n", end='<|im_end|>\n', generate=True), dict(role='BOT', begin='<|im_start|>assistant\n', end='<|im_end|>\n', generate=True),
], ],
) )
......
...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin='### User:\n', end='\n'), dict(role='HUMAN', begin='### User:\n', end='\n'),
dict(role="BOT", begin="### Assistant:\n", generate=True), dict(role='BOT', begin='### Assistant:\n', generate=True),
], ],
) )
......
...@@ -4,8 +4,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -4,8 +4,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
begin='Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n', begin='Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n',
round=[ round=[
dict(role="HUMAN", begin='### Instruction:\n', end='\n\n'), dict(role='HUMAN', begin='### Instruction:\n', end='\n\n'),
dict(role="BOT", begin="### Response:", end='\n\n', generate=True), dict(role='BOT', begin='### Response:', end='\n\n', generate=True),
], ],
) )
......
...@@ -2,10 +2,10 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -2,10 +2,10 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
begin="Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n", begin='Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n',
round=[ round=[
dict(role="HUMAN", begin='### Instruction:\n', end='\n\n'), dict(role='HUMAN', begin='### Instruction:\n', end='\n\n'),
dict(role="BOT", begin="### Response: ", end='\n\n', generate=True), dict(role='BOT', begin='### Response: ', end='\n\n', generate=True),
], ],
) )
......
...@@ -2,10 +2,10 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -2,10 +2,10 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
begin="Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n", begin='Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n',
round=[ round=[
dict(role="HUMAN", begin='### Instruction:\n', end='\n\n'), dict(role='HUMAN', begin='### Instruction:\n', end='\n\n'),
dict(role="BOT", begin="### Response: ", end='\n\n', generate=True), dict(role='BOT', begin='### Response: ', end='\n\n', generate=True),
], ],
) )
......
...@@ -2,10 +2,10 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -2,10 +2,10 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
begin="Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n", begin='Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n',
round=[ round=[
dict(role="HUMAN", begin='### Instruction:\n', end='\n\n'), dict(role='HUMAN', begin='### Instruction:\n', end='\n\n'),
dict(role="BOT", begin="### Response: ", end='\n\n', generate=True), dict(role='BOT', begin='### Response: ', end='\n\n', generate=True),
], ],
) )
......
...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin='GPT4 Correct User: ', end='<|end_of_turn|>'), dict(role='HUMAN', begin='GPT4 Correct User: ', end='<|end_of_turn|>'),
dict(role="BOT", begin="GPT4 Correct Assistant: ", end='<|end_of_turn|>', generate=True), dict(role='BOT', begin='GPT4 Correct Assistant: ', end='<|end_of_turn|>', generate=True),
], ],
) )
......
...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin='GPT4 Correct User: ', end='<|end_of_turn|>'), dict(role='HUMAN', begin='GPT4 Correct User: ', end='<|end_of_turn|>'),
dict(role="BOT", begin="GPT4 Correct Assistant: ", end='<|end_of_turn|>', generate=True), dict(role='BOT', begin='GPT4 Correct Assistant: ', end='<|end_of_turn|>', generate=True),
], ],
) )
......
...@@ -4,8 +4,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -4,8 +4,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
begin='<|startoftext|>', begin='<|startoftext|>',
round=[ round=[
dict(role="HUMAN", begin='Human: ', end='\n\n'), dict(role='HUMAN', begin='Human: ', end='\n\n'),
dict(role="BOT", begin="Assistant: <|endoftext|>", end='<|endoftext|>', generate=True), dict(role='BOT', begin='Assistant: <|endoftext|>', end='<|endoftext|>', generate=True),
], ],
) )
......
...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM ...@@ -3,8 +3,8 @@ from opencompass.models import HuggingFaceCausalLM
_meta_template = dict( _meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin='<_user>'), dict(role='HUMAN', begin='<_user>'),
dict(role="BOT", begin="<_bot>", end='<_end>', generate=True), dict(role='BOT', begin='<_bot>', end='<_end>', 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