推理时llava.eval.model_vqa中的metadata是否忘记改为category？

[wxxu-cnic]

ans_file.write(json.dumps({"question_id": idx,
"prompt": cur_prompt,
"text": outputs,
"answer_id": ans_id,
"model_id": model_name,
"metadata": {}}) + "\n")
然而输出格式要求为
{ 'question_id': 'TABMWP_8', # item_id
'prompt': 'Problem: \nHannah baked cookies each day ...', # input_prompt
'text': 'Find the numbers in the table.\n\nSaturday: ...', # model_output
'answer_id': 'jELcxSPcXHBj3xvHfm5r8T', # answer_id
'model_id': 'table-llava-7b', # model_id
'category': 'TABMWP_for_TQA'
} # item category

[wxxu-cnic]

ans_file.write(json.dumps({"question_id": idx, "prompt": cur_prompt, "text": outputs, "answer_id": ans_id, "model_id": model_name, "metadata": {}}) + "\n") 然而输出格式要求为 { 'question_id': 'TABMWP_8', # item_id 'prompt': 'Problem: \nHannah baked cookies each day ...', # input_prompt 'text': 'Find the numbers in the table.\n\nSaturday: ...', # model_output 'answer_id': 'jELcxSPcXHBj3xvHfm5r8T', # answer_id 'model_id': 'table-llava-7b', # model_id 'category': 'TABMWP_for_TQA' } # item category

我按照给定脚本运行无法得到category

[SpursGoZmy]

感谢同学指正，可能是我上传的llava代码版本太新了，和老版本有改动，然后sh脚本对应的是老脚本，我后面修正一下。