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."