@@ -7,8 +7,8 @@ from opencompass.utils.text_postprocessors import first_capital_postprocess
py150_reader_cfg=dict(
input_columns="input",
output_column="gt",
input_columns='input',
output_column='gt',
)
py150_infer_cfg=dict(
...
...
@@ -16,8 +16,8 @@ py150_infer_cfg = dict(
type=PromptTemplate,
template=dict(
round=[
dict(role="HUMAN",prompt="I will give you a part of python code. Please write down what the next line of code is. Note that you only need to give the next line of code, and you don't need to give any other reply.\nCode:{input}\nNext line:"),
dict(role="BOT",prompt="{gt}"),
dict(role='HUMAN',prompt="I will give you a part of python code. Please write down what the next line of code is. Note that you only need to give the next line of code, and you don't need to give any other reply.\nCode:{input}\nNext line:"),