diff --git a/models/dummy_model.py b/models/dummy_model.py index 2369a2e8968b4448c4688985ccba1b03739fa85c..4517b2502170f304b0cfa720e25332c458e7a4a7 100644 --- a/models/dummy_model.py +++ b/models/dummy_model.py @@ -91,7 +91,7 @@ class llama3_8b_FewShot_vllm(ShopBenchBaseModel): passage = '' t_data = json.loads(i.strip()) if "input" in t_data: - passage = t_data['instruction'] + t_data['input'] + '\nOutput:' + str( t_data['output']) + '\n' + passage = t_data['instruction'] + str(t_data['input']) + '\nOutput:' + str( t_data['output']) + '\n' else: passage = t_data['instruction'] + str(t_data['output']) + '\n' passage = passage.replace('\\n','\n')