crfm-helm 0.4.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of crfm-helm might be problematic. Click here for more details.

Files changed (499) hide show
  1. {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/METADATA +138 -31
  2. crfm_helm-0.5.1.dist-info/RECORD +654 -0
  3. {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/WHEEL +1 -1
  4. helm/benchmark/adaptation/adapter_spec.py +31 -3
  5. helm/benchmark/adaptation/adapters/adapter.py +2 -2
  6. helm/benchmark/adaptation/adapters/adapter_factory.py +24 -27
  7. helm/benchmark/adaptation/adapters/generation_adapter.py +1 -0
  8. helm/benchmark/adaptation/adapters/in_context_learning_adapter.py +20 -4
  9. helm/benchmark/adaptation/adapters/language_modeling_adapter.py +2 -3
  10. helm/benchmark/adaptation/adapters/multimodal/in_context_learning_multimodal_adapter.py +1 -0
  11. helm/benchmark/adaptation/adapters/multimodal/multimodal_prompt.py +7 -0
  12. helm/benchmark/adaptation/adapters/multimodal/multiple_choice_joint_multimodal_adapter.py +104 -0
  13. helm/benchmark/adaptation/adapters/multimodal/test_in_context_learning_multimodal_adapter.py +2 -1
  14. helm/benchmark/adaptation/adapters/multimodal/test_multimodal_prompt.py +2 -0
  15. helm/benchmark/adaptation/adapters/test_adapter.py +2 -1
  16. helm/benchmark/adaptation/adapters/test_generation_adapter.py +32 -8
  17. helm/benchmark/adaptation/adapters/test_language_modeling_adapter.py +7 -19
  18. helm/benchmark/adaptation/adapters/test_multiple_choice_joint_adapter.py +60 -6
  19. helm/benchmark/adaptation/common_adapter_specs.py +376 -0
  20. helm/benchmark/adaptation/request_state.py +6 -1
  21. helm/benchmark/adaptation/scenario_state.py +6 -2
  22. helm/benchmark/annotation/annotator.py +43 -0
  23. helm/benchmark/annotation/annotator_factory.py +61 -0
  24. helm/benchmark/annotation/image2structure/image_compiler_annotator.py +88 -0
  25. helm/benchmark/annotation/image2structure/latex_compiler_annotator.py +59 -0
  26. helm/benchmark/annotation/image2structure/lilypond_compiler_annotator.py +84 -0
  27. helm/benchmark/annotation/image2structure/webpage_compiler_annotator.py +132 -0
  28. helm/benchmark/annotation/test_annotator_factory.py +26 -0
  29. helm/benchmark/annotation/test_dummy_annotator.py +44 -0
  30. helm/benchmark/annotation_executor.py +124 -0
  31. helm/benchmark/augmentations/data_augmenter.py +0 -2
  32. helm/benchmark/augmentations/gender_perturbation.py +1 -1
  33. helm/benchmark/augmentations/perturbation.py +25 -3
  34. helm/benchmark/augmentations/perturbation_description.py +1 -1
  35. helm/benchmark/augmentations/suffix_perturbation.py +29 -0
  36. helm/benchmark/augmentations/test_perturbation.py +41 -7
  37. helm/benchmark/augmentations/translate_perturbation.py +30 -0
  38. helm/benchmark/config_registry.py +7 -1
  39. helm/benchmark/executor.py +46 -16
  40. helm/benchmark/huggingface_registration.py +20 -7
  41. helm/benchmark/metrics/basic_metrics.py +169 -664
  42. helm/benchmark/metrics/bbq_metrics.py +3 -4
  43. helm/benchmark/metrics/bias_metrics.py +6 -6
  44. helm/benchmark/metrics/classification_metrics.py +11 -8
  45. helm/benchmark/metrics/cleva_accuracy_metrics.py +8 -5
  46. helm/benchmark/metrics/cleva_harms_metrics.py +2 -2
  47. helm/benchmark/metrics/code_metrics_helper.py +0 -2
  48. helm/benchmark/metrics/common_metric_specs.py +167 -0
  49. helm/benchmark/metrics/decodingtrust_fairness_metrics.py +72 -0
  50. helm/benchmark/metrics/decodingtrust_ood_knowledge_metrics.py +66 -0
  51. helm/benchmark/metrics/decodingtrust_privacy_metrics.py +101 -0
  52. helm/benchmark/metrics/decodingtrust_stereotype_bias_metrics.py +202 -0
  53. helm/benchmark/metrics/disinformation_metrics.py +4 -110
  54. helm/benchmark/metrics/dry_run_metrics.py +2 -2
  55. helm/benchmark/metrics/efficiency_metrics.py +213 -0
  56. helm/benchmark/metrics/evaluate_instances_metric.py +59 -0
  57. helm/benchmark/metrics/evaluate_reference_metrics.py +392 -0
  58. helm/benchmark/metrics/image_generation/aesthetics_metrics.py +54 -0
  59. helm/benchmark/metrics/image_generation/aesthetics_scorer.py +66 -0
  60. helm/benchmark/metrics/image_generation/clip_score_metrics.py +73 -0
  61. helm/benchmark/metrics/image_generation/denoised_runtime_metric.py +42 -0
  62. helm/benchmark/metrics/image_generation/detection_metrics.py +57 -0
  63. helm/benchmark/metrics/image_generation/detectors/base_detector.py +8 -0
  64. helm/benchmark/metrics/image_generation/detectors/vitdet.py +178 -0
  65. helm/benchmark/metrics/image_generation/efficiency_metrics.py +41 -0
  66. helm/benchmark/metrics/image_generation/fidelity_metrics.py +168 -0
  67. helm/benchmark/metrics/image_generation/fractal_dimension/__init__.py +0 -0
  68. helm/benchmark/metrics/image_generation/fractal_dimension/fractal_dimension_util.py +63 -0
  69. helm/benchmark/metrics/image_generation/fractal_dimension/test_fractal_dimension_util.py +33 -0
  70. helm/benchmark/metrics/image_generation/fractal_dimension_metric.py +50 -0
  71. helm/benchmark/metrics/image_generation/gender_metrics.py +58 -0
  72. helm/benchmark/metrics/image_generation/image_critique_metrics.py +284 -0
  73. helm/benchmark/metrics/image_generation/lpips_metrics.py +82 -0
  74. helm/benchmark/metrics/image_generation/multi_scale_ssim_metrics.py +82 -0
  75. helm/benchmark/metrics/image_generation/nsfw_detector.py +96 -0
  76. helm/benchmark/metrics/image_generation/nsfw_metrics.py +103 -0
  77. helm/benchmark/metrics/image_generation/nudity_metrics.py +38 -0
  78. helm/benchmark/metrics/image_generation/photorealism_critique_metrics.py +153 -0
  79. helm/benchmark/metrics/image_generation/psnr_metrics.py +78 -0
  80. helm/benchmark/metrics/image_generation/q16/__init__.py +0 -0
  81. helm/benchmark/metrics/image_generation/q16/q16_toxicity_detector.py +90 -0
  82. helm/benchmark/metrics/image_generation/q16/test_q16.py +18 -0
  83. helm/benchmark/metrics/image_generation/q16_toxicity_metrics.py +48 -0
  84. helm/benchmark/metrics/image_generation/skin_tone_metrics.py +164 -0
  85. helm/benchmark/metrics/image_generation/uiqi_metrics.py +92 -0
  86. helm/benchmark/metrics/image_generation/watermark/__init__.py +0 -0
  87. helm/benchmark/metrics/image_generation/watermark/test_watermark_detector.py +16 -0
  88. helm/benchmark/metrics/image_generation/watermark/watermark_detector.py +87 -0
  89. helm/benchmark/metrics/image_generation/watermark_metrics.py +48 -0
  90. helm/benchmark/metrics/instruction_following_critique_metrics.py +3 -1
  91. helm/benchmark/metrics/language_modeling_metrics.py +99 -0
  92. helm/benchmark/metrics/machine_translation_metrics.py +89 -0
  93. helm/benchmark/metrics/metric.py +93 -172
  94. helm/benchmark/metrics/metric_name.py +0 -1
  95. helm/benchmark/metrics/metric_service.py +16 -0
  96. helm/benchmark/metrics/paraphrase_generation_metrics.py +3 -4
  97. helm/benchmark/metrics/ranking_metrics.py +2 -2
  98. helm/benchmark/metrics/reference_metric.py +148 -0
  99. helm/benchmark/metrics/summac/model_summac.py +0 -2
  100. helm/benchmark/metrics/summarization_metrics.py +2 -2
  101. helm/benchmark/metrics/test_classification_metrics.py +8 -5
  102. helm/benchmark/metrics/test_disinformation_metrics.py +78 -0
  103. helm/benchmark/metrics/{test_basic_metrics.py → test_evaluate_reference_metrics.py} +5 -1
  104. helm/benchmark/metrics/test_metric.py +2 -2
  105. helm/benchmark/metrics/tokens/gooseai_token_cost_estimator.py +10 -2
  106. helm/benchmark/metrics/toxicity_metrics.py +1 -1
  107. helm/benchmark/metrics/toxicity_utils.py +23 -0
  108. helm/benchmark/metrics/unitxt_metrics.py +81 -0
  109. helm/benchmark/metrics/vision_language/__init__.py +0 -0
  110. helm/benchmark/metrics/vision_language/emd_utils.py +341 -0
  111. helm/benchmark/metrics/vision_language/image_metrics.py +575 -0
  112. helm/benchmark/metrics/vision_language/image_utils.py +100 -0
  113. helm/benchmark/model_deployment_registry.py +74 -0
  114. helm/benchmark/model_metadata_registry.py +41 -1
  115. helm/benchmark/multi_gpu_runner.py +133 -0
  116. helm/benchmark/presentation/create_plots.py +8 -7
  117. helm/benchmark/presentation/run_display.py +26 -10
  118. helm/benchmark/presentation/schema.py +15 -40
  119. helm/benchmark/presentation/summarize.py +119 -79
  120. helm/benchmark/presentation/table.py +8 -8
  121. helm/benchmark/presentation/test_contamination.py +2 -2
  122. helm/benchmark/presentation/test_run_entry.py +1 -2
  123. helm/benchmark/presentation/test_summarize.py +3 -3
  124. helm/benchmark/run.py +54 -26
  125. helm/benchmark/run_expander.py +205 -35
  126. helm/benchmark/run_spec.py +93 -0
  127. helm/benchmark/run_spec_factory.py +163 -0
  128. helm/benchmark/run_specs/__init__.py +0 -0
  129. helm/benchmark/run_specs/classic_run_specs.py +1510 -0
  130. helm/benchmark/run_specs/cleva_run_specs.py +277 -0
  131. helm/benchmark/run_specs/decodingtrust_run_specs.py +314 -0
  132. helm/benchmark/run_specs/heim_run_specs.py +623 -0
  133. helm/benchmark/run_specs/instruction_following_run_specs.py +129 -0
  134. helm/benchmark/run_specs/lite_run_specs.py +307 -0
  135. helm/benchmark/run_specs/simple_run_specs.py +104 -0
  136. helm/benchmark/run_specs/unitxt_run_specs.py +42 -0
  137. helm/benchmark/run_specs/vlm_run_specs.py +757 -0
  138. helm/benchmark/runner.py +51 -57
  139. helm/benchmark/runner_config_registry.py +21 -0
  140. helm/benchmark/scenarios/bbq_scenario.py +1 -1
  141. helm/benchmark/scenarios/bold_scenario.py +2 -2
  142. helm/benchmark/scenarios/code_scenario.py +1 -0
  143. helm/benchmark/scenarios/decodingtrust_adv_demonstration_scenario.py +169 -0
  144. helm/benchmark/scenarios/decodingtrust_adv_robustness_scenario.py +121 -0
  145. helm/benchmark/scenarios/decodingtrust_fairness_scenario.py +77 -0
  146. helm/benchmark/scenarios/decodingtrust_machine_ethics_scenario.py +324 -0
  147. helm/benchmark/scenarios/decodingtrust_ood_robustness_scenario.py +204 -0
  148. helm/benchmark/scenarios/decodingtrust_privacy_scenario.py +559 -0
  149. helm/benchmark/scenarios/decodingtrust_stereotype_bias_scenario.py +67 -0
  150. helm/benchmark/scenarios/decodingtrust_toxicity_prompts_scenario.py +78 -0
  151. helm/benchmark/scenarios/dialogue_scenarios.py +0 -1
  152. helm/benchmark/scenarios/image_generation/__init__.py +0 -0
  153. helm/benchmark/scenarios/image_generation/common_syntactic_processes_scenario.py +105 -0
  154. helm/benchmark/scenarios/image_generation/cub200_scenario.py +95 -0
  155. helm/benchmark/scenarios/image_generation/daily_dalle_scenario.py +124 -0
  156. helm/benchmark/scenarios/image_generation/demographic_stereotypes_scenario.py +82 -0
  157. helm/benchmark/scenarios/image_generation/detection_scenario.py +83 -0
  158. helm/benchmark/scenarios/image_generation/draw_bench_scenario.py +74 -0
  159. helm/benchmark/scenarios/image_generation/i2p_scenario.py +57 -0
  160. helm/benchmark/scenarios/image_generation/landing_page_scenario.py +46 -0
  161. helm/benchmark/scenarios/image_generation/logos_scenario.py +223 -0
  162. helm/benchmark/scenarios/image_generation/magazine_cover_scenario.py +91 -0
  163. helm/benchmark/scenarios/image_generation/mental_disorders_scenario.py +46 -0
  164. helm/benchmark/scenarios/image_generation/mscoco_scenario.py +91 -0
  165. helm/benchmark/scenarios/image_generation/paint_skills_scenario.py +72 -0
  166. helm/benchmark/scenarios/image_generation/parti_prompts_scenario.py +94 -0
  167. helm/benchmark/scenarios/image_generation/radiology_scenario.py +42 -0
  168. helm/benchmark/scenarios/image_generation/relational_understanding_scenario.py +52 -0
  169. helm/benchmark/scenarios/image_generation/time_most_significant_historical_figures_scenario.py +124 -0
  170. helm/benchmark/scenarios/image_generation/winoground_scenario.py +62 -0
  171. helm/benchmark/scenarios/imdb_scenario.py +0 -1
  172. helm/benchmark/scenarios/legalbench_scenario.py +6 -2
  173. helm/benchmark/scenarios/live_qa_scenario.py +94 -0
  174. helm/benchmark/scenarios/lm_entry_scenario.py +185 -0
  175. helm/benchmark/scenarios/math_scenario.py +19 -2
  176. helm/benchmark/scenarios/medication_qa_scenario.py +60 -0
  177. helm/benchmark/scenarios/numeracy_scenario.py +1 -1
  178. helm/benchmark/scenarios/opinions_qa_scenario.py +0 -4
  179. helm/benchmark/scenarios/scenario.py +4 -0
  180. helm/benchmark/scenarios/simple_scenarios.py +122 -1
  181. helm/benchmark/scenarios/test_math_scenario.py +6 -0
  182. helm/benchmark/scenarios/test_scenario.py +6 -3
  183. helm/benchmark/scenarios/test_simple_scenarios.py +50 -0
  184. helm/benchmark/scenarios/thai_exam_scenario.py +135 -0
  185. helm/benchmark/scenarios/unitxt_scenario.py +56 -0
  186. helm/benchmark/scenarios/verifiability_judgment_scenario.py +3 -1
  187. helm/benchmark/scenarios/vicuna_scenario.py +1 -1
  188. helm/benchmark/scenarios/vision_language/a_okvqa_scenario.py +83 -0
  189. helm/benchmark/scenarios/vision_language/bingo_scenario.py +103 -0
  190. helm/benchmark/scenarios/vision_language/crossmodal_3600_scenario.py +134 -0
  191. helm/benchmark/scenarios/vision_language/flickr30k_scenario.py +74 -0
  192. helm/benchmark/scenarios/vision_language/gqa_scenario.py +91 -0
  193. helm/benchmark/scenarios/vision_language/hateful_memes_scenario.py +94 -0
  194. helm/benchmark/scenarios/vision_language/heim_human_eval_scenario.py +113 -0
  195. helm/benchmark/scenarios/vision_language/image2structure/__init__.py +0 -0
  196. helm/benchmark/scenarios/vision_language/image2structure/chart2csv_scenario.py +55 -0
  197. helm/benchmark/scenarios/vision_language/image2structure/image2structure_scenario.py +214 -0
  198. helm/benchmark/scenarios/vision_language/image2structure/latex_scenario.py +25 -0
  199. helm/benchmark/scenarios/vision_language/image2structure/musicsheet_scenario.py +20 -0
  200. helm/benchmark/scenarios/vision_language/image2structure/utils_latex.py +347 -0
  201. helm/benchmark/scenarios/vision_language/image2structure/webpage/__init__.py +0 -0
  202. helm/benchmark/scenarios/vision_language/image2structure/webpage/driver.py +84 -0
  203. helm/benchmark/scenarios/vision_language/image2structure/webpage/jekyll_server.py +182 -0
  204. helm/benchmark/scenarios/vision_language/image2structure/webpage/utils.py +31 -0
  205. helm/benchmark/scenarios/vision_language/image2structure/webpage_scenario.py +225 -0
  206. helm/benchmark/scenarios/vision_language/math_vista_scenario.py +117 -0
  207. helm/benchmark/scenarios/vision_language/mementos_scenario.py +124 -0
  208. helm/benchmark/scenarios/vision_language/mm_safety_bench_scenario.py +103 -0
  209. helm/benchmark/scenarios/vision_language/mme_scenario.py +145 -0
  210. helm/benchmark/scenarios/vision_language/mmmu_scenario.py +187 -0
  211. helm/benchmark/scenarios/vision_language/mscoco_captioning_scenario.py +92 -0
  212. helm/benchmark/scenarios/vision_language/mscoco_categorization_scenario.py +117 -0
  213. helm/benchmark/scenarios/vision_language/multipanelvqa_scenario.py +169 -0
  214. helm/benchmark/scenarios/vision_language/originality_scenario.py +35 -0
  215. helm/benchmark/scenarios/vision_language/pairs_scenario.py +246 -0
  216. helm/benchmark/scenarios/vision_language/pope_scenario.py +104 -0
  217. helm/benchmark/scenarios/vision_language/seed_bench_scenario.py +129 -0
  218. helm/benchmark/scenarios/vision_language/unicorn_scenario.py +108 -0
  219. helm/benchmark/scenarios/vision_language/viz_wiz_scenario.py +3 -4
  220. helm/benchmark/scenarios/vision_language/vqa_scenario.py +5 -3
  221. helm/benchmark/scenarios/wmt_14_scenario.py +1 -1
  222. helm/benchmark/server.py +24 -1
  223. helm/benchmark/slurm_runner.py +70 -49
  224. helm/benchmark/static/benchmarking.js +1 -1
  225. helm/benchmark/static/schema_classic.yaml +258 -1066
  226. helm/benchmark/static/schema_image2structure.yaml +304 -0
  227. helm/benchmark/static/schema_instruction_following.yaml +210 -0
  228. helm/benchmark/static/schema_lite.yaml +2 -227
  229. helm/benchmark/static/schema_mmlu.yaml +1507 -0
  230. helm/benchmark/static/schema_unitxt.yaml +428 -0
  231. helm/benchmark/static/schema_vhelm_lite.yaml +164 -0
  232. helm/benchmark/static/schema_vlm.yaml +823 -0
  233. helm/benchmark/static_build/assets/01-694cb9b7.png +0 -0
  234. helm/benchmark/static_build/assets/ai21-0eb91ec3.png +0 -0
  235. helm/benchmark/static_build/assets/aleph-alpha-7ce10034.png +0 -0
  236. helm/benchmark/static_build/assets/anthropic-70d8bc39.png +0 -0
  237. helm/benchmark/static_build/assets/bigscience-7f0400c0.png +0 -0
  238. helm/benchmark/static_build/assets/cohere-3550c6cb.png +0 -0
  239. helm/benchmark/static_build/assets/crfm-logo-74391ab8.png +0 -0
  240. helm/benchmark/static_build/assets/eleutherai-b9451114.png +0 -0
  241. helm/benchmark/static_build/assets/google-06d997ad.png +0 -0
  242. helm/benchmark/static_build/assets/heim-logo-3e5e3aa4.png +0 -0
  243. helm/benchmark/static_build/assets/helm-logo-simple-2ed5400b.png +0 -0
  244. helm/benchmark/static_build/assets/helmhero-28e90f4d.png +0 -0
  245. helm/benchmark/static_build/assets/index-737eef9e.js +10 -0
  246. helm/benchmark/static_build/assets/index-878a1094.css +1 -0
  247. helm/benchmark/static_build/assets/meta-5580e9f1.png +0 -0
  248. helm/benchmark/static_build/assets/microsoft-f5ee5016.png +0 -0
  249. helm/benchmark/static_build/assets/mistral-18e1be23.png +0 -0
  250. helm/benchmark/static_build/assets/nvidia-86fa75c1.png +0 -0
  251. helm/benchmark/static_build/assets/openai-3f8653e4.png +0 -0
  252. helm/benchmark/static_build/assets/react-d4a0b69b.js +85 -0
  253. helm/benchmark/static_build/assets/recharts-6d337683.js +97 -0
  254. helm/benchmark/static_build/assets/tii-24de195c.png +0 -0
  255. helm/benchmark/static_build/assets/together-a665a35b.png +0 -0
  256. helm/benchmark/static_build/assets/tremor-54a99cc4.js +10 -0
  257. helm/benchmark/static_build/assets/tsinghua-keg-97d4b395.png +0 -0
  258. helm/benchmark/static_build/assets/vhelm-framework-cde7618a.png +0 -0
  259. helm/benchmark/static_build/assets/vhelm-model-6d812526.png +0 -0
  260. helm/benchmark/static_build/assets/yandex-38e09d70.png +0 -0
  261. helm/benchmark/static_build/config.js +4 -0
  262. helm/benchmark/static_build/index.html +20 -0
  263. helm/benchmark/test_data_preprocessor.py +3 -3
  264. helm/benchmark/test_run_expander.py +1 -1
  265. helm/benchmark/window_services/ai21_window_service.py +22 -33
  266. helm/benchmark/window_services/cohere_window_service.py +1 -63
  267. helm/benchmark/window_services/default_window_service.py +2 -44
  268. helm/benchmark/window_services/encoder_decoder_window_service.py +0 -11
  269. helm/benchmark/window_services/ice_window_service.py +0 -34
  270. helm/benchmark/window_services/image_generation/__init__.py +0 -0
  271. helm/benchmark/window_services/image_generation/clip_window_service.py +15 -0
  272. helm/benchmark/window_services/image_generation/lexica_search_window_service.py +9 -0
  273. helm/benchmark/window_services/image_generation/openai_dalle_window_service.py +9 -0
  274. helm/benchmark/window_services/image_generation/test_clip_window_service.py +29 -0
  275. helm/benchmark/window_services/image_generation/test_openai_dalle_window_service.py +30 -0
  276. helm/benchmark/window_services/local_window_service.py +21 -4
  277. helm/benchmark/window_services/test_anthropic_window_service.py +2 -1
  278. helm/benchmark/window_services/test_bloom_window_service.py +2 -1
  279. helm/benchmark/window_services/test_cohere_window_service.py +2 -1
  280. helm/benchmark/window_services/test_flan_t5_window_service.py +2 -1
  281. helm/benchmark/window_services/test_gpt2_window_service.py +2 -2
  282. helm/benchmark/window_services/test_gpt4_window_service.py +2 -1
  283. helm/benchmark/window_services/test_gptj_window_service.py +3 -2
  284. helm/benchmark/window_services/test_gptneox_window_service.py +3 -2
  285. helm/benchmark/window_services/test_ice_window_service.py +2 -1
  286. helm/benchmark/window_services/test_openai_window_service.py +2 -1
  287. helm/benchmark/window_services/test_opt_window_service.py +3 -2
  288. helm/benchmark/window_services/test_palmyra_window_service.py +2 -1
  289. helm/benchmark/window_services/test_t0pp_window_service.py +2 -1
  290. helm/benchmark/window_services/test_t511b_window_service.py +2 -1
  291. helm/benchmark/window_services/test_ul2_window_service.py +2 -1
  292. helm/benchmark/window_services/test_utils.py +3 -2
  293. helm/benchmark/window_services/test_yalm_window_service.py +2 -1
  294. helm/benchmark/window_services/window_service.py +42 -0
  295. helm/benchmark/window_services/window_service_factory.py +4 -1
  296. helm/benchmark/window_services/yalm_window_service.py +0 -27
  297. helm/clients/__init__.py +0 -0
  298. helm/{proxy/clients → clients}/ai21_client.py +3 -9
  299. helm/clients/aleph_alpha_client.py +112 -0
  300. helm/{proxy/clients → clients}/anthropic_client.py +233 -18
  301. helm/{proxy/clients → clients}/auto_client.py +59 -31
  302. helm/clients/bedrock_client.py +128 -0
  303. helm/clients/bedrock_utils.py +72 -0
  304. helm/{proxy/clients → clients}/client.py +65 -7
  305. helm/clients/clip_score_client.py +49 -0
  306. helm/clients/clip_scorers/__init__.py +0 -0
  307. helm/clients/clip_scorers/base_clip_scorer.py +18 -0
  308. helm/clients/clip_scorers/clip_scorer.py +50 -0
  309. helm/clients/clip_scorers/multilingual_clip_scorer.py +50 -0
  310. helm/{proxy/clients → clients}/cohere_client.py +4 -11
  311. helm/clients/gcs_client.py +82 -0
  312. helm/{proxy/clients → clients}/google_client.py +5 -5
  313. helm/clients/google_translate_client.py +35 -0
  314. helm/{proxy/clients → clients}/http_model_client.py +5 -7
  315. helm/{proxy/clients → clients}/huggingface_client.py +43 -64
  316. helm/clients/image_generation/__init__.py +0 -0
  317. helm/clients/image_generation/adobe_vision_client.py +78 -0
  318. helm/clients/image_generation/aleph_alpha_image_generation_client.py +98 -0
  319. helm/clients/image_generation/cogview2/__init__.py +0 -0
  320. helm/clients/image_generation/cogview2/coglm_strategy.py +96 -0
  321. helm/clients/image_generation/cogview2/coglm_utils.py +82 -0
  322. helm/clients/image_generation/cogview2/sr_pipeline/__init__.py +15 -0
  323. helm/clients/image_generation/cogview2/sr_pipeline/direct_sr.py +96 -0
  324. helm/clients/image_generation/cogview2/sr_pipeline/dsr_model.py +254 -0
  325. helm/clients/image_generation/cogview2/sr_pipeline/dsr_sampling.py +190 -0
  326. helm/clients/image_generation/cogview2/sr_pipeline/iterative_sr.py +141 -0
  327. helm/clients/image_generation/cogview2/sr_pipeline/itersr_model.py +269 -0
  328. helm/clients/image_generation/cogview2/sr_pipeline/itersr_sampling.py +120 -0
  329. helm/clients/image_generation/cogview2/sr_pipeline/sr_group.py +42 -0
  330. helm/clients/image_generation/cogview2_client.py +191 -0
  331. helm/clients/image_generation/dalle2_client.py +192 -0
  332. helm/clients/image_generation/dalle3_client.py +108 -0
  333. helm/clients/image_generation/dalle_mini/__init__.py +3 -0
  334. helm/clients/image_generation/dalle_mini/data.py +442 -0
  335. helm/clients/image_generation/dalle_mini/model/__init__.py +5 -0
  336. helm/clients/image_generation/dalle_mini/model/configuration.py +175 -0
  337. helm/clients/image_generation/dalle_mini/model/modeling.py +1834 -0
  338. helm/clients/image_generation/dalle_mini/model/partitions.py +84 -0
  339. helm/clients/image_generation/dalle_mini/model/processor.py +63 -0
  340. helm/clients/image_generation/dalle_mini/model/text.py +251 -0
  341. helm/clients/image_generation/dalle_mini/model/tokenizer.py +9 -0
  342. helm/clients/image_generation/dalle_mini/model/utils.py +29 -0
  343. helm/clients/image_generation/dalle_mini/vqgan_jax/__init__.py +1 -0
  344. helm/clients/image_generation/dalle_mini/vqgan_jax/configuration_vqgan.py +40 -0
  345. helm/clients/image_generation/dalle_mini/vqgan_jax/convert_pt_model_to_jax.py +107 -0
  346. helm/clients/image_generation/dalle_mini/vqgan_jax/modeling_flax_vqgan.py +610 -0
  347. helm/clients/image_generation/dalle_mini_client.py +190 -0
  348. helm/clients/image_generation/deep_floyd_client.py +78 -0
  349. helm/clients/image_generation/huggingface_diffusers_client.py +249 -0
  350. helm/clients/image_generation/image_generation_client_utils.py +9 -0
  351. helm/clients/image_generation/lexica_client.py +86 -0
  352. helm/clients/image_generation/mindalle/__init__.py +0 -0
  353. helm/clients/image_generation/mindalle/models/__init__.py +216 -0
  354. helm/clients/image_generation/mindalle/models/stage1/__init__.py +0 -0
  355. helm/clients/image_generation/mindalle/models/stage1/layers.py +312 -0
  356. helm/clients/image_generation/mindalle/models/stage1/vqgan.py +103 -0
  357. helm/clients/image_generation/mindalle/models/stage2/__init__.py +0 -0
  358. helm/clients/image_generation/mindalle/models/stage2/layers.py +144 -0
  359. helm/clients/image_generation/mindalle/models/stage2/transformer.py +268 -0
  360. helm/clients/image_generation/mindalle/models/tokenizer.py +30 -0
  361. helm/clients/image_generation/mindalle/utils/__init__.py +3 -0
  362. helm/clients/image_generation/mindalle/utils/config.py +129 -0
  363. helm/clients/image_generation/mindalle/utils/sampling.py +149 -0
  364. helm/clients/image_generation/mindalle/utils/utils.py +89 -0
  365. helm/clients/image_generation/mindalle_client.py +115 -0
  366. helm/clients/image_generation/nudity_check_client.py +64 -0
  367. helm/clients/image_generation/together_image_generation_client.py +111 -0
  368. helm/{proxy/clients → clients}/lit_gpt_client.py +4 -4
  369. helm/{proxy/clients → clients}/megatron_client.py +5 -5
  370. helm/clients/mistral_client.py +134 -0
  371. helm/clients/moderation_api_client.py +109 -0
  372. helm/clients/open_lm_client.py +43 -0
  373. helm/clients/openai_client.py +301 -0
  374. helm/{proxy/clients → clients}/palmyra_client.py +6 -8
  375. helm/{proxy/clients → clients}/perspective_api_client.py +7 -8
  376. helm/clients/simple_client.py +64 -0
  377. helm/{proxy/clients → clients}/test_auto_client.py +13 -15
  378. helm/clients/test_client.py +100 -0
  379. helm/{proxy/clients → clients}/test_huggingface_client.py +15 -16
  380. helm/clients/test_simple_client.py +19 -0
  381. helm/{proxy/clients → clients}/test_together_client.py +20 -8
  382. helm/{proxy/clients → clients}/together_client.py +104 -73
  383. helm/clients/vertexai_client.py +400 -0
  384. helm/clients/vision_language/__init__.py +0 -0
  385. helm/clients/vision_language/huggingface_vision2seq_client.py +145 -0
  386. helm/clients/vision_language/huggingface_vlm_client.py +111 -0
  387. helm/{proxy/clients → clients}/vision_language/idefics_client.py +54 -49
  388. helm/clients/vision_language/open_flamingo/__init__.py +2 -0
  389. helm/clients/vision_language/open_flamingo/src/__init__.py +0 -0
  390. helm/clients/vision_language/open_flamingo/src/factory.py +147 -0
  391. helm/clients/vision_language/open_flamingo/src/flamingo.py +337 -0
  392. helm/clients/vision_language/open_flamingo/src/flamingo_lm.py +155 -0
  393. helm/clients/vision_language/open_flamingo/src/helpers.py +267 -0
  394. helm/clients/vision_language/open_flamingo/src/utils.py +47 -0
  395. helm/clients/vision_language/open_flamingo_client.py +155 -0
  396. helm/clients/vision_language/qwen_vlm_client.py +171 -0
  397. helm/clients/vllm_client.py +46 -0
  398. helm/common/cache.py +16 -4
  399. helm/common/cache_backend_config.py +47 -0
  400. helm/common/clip_score_request.py +41 -0
  401. helm/common/file_caches/__init__.py +0 -0
  402. helm/common/file_caches/file_cache.py +16 -0
  403. helm/common/file_caches/local_file_cache.py +61 -0
  404. helm/common/file_caches/test_local_file_cache.py +25 -0
  405. helm/common/file_upload_request.py +27 -0
  406. helm/common/general.py +1 -1
  407. helm/common/image_generation_parameters.py +25 -0
  408. helm/common/images_utils.py +33 -3
  409. helm/common/key_value_store.py +35 -4
  410. helm/common/media_object.py +13 -0
  411. helm/common/moderations_api_request.py +71 -0
  412. helm/common/mongo_key_value_store.py +3 -3
  413. helm/common/multimodal_request_utils.py +31 -0
  414. helm/common/nudity_check_request.py +29 -0
  415. helm/common/request.py +15 -17
  416. helm/common/test_general.py +6 -0
  417. helm/common/tokenization_request.py +1 -1
  418. helm/config/model_deployments.yaml +1159 -538
  419. helm/config/model_metadata.yaml +868 -41
  420. helm/config/tokenizer_configs.yaml +149 -43
  421. helm/proxy/accounts.py +31 -4
  422. helm/proxy/critique/mechanical_turk_critique_importer.py +3 -0
  423. helm/proxy/critique/model_critique_client.py +8 -6
  424. helm/proxy/example_queries.py +29 -17
  425. helm/proxy/server.py +70 -5
  426. helm/proxy/services/remote_service.py +31 -0
  427. helm/proxy/services/server_service.py +96 -16
  428. helm/proxy/services/service.py +30 -0
  429. helm/proxy/services/test_remote_service.py +4 -3
  430. helm/proxy/services/test_service.py +0 -12
  431. helm/proxy/test_accounts.py +32 -0
  432. helm/proxy/token_counters/auto_token_counter.py +37 -37
  433. helm/proxy/token_counters/test_auto_token_counter.py +164 -0
  434. helm/proxy/token_counters/token_counter.py +3 -5
  435. helm/tokenizers/__init__.py +0 -0
  436. helm/{proxy/tokenizers → tokenizers}/ai21_tokenizer.py +3 -3
  437. helm/{proxy/tokenizers → tokenizers}/anthropic_tokenizer.py +1 -1
  438. helm/{proxy/tokenizers → tokenizers}/auto_tokenizer.py +6 -9
  439. helm/{proxy/tokenizers → tokenizers}/cohere_tokenizer.py +1 -1
  440. helm/{proxy/tokenizers → tokenizers}/http_model_tokenizer.py +3 -3
  441. helm/{proxy/tokenizers → tokenizers}/huggingface_tokenizer.py +7 -26
  442. helm/tokenizers/simple_tokenizer.py +33 -0
  443. helm/{proxy/tokenizers → tokenizers}/test_anthropic_tokenizer.py +1 -1
  444. helm/{proxy/tokenizers → tokenizers}/test_huggingface_tokenizer.py +3 -0
  445. helm/tokenizers/test_simple_tokenizer.py +33 -0
  446. helm/{proxy/tokenizers → tokenizers}/vertexai_tokenizer.py +1 -1
  447. helm/{proxy/tokenizers → tokenizers}/yalm_tokenizer.py +5 -3
  448. helm/tokenizers/yalm_tokenizer_data/__init__.py +0 -0
  449. helm/tokenizers/yalm_tokenizer_data/voc_100b.sp +0 -0
  450. helm/{proxy/tokenizers → tokenizers}/yalm_tokenizer_data/yalm_tokenizer.py +1 -1
  451. crfm_helm-0.4.0.dist-info/RECORD +0 -397
  452. helm/benchmark/run_specs.py +0 -2762
  453. helm/benchmark/test_model_deployment_definition.py +0 -92
  454. helm/benchmark/test_model_properties.py +0 -1570
  455. helm/benchmark/vlm_run_specs.py +0 -97
  456. helm/benchmark/window_services/flan_t5_window_service.py +0 -29
  457. helm/benchmark/window_services/gpt2_window_service.py +0 -32
  458. helm/benchmark/window_services/huggingface_window_service.py +0 -60
  459. helm/benchmark/window_services/t0pp_window_service.py +0 -35
  460. helm/benchmark/window_services/t511b_window_service.py +0 -30
  461. helm/benchmark/window_services/test_mt_nlg_window_service.py +0 -48
  462. helm/benchmark/window_services/ul2_window_service.py +0 -30
  463. helm/benchmark/window_services/wider_ai21_window_service.py +0 -24
  464. helm/common/cache_utils.py +0 -14
  465. helm/proxy/clients/aleph_alpha_client.py +0 -95
  466. helm/proxy/clients/goose_ai_client.py +0 -99
  467. helm/proxy/clients/microsoft_client.py +0 -180
  468. helm/proxy/clients/openai_client.py +0 -206
  469. helm/proxy/clients/simple_client.py +0 -60
  470. helm/proxy/clients/test_client.py +0 -49
  471. helm/proxy/clients/vertexai_client.py +0 -115
  472. helm/proxy/token_counters/ai21_token_counter.py +0 -20
  473. helm/proxy/token_counters/cohere_token_counter.py +0 -13
  474. helm/proxy/token_counters/free_token_counter.py +0 -12
  475. helm/proxy/token_counters/gooseai_token_counter.py +0 -24
  476. helm/proxy/token_counters/openai_token_counter.py +0 -22
  477. helm/proxy/token_counters/test_ai21_token_counter.py +0 -88
  478. helm/proxy/token_counters/test_openai_token_counter.py +0 -81
  479. helm/proxy/tokenizers/simple_tokenizer.py +0 -32
  480. {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/LICENSE +0 -0
  481. {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/entry_points.txt +0 -0
  482. {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/top_level.txt +0 -0
  483. /helm/{proxy/clients → benchmark/annotation}/__init__.py +0 -0
  484. /helm/{proxy/clients/vision_language → benchmark/annotation/image2structure}/__init__.py +0 -0
  485. /helm/{proxy/tokenizers → benchmark/metrics/image_generation}/__init__.py +0 -0
  486. /helm/{proxy/tokenizers/yalm_tokenizer_data → benchmark/metrics/image_generation/detectors}/__init__.py +0 -0
  487. /helm/{proxy/clients → clients}/ai21_utils.py +0 -0
  488. /helm/{proxy/clients → clients}/cohere_utils.py +0 -0
  489. /helm/{proxy/clients → clients}/lit_gpt_generate.py +0 -0
  490. /helm/{proxy/clients → clients}/toxicity_classifier_client.py +0 -0
  491. /helm/{proxy/tokenizers → tokenizers}/aleph_alpha_tokenizer.py +0 -0
  492. /helm/{proxy/tokenizers → tokenizers}/caching_tokenizer.py +0 -0
  493. /helm/{proxy/tokenizers → tokenizers}/ice_tokenizer.py +0 -0
  494. /helm/{proxy/tokenizers → tokenizers}/lit_gpt_tokenizer.py +0 -0
  495. /helm/{proxy/tokenizers → tokenizers}/test_ice_tokenizer.py +0 -0
  496. /helm/{proxy/tokenizers → tokenizers}/test_yalm_tokenizer.py +0 -0
  497. /helm/{proxy/tokenizers → tokenizers}/tiktoken_tokenizer.py +0 -0
  498. /helm/{proxy/tokenizers → tokenizers}/tokenizer.py +0 -0
  499. /helm/{proxy/tokenizers → tokenizers}/yalm_tokenizer_data/test_yalm_tokenizer.py +0 -0
@@ -0,0 +1,757 @@
1
+ """Run spec functions for evaluating Vision-Language Models."""
2
+
3
+ from typing import List, Optional, Dict
4
+
5
+ from helm.benchmark.adaptation.adapter_spec import AdapterSpec
6
+ from helm.benchmark.adaptation.adapters.adapter_factory import (
7
+ ADAPT_GENERATION_MULTIMODAL,
8
+ ADAPT_MULTIPLE_CHOICE_JOINT_MULTIMODAL,
9
+ )
10
+ from helm.benchmark.metrics.common_metric_specs import (
11
+ get_basic_reference_metric_specs,
12
+ get_exact_match_metric_specs,
13
+ get_generative_harms_metric_specs,
14
+ get_basic_metric_specs,
15
+ )
16
+ from helm.benchmark.metrics.metric import MetricSpec
17
+ from helm.benchmark.run_spec import RunSpec, run_spec_function
18
+ from helm.benchmark.scenarios.scenario import ScenarioSpec
19
+ from helm.benchmark.annotation.annotator import AnnotatorSpec
20
+
21
+
22
+ ############################################################
23
+ # Prototypical adapter specs for VLM evaluation
24
+
25
+
26
+ def _get_generation_adapter_spec(
27
+ instructions: str = "",
28
+ input_prefix: str = "",
29
+ input_suffix: str = "",
30
+ output_prefix: str = "",
31
+ output_suffix: str = "",
32
+ max_tokens: int = 100,
33
+ stop_sequences: Optional[List[str]] = None,
34
+ ) -> AdapterSpec:
35
+ return AdapterSpec(
36
+ method=ADAPT_GENERATION_MULTIMODAL,
37
+ global_prefix="",
38
+ instructions=instructions,
39
+ input_prefix=input_prefix,
40
+ input_suffix=input_suffix,
41
+ output_prefix=output_prefix,
42
+ output_suffix=output_suffix,
43
+ instance_prefix="\n",
44
+ # We focus on zero-shot evaluation for now as most open VLMs only support a single image input
45
+ max_train_instances=0,
46
+ num_outputs=1,
47
+ max_tokens=max_tokens,
48
+ stop_sequences=stop_sequences if stop_sequences is not None else [],
49
+ temperature=0.0,
50
+ random=None,
51
+ )
52
+
53
+
54
+ def _get_short_answer_generation_adapter_spec(instructions: Optional[str] = None) -> AdapterSpec:
55
+ return _get_generation_adapter_spec(
56
+ instructions=(
57
+ "Just give a short answer without answering in a complete sentence."
58
+ if instructions is None
59
+ else instructions
60
+ ),
61
+ max_tokens=20,
62
+ )
63
+
64
+
65
+ def _get_captioning_adapter_spec() -> AdapterSpec:
66
+ return _get_generation_adapter_spec(
67
+ instructions="Generate a caption for the following image. The caption should be short and does "
68
+ "not need to be a complete sentence.",
69
+ max_tokens=20,
70
+ )
71
+
72
+
73
+ def _get_multiple_choice_joint_adapter_spec(
74
+ input_noun: Optional[str],
75
+ output_noun: str,
76
+ max_train_instances: int = 0,
77
+ num_outputs: int = 1,
78
+ ) -> AdapterSpec:
79
+ return AdapterSpec(
80
+ method=ADAPT_MULTIPLE_CHOICE_JOINT_MULTIMODAL,
81
+ global_prefix="",
82
+ instructions="Answer the multiple choice question by just giving the letter of the correct answer.",
83
+ input_prefix=f"{input_noun}: " if input_noun is not None else "",
84
+ input_suffix="\n",
85
+ output_prefix=f"{output_noun}: ",
86
+ output_suffix="\n",
87
+ instance_prefix="\n",
88
+ max_train_instances=max_train_instances,
89
+ num_outputs=num_outputs,
90
+ max_tokens=1,
91
+ stop_sequences=["\n"],
92
+ temperature=0.0,
93
+ random=None,
94
+ )
95
+
96
+
97
+ ############################################################
98
+ # VHELM metric specs
99
+
100
+
101
+ def _get_open_ended_generation_metric_specs() -> List[MetricSpec]:
102
+ return get_basic_metric_specs(
103
+ ["exact_match", "quasi_exact_match", "f1_score", "rouge_l", "bleu_1", "bleu_4", "cider"]
104
+ )
105
+
106
+
107
+ def _get_image2structure_metric_specs(
108
+ generation_type: str,
109
+ metric_names: Optional[List[str]] = None,
110
+ args: Optional[Dict] = None,
111
+ include_edit_similarity: bool = True,
112
+ size_handling_method: str = "resize",
113
+ ) -> List[MetricSpec]:
114
+ from helm.benchmark.metrics.vision_language.image_metrics import AnnotatedImageMetrics
115
+
116
+ if metric_names is None:
117
+ metric_names = [
118
+ AnnotatedImageMetrics.PIXEL_SIMILARITY,
119
+ AnnotatedImageMetrics.FID_SIMILARITY,
120
+ AnnotatedImageMetrics.BLOCK_EARTH_MOVER_SIMILARITY,
121
+ AnnotatedImageMetrics.BLOCK_EARTH_MOVER_SIMILARITY_NORM2,
122
+ AnnotatedImageMetrics.BLOCK_EARTH_MOVER_SIMILARITY_NORM1,
123
+ ]
124
+ if include_edit_similarity:
125
+ metric_names.append(AnnotatedImageMetrics.EDIT_SIMILARITY)
126
+ if args is None:
127
+ args = {}
128
+ metric_specs = [
129
+ MetricSpec(
130
+ class_name="helm.benchmark.metrics.vision_language.image_metrics.AnnotatedImageMetrics",
131
+ args={
132
+ "generation_type": generation_type,
133
+ "metric_names": metric_names,
134
+ "size_handling_method": size_handling_method,
135
+ **args,
136
+ },
137
+ ),
138
+ ]
139
+ return metric_specs + get_basic_reference_metric_specs()
140
+
141
+
142
+ ############################################################
143
+ # VHELM run specs
144
+
145
+
146
+ @run_spec_function("a_okvqa")
147
+ def get_a_okvqa_spec() -> RunSpec:
148
+ scenario_spec = ScenarioSpec(
149
+ class_name="helm.benchmark.scenarios.vision_language.a_okvqa_scenario.AOKVQAScenario",
150
+ args={},
151
+ )
152
+
153
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
154
+ input_noun=None, output_noun="Answer", max_train_instances=0
155
+ )
156
+
157
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
158
+ run_spec_name: str = "a_okvqa"
159
+ return RunSpec(
160
+ name=run_spec_name,
161
+ scenario_spec=scenario_spec,
162
+ adapter_spec=adapter_spec,
163
+ metric_specs=metric_specs,
164
+ groups=[run_spec_name],
165
+ )
166
+
167
+
168
+ @run_spec_function("chart2csv")
169
+ def get_chart2csv_spec() -> RunSpec:
170
+ scenario_spec = ScenarioSpec(
171
+ class_name="helm.benchmark.scenarios.vision_language.image2structure.chart2csv_scenario.Chart2CSVScenario",
172
+ args={},
173
+ )
174
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(
175
+ instructions="Generate the CSV for the chart. Some of the labels may be missing due to the size of the chart. "
176
+ "Please infer the missing labels based on the surrounding context. "
177
+ "Just give the CSV without any explanation.",
178
+ max_tokens=1000,
179
+ )
180
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
181
+
182
+ run_spec_name: str = "chart2csv"
183
+ return RunSpec(
184
+ name=run_spec_name,
185
+ scenario_spec=scenario_spec,
186
+ adapter_spec=adapter_spec,
187
+ metric_specs=metric_specs,
188
+ groups=[run_spec_name],
189
+ )
190
+
191
+
192
+ @run_spec_function("crossmodal_3600")
193
+ def get_crossmodal_3600_spec(location: str, language: str) -> RunSpec:
194
+ scenario_spec = ScenarioSpec(
195
+ class_name="helm.benchmark.scenarios.vision_language.crossmodal_3600_scenario.Crossmodal3600Scenario",
196
+ args={"location": location, "language": language},
197
+ )
198
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(max_tokens=20)
199
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs() + _get_open_ended_generation_metric_specs()
200
+
201
+ run_spec_name: str = "crossmodal_3600"
202
+ return RunSpec(
203
+ name=f"{run_spec_name}:location={location},language={language}",
204
+ scenario_spec=scenario_spec,
205
+ adapter_spec=adapter_spec,
206
+ metric_specs=metric_specs,
207
+ groups=[run_spec_name],
208
+ )
209
+
210
+
211
+ @run_spec_function("flickr30k")
212
+ def get_flickr30k_spec() -> RunSpec:
213
+ scenario_spec = ScenarioSpec(
214
+ class_name="helm.benchmark.scenarios.vision_language.flickr30k_scenario.Flickr30KScenario", args={}
215
+ )
216
+ adapter_spec: AdapterSpec = _get_captioning_adapter_spec()
217
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs() + _get_open_ended_generation_metric_specs()
218
+
219
+ run_spec_name: str = "flickr30k"
220
+ return RunSpec(
221
+ name=run_spec_name,
222
+ scenario_spec=scenario_spec,
223
+ adapter_spec=adapter_spec,
224
+ metric_specs=metric_specs,
225
+ groups=[run_spec_name],
226
+ )
227
+
228
+
229
+ @run_spec_function("gqa")
230
+ def get_gqa_spec() -> RunSpec:
231
+ scenario_spec = ScenarioSpec(
232
+ class_name="helm.benchmark.scenarios.vision_language.gqa_scenario.GQAScenario", args={}
233
+ )
234
+ adapter_spec: AdapterSpec = _get_short_answer_generation_adapter_spec(
235
+ instructions="Answer the question using a single word or phrase."
236
+ )
237
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs() + _get_open_ended_generation_metric_specs()
238
+
239
+ run_spec_name: str = "gqa"
240
+ return RunSpec(
241
+ name=run_spec_name,
242
+ scenario_spec=scenario_spec,
243
+ adapter_spec=adapter_spec,
244
+ metric_specs=metric_specs,
245
+ groups=[run_spec_name],
246
+ )
247
+
248
+
249
+ @run_spec_function("hateful_memes")
250
+ def get_hateful_memes_spec() -> RunSpec:
251
+ scenario_spec = ScenarioSpec(
252
+ class_name="helm.benchmark.scenarios.vision_language.hateful_memes_scenario.HatefulMemesScenario", args={}
253
+ )
254
+ adapter_spec = _get_multiple_choice_joint_adapter_spec(input_noun=None, output_noun="Answer", max_train_instances=0)
255
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
256
+
257
+ run_spec_name: str = "hateful_memes"
258
+ return RunSpec(
259
+ name=run_spec_name,
260
+ scenario_spec=scenario_spec,
261
+ adapter_spec=adapter_spec,
262
+ metric_specs=metric_specs,
263
+ groups=[run_spec_name],
264
+ )
265
+
266
+
267
+ @run_spec_function("mm_safety_bench")
268
+ def get_mm_safety_bench_spec(subset: str) -> RunSpec:
269
+ scenario_spec = ScenarioSpec(
270
+ class_name="helm.benchmark.scenarios.vision_language.mm_safety_bench_scenario.MMSafetyBenchScenario",
271
+ args={"subset": subset},
272
+ )
273
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(max_tokens=500)
274
+ metric_specs: List[MetricSpec] = get_generative_harms_metric_specs(
275
+ include_basic_metrics=True, include_generative_harms_metrics=True
276
+ )
277
+
278
+ run_spec_name: str = "mm_safety_bench"
279
+ return RunSpec(
280
+ name=f"{run_spec_name}:subset={subset}",
281
+ scenario_spec=scenario_spec,
282
+ adapter_spec=adapter_spec,
283
+ metric_specs=metric_specs,
284
+ groups=[run_spec_name],
285
+ )
286
+
287
+
288
+ @run_spec_function("mscoco_captioning")
289
+ def get_mscoco_captioning_spec(long: bool = False) -> RunSpec:
290
+ scenario_spec = ScenarioSpec(
291
+ class_name="helm.benchmark.scenarios.vision_language.mscoco_captioning_scenario.MSCOCOCaptioningScenario",
292
+ args={},
293
+ )
294
+
295
+ adapter_spec: AdapterSpec
296
+ if long:
297
+ adapter_spec = _get_generation_adapter_spec(
298
+ instructions="Generate a long, detailed caption for the following image.",
299
+ max_tokens=150,
300
+ )
301
+ else:
302
+ adapter_spec = _get_captioning_adapter_spec()
303
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs() + _get_open_ended_generation_metric_specs()
304
+
305
+ run_spec_name: str = "mscoco_captioning"
306
+ if long:
307
+ run_spec_name += "_long"
308
+
309
+ return RunSpec(
310
+ name=run_spec_name,
311
+ scenario_spec=scenario_spec,
312
+ adapter_spec=adapter_spec,
313
+ metric_specs=metric_specs,
314
+ groups=[run_spec_name],
315
+ )
316
+
317
+
318
+ @run_spec_function("mscoco_categorization")
319
+ def get_mscoco_categorization_spec() -> RunSpec:
320
+ scenario_spec = ScenarioSpec(
321
+ class_name="helm.benchmark.scenarios.vision_language.mscoco_categorization_scenario."
322
+ "MSCOCOCategorizationScenario",
323
+ args={},
324
+ )
325
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
326
+ input_noun=None, output_noun="Answer", max_train_instances=0
327
+ )
328
+
329
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
330
+
331
+ run_spec_name: str = "mscoco_categorization"
332
+ return RunSpec(
333
+ name=run_spec_name,
334
+ scenario_spec=scenario_spec,
335
+ adapter_spec=adapter_spec,
336
+ metric_specs=metric_specs,
337
+ groups=[run_spec_name],
338
+ )
339
+
340
+
341
+ @run_spec_function("originality_vlm")
342
+ def get_originality_vlm_spec() -> RunSpec:
343
+ scenario_spec = ScenarioSpec(
344
+ class_name="helm.benchmark.scenarios.vision_language.originality_scenario.OriginalityScenario", args={}
345
+ )
346
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(max_tokens=500)
347
+ metric_specs: List[MetricSpec] = get_generative_harms_metric_specs(
348
+ include_basic_metrics=True, include_generative_harms_metrics=True
349
+ )
350
+
351
+ run_spec_name: str = "originality_vlm"
352
+ return RunSpec(
353
+ name=run_spec_name,
354
+ scenario_spec=scenario_spec,
355
+ adapter_spec=adapter_spec,
356
+ metric_specs=metric_specs,
357
+ groups=[run_spec_name],
358
+ )
359
+
360
+
361
+ @run_spec_function("viz_wiz")
362
+ def get_viz_wiz_spec() -> RunSpec:
363
+ scenario_spec = ScenarioSpec(
364
+ class_name="helm.benchmark.scenarios.vision_language.viz_wiz_scenario.VizWizScenario", args={}
365
+ )
366
+ adapter_spec: AdapterSpec = _get_short_answer_generation_adapter_spec(
367
+ # Following https://arxiv.org/abs/2310.03744
368
+ instructions="When the provided information is insufficient, respond with 'Unanswerable'. "
369
+ "Answer the question using a single word or phrase."
370
+ )
371
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs() + _get_open_ended_generation_metric_specs()
372
+
373
+ run_spec_name: str = "viz_wiz"
374
+ return RunSpec(
375
+ name=run_spec_name,
376
+ scenario_spec=scenario_spec,
377
+ adapter_spec=adapter_spec,
378
+ metric_specs=metric_specs,
379
+ groups=[run_spec_name],
380
+ )
381
+
382
+
383
+ @run_spec_function("vqa")
384
+ def get_vqa_spec() -> RunSpec:
385
+ scenario_spec = ScenarioSpec(
386
+ class_name="helm.benchmark.scenarios.vision_language.vqa_scenario.VQAScenario", args={}
387
+ )
388
+ # Following https://arxiv.org/abs/2310.03744
389
+ adapter_spec: AdapterSpec = _get_short_answer_generation_adapter_spec(
390
+ instructions='Answer the question using a single word or phrase. When the question asks "How many...", '
391
+ "respond with just a number (e.g., 3) and not the word corresponding to the number."
392
+ )
393
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs() + _get_open_ended_generation_metric_specs()
394
+
395
+ run_spec_name: str = "vqa"
396
+ return RunSpec(
397
+ name=run_spec_name,
398
+ scenario_spec=scenario_spec,
399
+ adapter_spec=adapter_spec,
400
+ metric_specs=metric_specs,
401
+ groups=[run_spec_name],
402
+ )
403
+
404
+
405
+ @run_spec_function("image2latex")
406
+ def get_image2latex_spec(subset: str, recompile_prompt: bool = False, args: Optional[Dict] = None) -> RunSpec:
407
+ scenario_spec = ScenarioSpec(
408
+ class_name="helm.benchmark.scenarios.vision_language.image2structure.latex_scenario.LatexScenario",
409
+ args={"subset": subset, "recompile_prompt": recompile_prompt},
410
+ )
411
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(
412
+ instructions="Just give a short answer without answering in a complete sentence.",
413
+ max_tokens=2000,
414
+ )
415
+ metric_specs: List[MetricSpec] = _get_image2structure_metric_specs(
416
+ generation_type="latex",
417
+ args=args,
418
+ include_edit_similarity=True,
419
+ size_handling_method="padding",
420
+ )
421
+ annotator_specs: List[AnnotatorSpec] = [
422
+ AnnotatorSpec(
423
+ class_name="helm.benchmark.annotation.image2structure.latex_compiler_annotator.LatexCompilerAnnotator",
424
+ )
425
+ ]
426
+
427
+ run_spec_name: str = "image2latex"
428
+ return RunSpec(
429
+ name=f"{run_spec_name}:subset={subset}",
430
+ scenario_spec=scenario_spec,
431
+ adapter_spec=adapter_spec,
432
+ metric_specs=metric_specs,
433
+ groups=[run_spec_name],
434
+ annotators=annotator_specs,
435
+ )
436
+
437
+
438
+ @run_spec_function("image2webpage")
439
+ def get_image2webpage_spec(subset: str, recompile_prompt: bool = False, args: Optional[Dict] = None) -> RunSpec:
440
+ scenario_spec = ScenarioSpec(
441
+ class_name="helm.benchmark.scenarios.vision_language.image2structure.webpage_scenario.WebpageScenario",
442
+ args={"subset": subset, "recompile_prompt": recompile_prompt},
443
+ )
444
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(
445
+ instructions="Just give a short answer without answering in a complete sentence.",
446
+ max_tokens=2000,
447
+ )
448
+ metric_specs: List[MetricSpec] = _get_image2structure_metric_specs(
449
+ generation_type="webpage",
450
+ args=args,
451
+ include_edit_similarity=True,
452
+ size_handling_method="none",
453
+ )
454
+ annotator_specs: List[AnnotatorSpec] = [
455
+ AnnotatorSpec(
456
+ class_name="helm.benchmark.annotation.image2structure.webpage_compiler_annotator.WebpageCompilerAnnotator",
457
+ )
458
+ ]
459
+
460
+ run_spec_name: str = "image2webpage"
461
+ return RunSpec(
462
+ name=f"{run_spec_name}:subset={subset}",
463
+ scenario_spec=scenario_spec,
464
+ adapter_spec=adapter_spec,
465
+ metric_specs=metric_specs,
466
+ groups=[run_spec_name],
467
+ annotators=annotator_specs,
468
+ )
469
+
470
+
471
+ @run_spec_function("math_vista")
472
+ def get_math_vista_spec(grade: str, question_type: str) -> RunSpec:
473
+ scenario_spec = ScenarioSpec(
474
+ class_name="helm.benchmark.scenarios.vision_language.math_vista_scenario.MathVistaScenario",
475
+ args={"grade": grade, "question_type": question_type},
476
+ )
477
+
478
+ adapter_spec: AdapterSpec
479
+ if question_type == "free_form":
480
+ adapter_spec = _get_short_answer_generation_adapter_spec()
481
+ elif question_type == "multi_choice":
482
+ adapter_spec = _get_multiple_choice_joint_adapter_spec(
483
+ input_noun=None, output_noun="Answer", max_train_instances=0
484
+ )
485
+ else:
486
+ raise ValueError(f"Invalid question type: {question_type}")
487
+
488
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
489
+ run_spec_name: str = "math_vista"
490
+ return RunSpec(
491
+ name=f"{run_spec_name}:grade={grade},question_type={question_type}",
492
+ scenario_spec=scenario_spec,
493
+ adapter_spec=adapter_spec,
494
+ metric_specs=metric_specs,
495
+ groups=[run_spec_name],
496
+ )
497
+
498
+
499
+ @run_spec_function("image2musicsheet")
500
+ def get_image2musicsheet_spec(args: Optional[Dict] = None) -> RunSpec:
501
+ scenario_spec = ScenarioSpec(
502
+ class_name="helm.benchmark.scenarios.vision_language.image2structure.musicsheet_scenario.MusicSheetScenario",
503
+ args={"subset": "music", "recompile_prompt": False}, # There os only one subset for music sheets
504
+ )
505
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(
506
+ instructions="Just give a short answer without answering in a complete sentence.",
507
+ max_tokens=2000,
508
+ )
509
+ metric_specs: List[MetricSpec] = _get_image2structure_metric_specs(
510
+ generation_type="lilypond",
511
+ args=args,
512
+ include_edit_similarity=False, # No ground truth for music sheets
513
+ size_handling_method="padding",
514
+ )
515
+ annotator_specs: List[AnnotatorSpec] = [
516
+ AnnotatorSpec(
517
+ class_name="helm.benchmark.annotation.image2structure.lilypond_compiler_annotator.LilypondCompilerAnnotator", # noqa: E501
518
+ )
519
+ ]
520
+
521
+ run_spec_name: str = "image2musicsheet"
522
+ return RunSpec(
523
+ name=run_spec_name,
524
+ scenario_spec=scenario_spec,
525
+ adapter_spec=adapter_spec,
526
+ metric_specs=metric_specs,
527
+ groups=[run_spec_name],
528
+ annotators=annotator_specs,
529
+ )
530
+
531
+
532
+ @run_spec_function("mmmu")
533
+ def get_mmmu_spec(subject: str, question_type: str) -> RunSpec:
534
+ scenario_spec = ScenarioSpec(
535
+ class_name="helm.benchmark.scenarios.vision_language.mmmu_scenario.MMMUScenario",
536
+ args={"subject": subject, "question_type": question_type},
537
+ )
538
+
539
+ adapter_spec: AdapterSpec
540
+ if question_type == "open":
541
+ adapter_spec = _get_short_answer_generation_adapter_spec()
542
+ elif question_type == "multiple-choice":
543
+ adapter_spec = _get_multiple_choice_joint_adapter_spec(
544
+ input_noun=None,
545
+ output_noun="Answer",
546
+ max_train_instances=0,
547
+ # instructions="Refer to the figure(s) and answer the multiple choice question by responding with just "
548
+ # "the letter of the correct answer (e.g., A, B, C, D, E).",
549
+ )
550
+ else:
551
+ raise ValueError(f"Invalid question type: {question_type}")
552
+
553
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
554
+ run_spec_name: str = "mmmu"
555
+ return RunSpec(
556
+ name=f"{run_spec_name}:subject={subject},question_type={question_type}",
557
+ scenario_spec=scenario_spec,
558
+ adapter_spec=adapter_spec,
559
+ metric_specs=metric_specs,
560
+ groups=[run_spec_name],
561
+ )
562
+
563
+
564
+ @run_spec_function("unicorn")
565
+ def get_unicorn_spec(subject: str) -> RunSpec:
566
+ scenario_spec = ScenarioSpec(
567
+ class_name="helm.benchmark.scenarios.vision_language.unicorn_scenario.UnicornScenario",
568
+ args={"subject": subject},
569
+ )
570
+ adapter_spec: AdapterSpec = _get_generation_adapter_spec(
571
+ instructions="Only give numerical or boolean answer without an explanation."
572
+ )
573
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
574
+
575
+ run_spec_name: str = "unicorn"
576
+ return RunSpec(
577
+ name=run_spec_name,
578
+ scenario_spec=scenario_spec,
579
+ adapter_spec=adapter_spec,
580
+ metric_specs=metric_specs,
581
+ groups=[run_spec_name],
582
+ )
583
+
584
+
585
+ @run_spec_function("bingo")
586
+ def get_bingo_spec(subject: str) -> RunSpec:
587
+ scenario_spec = ScenarioSpec(
588
+ class_name="helm.benchmark.scenarios.vision_language.bingo_scenario.BingoScenario", args={"subject": subject}
589
+ )
590
+ adapter_spec: AdapterSpec = _get_short_answer_generation_adapter_spec()
591
+ metric_specs: List[MetricSpec] = _get_open_ended_generation_metric_specs()
592
+
593
+ run_spec_name: str = "bingo"
594
+ return RunSpec(
595
+ name=run_spec_name,
596
+ scenario_spec=scenario_spec,
597
+ adapter_spec=adapter_spec,
598
+ metric_specs=metric_specs,
599
+ groups=[run_spec_name],
600
+ )
601
+
602
+
603
+ @run_spec_function("multipanelvqa")
604
+ def get_multipanelvqa_spec(subject: str, question_type: str) -> RunSpec:
605
+ scenario_spec = ScenarioSpec(
606
+ class_name="helm.benchmark.scenarios.vision_language.multipanelvqa_scenario.MultipanelVQAScenario",
607
+ args={"subject": subject, "question_type": question_type},
608
+ )
609
+
610
+ adapter_spec: AdapterSpec
611
+ if question_type == "open":
612
+ adapter_spec = _get_short_answer_generation_adapter_spec()
613
+ elif question_type == "multiple-choice":
614
+ adapter_spec = _get_multiple_choice_joint_adapter_spec(
615
+ input_noun=None, output_noun="Answer", max_train_instances=0
616
+ )
617
+ else:
618
+ raise ValueError(f"Invalid question type: {question_type}")
619
+
620
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
621
+ run_spec_name: str = "multipanelvqa"
622
+ return RunSpec(
623
+ name=f"{run_spec_name}:subject={subject},question_type={question_type}",
624
+ scenario_spec=scenario_spec,
625
+ adapter_spec=adapter_spec,
626
+ metric_specs=metric_specs,
627
+ groups=[run_spec_name],
628
+ )
629
+
630
+
631
+ @run_spec_function("pope")
632
+ def get_pope_spec() -> RunSpec:
633
+ scenario_spec = ScenarioSpec(
634
+ class_name="helm.benchmark.scenarios.vision_language.pope_scenario.POPEScenario",
635
+ )
636
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
637
+ input_noun=None, output_noun="Answer", max_train_instances=0
638
+ )
639
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
640
+
641
+ run_spec_name: str = "pope"
642
+ return RunSpec(
643
+ name=run_spec_name,
644
+ scenario_spec=scenario_spec,
645
+ adapter_spec=adapter_spec,
646
+ metric_specs=metric_specs,
647
+ groups=[run_spec_name],
648
+ )
649
+
650
+
651
+ @run_spec_function("seed_bench")
652
+ def get_seed_bench_spec(subject: str) -> RunSpec:
653
+ scenario_spec = ScenarioSpec(
654
+ class_name="helm.benchmark.scenarios.vision_language.seed_bench_scenario.SEEDBenchScenario",
655
+ args={"subject": subject},
656
+ )
657
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
658
+ input_noun=None, output_noun="Answer", max_train_instances=0
659
+ )
660
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
661
+
662
+ run_spec_name: str = "seed_bench"
663
+ return RunSpec(
664
+ name=run_spec_name,
665
+ scenario_spec=scenario_spec,
666
+ adapter_spec=adapter_spec,
667
+ metric_specs=metric_specs,
668
+ groups=[run_spec_name],
669
+ )
670
+
671
+
672
+ @run_spec_function("mme")
673
+ def get_mme_spec(subject: str) -> RunSpec:
674
+ scenario_spec = ScenarioSpec(
675
+ class_name="helm.benchmark.scenarios.vision_language.mme_scenario.MMEScenario",
676
+ args={"subject": subject},
677
+ )
678
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
679
+ input_noun=None, output_noun="Answer", max_train_instances=0
680
+ )
681
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
682
+
683
+ run_spec_name: str = "mme"
684
+ return RunSpec(
685
+ name=run_spec_name,
686
+ scenario_spec=scenario_spec,
687
+ adapter_spec=adapter_spec,
688
+ metric_specs=metric_specs,
689
+ groups=[run_spec_name],
690
+ )
691
+
692
+
693
+ @run_spec_function("heim_human_eval")
694
+ def get_heim_human_eval_spec(question_type: str) -> RunSpec:
695
+ scenario_spec = ScenarioSpec(
696
+ class_name="helm.benchmark.scenarios.vision_language.heim_human_eval_scenario.HEIMHumanEvalScenario",
697
+ args={"question_type": question_type},
698
+ )
699
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
700
+ input_noun=None,
701
+ output_noun="Answer",
702
+ num_outputs=1,
703
+ max_train_instances=0,
704
+ )
705
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
706
+
707
+ run_spec_name: str = "heim_human_eval"
708
+ return RunSpec(
709
+ name=f"{run_spec_name}:question_type={question_type}",
710
+ scenario_spec=scenario_spec,
711
+ adapter_spec=adapter_spec,
712
+ metric_specs=metric_specs,
713
+ groups=[run_spec_name],
714
+ )
715
+
716
+
717
+ @run_spec_function("pairs")
718
+ def get_pairs_spec(subset: str, person: str) -> RunSpec:
719
+ scenario_spec = ScenarioSpec(
720
+ class_name="helm.benchmark.scenarios.vision_language.pairs_scenario.PAIRSScenario",
721
+ args={"subset": subset, "person": person},
722
+ )
723
+ adapter_spec: AdapterSpec = _get_multiple_choice_joint_adapter_spec(
724
+ input_noun=None,
725
+ output_noun="Answer",
726
+ num_outputs=1,
727
+ max_train_instances=0,
728
+ )
729
+ metric_specs: List[MetricSpec] = get_exact_match_metric_specs()
730
+
731
+ run_spec_name: str = "pairs"
732
+ return RunSpec(
733
+ name=f"{run_spec_name}:subset={subset},person={person}",
734
+ scenario_spec=scenario_spec,
735
+ adapter_spec=adapter_spec,
736
+ metric_specs=metric_specs,
737
+ groups=[run_spec_name],
738
+ )
739
+
740
+
741
+ @run_spec_function("mementos")
742
+ def get_mementos_spec(subject: str) -> RunSpec:
743
+ scenario_spec = ScenarioSpec(
744
+ class_name="helm.benchmark.scenarios.vision_language.mementos_scenario.MementosScenario",
745
+ args={"subject": subject},
746
+ )
747
+ adapter_spec: AdapterSpec = _get_short_answer_generation_adapter_spec()
748
+ metric_specs: List[MetricSpec] = _get_open_ended_generation_metric_specs()
749
+
750
+ run_spec_name: str = "mementos"
751
+ return RunSpec(
752
+ name=run_spec_name,
753
+ scenario_spec=scenario_spec,
754
+ adapter_spec=adapter_spec,
755
+ metric_specs=metric_specs,
756
+ groups=[run_spec_name],
757
+ )