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