lmms_eval/tasks/stvqa/stvqa.yaml (19 lines of code) (raw):

dataset_path: lmms-lab/ST-VQA task: "stvqa" test_split: test output_type: generate_until doc_to_visual: !function utils.stvqa_doc_to_visual doc_to_text: !function utils.stvqa_doc_to_text doc_to_target: "answers" generation_kwargs: max_new_tokens: 32 temperature: 0 do_sample: False process_results: !function utils.stvqa_process_results metric_list: - metric: submission aggregation: !function utils.stvqa_aggregate_submissions model_specific_prompt_kwargs: default: pre_prompt: "" post_prompt: "\nAnswer the question using a single word or phrase."