doc_to_image is list[img]

c6259cf7 · Baber · b27bc18e · c6259cf7
Commit c6259cf7 authored Sep 18, 2024 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

lm_eval/tasks/mathvista/mathvista.yaml lm_eval/tasks/mathvista/mathvista.yaml +2 -1

No files found.
--- a/lm_eval/tasks/mathvista/mathvista.yaml
+++ b/lm_eval/tasks/mathvista/mathvista.yaml
@@ -3,7 +3,8 @@ task: mathvista
 test_split: testmini
 output_type: "generate_until"
 #process_docs: !function utils.process_docs
-doc_to_image: decoded_image
+doc_to_image:
+  - decoded_image
 doc_to_text: "<image> {{query}}"
 #doc_to_choice: '{{ ["A", "B", "C", "D", "E", "F"][:choices.length] }}'
 doc_to_target: answer