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.
- {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/METADATA +138 -31
- crfm_helm-0.5.1.dist-info/RECORD +654 -0
- {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/WHEEL +1 -1
- helm/benchmark/adaptation/adapter_spec.py +31 -3
- helm/benchmark/adaptation/adapters/adapter.py +2 -2
- helm/benchmark/adaptation/adapters/adapter_factory.py +24 -27
- helm/benchmark/adaptation/adapters/generation_adapter.py +1 -0
- helm/benchmark/adaptation/adapters/in_context_learning_adapter.py +20 -4
- helm/benchmark/adaptation/adapters/language_modeling_adapter.py +2 -3
- helm/benchmark/adaptation/adapters/multimodal/in_context_learning_multimodal_adapter.py +1 -0
- helm/benchmark/adaptation/adapters/multimodal/multimodal_prompt.py +7 -0
- helm/benchmark/adaptation/adapters/multimodal/multiple_choice_joint_multimodal_adapter.py +104 -0
- helm/benchmark/adaptation/adapters/multimodal/test_in_context_learning_multimodal_adapter.py +2 -1
- helm/benchmark/adaptation/adapters/multimodal/test_multimodal_prompt.py +2 -0
- helm/benchmark/adaptation/adapters/test_adapter.py +2 -1
- helm/benchmark/adaptation/adapters/test_generation_adapter.py +32 -8
- helm/benchmark/adaptation/adapters/test_language_modeling_adapter.py +7 -19
- helm/benchmark/adaptation/adapters/test_multiple_choice_joint_adapter.py +60 -6
- helm/benchmark/adaptation/common_adapter_specs.py +376 -0
- helm/benchmark/adaptation/request_state.py +6 -1
- helm/benchmark/adaptation/scenario_state.py +6 -2
- helm/benchmark/annotation/annotator.py +43 -0
- helm/benchmark/annotation/annotator_factory.py +61 -0
- helm/benchmark/annotation/image2structure/image_compiler_annotator.py +88 -0
- helm/benchmark/annotation/image2structure/latex_compiler_annotator.py +59 -0
- helm/benchmark/annotation/image2structure/lilypond_compiler_annotator.py +84 -0
- helm/benchmark/annotation/image2structure/webpage_compiler_annotator.py +132 -0
- helm/benchmark/annotation/test_annotator_factory.py +26 -0
- helm/benchmark/annotation/test_dummy_annotator.py +44 -0
- helm/benchmark/annotation_executor.py +124 -0
- helm/benchmark/augmentations/data_augmenter.py +0 -2
- helm/benchmark/augmentations/gender_perturbation.py +1 -1
- helm/benchmark/augmentations/perturbation.py +25 -3
- helm/benchmark/augmentations/perturbation_description.py +1 -1
- helm/benchmark/augmentations/suffix_perturbation.py +29 -0
- helm/benchmark/augmentations/test_perturbation.py +41 -7
- helm/benchmark/augmentations/translate_perturbation.py +30 -0
- helm/benchmark/config_registry.py +7 -1
- helm/benchmark/executor.py +46 -16
- helm/benchmark/huggingface_registration.py +20 -7
- helm/benchmark/metrics/basic_metrics.py +169 -664
- helm/benchmark/metrics/bbq_metrics.py +3 -4
- helm/benchmark/metrics/bias_metrics.py +6 -6
- helm/benchmark/metrics/classification_metrics.py +11 -8
- helm/benchmark/metrics/cleva_accuracy_metrics.py +8 -5
- helm/benchmark/metrics/cleva_harms_metrics.py +2 -2
- helm/benchmark/metrics/code_metrics_helper.py +0 -2
- helm/benchmark/metrics/common_metric_specs.py +167 -0
- helm/benchmark/metrics/decodingtrust_fairness_metrics.py +72 -0
- helm/benchmark/metrics/decodingtrust_ood_knowledge_metrics.py +66 -0
- helm/benchmark/metrics/decodingtrust_privacy_metrics.py +101 -0
- helm/benchmark/metrics/decodingtrust_stereotype_bias_metrics.py +202 -0
- helm/benchmark/metrics/disinformation_metrics.py +4 -110
- helm/benchmark/metrics/dry_run_metrics.py +2 -2
- helm/benchmark/metrics/efficiency_metrics.py +213 -0
- helm/benchmark/metrics/evaluate_instances_metric.py +59 -0
- helm/benchmark/metrics/evaluate_reference_metrics.py +392 -0
- helm/benchmark/metrics/image_generation/aesthetics_metrics.py +54 -0
- helm/benchmark/metrics/image_generation/aesthetics_scorer.py +66 -0
- helm/benchmark/metrics/image_generation/clip_score_metrics.py +73 -0
- helm/benchmark/metrics/image_generation/denoised_runtime_metric.py +42 -0
- helm/benchmark/metrics/image_generation/detection_metrics.py +57 -0
- helm/benchmark/metrics/image_generation/detectors/base_detector.py +8 -0
- helm/benchmark/metrics/image_generation/detectors/vitdet.py +178 -0
- helm/benchmark/metrics/image_generation/efficiency_metrics.py +41 -0
- helm/benchmark/metrics/image_generation/fidelity_metrics.py +168 -0
- helm/benchmark/metrics/image_generation/fractal_dimension/__init__.py +0 -0
- helm/benchmark/metrics/image_generation/fractal_dimension/fractal_dimension_util.py +63 -0
- helm/benchmark/metrics/image_generation/fractal_dimension/test_fractal_dimension_util.py +33 -0
- helm/benchmark/metrics/image_generation/fractal_dimension_metric.py +50 -0
- helm/benchmark/metrics/image_generation/gender_metrics.py +58 -0
- helm/benchmark/metrics/image_generation/image_critique_metrics.py +284 -0
- helm/benchmark/metrics/image_generation/lpips_metrics.py +82 -0
- helm/benchmark/metrics/image_generation/multi_scale_ssim_metrics.py +82 -0
- helm/benchmark/metrics/image_generation/nsfw_detector.py +96 -0
- helm/benchmark/metrics/image_generation/nsfw_metrics.py +103 -0
- helm/benchmark/metrics/image_generation/nudity_metrics.py +38 -0
- helm/benchmark/metrics/image_generation/photorealism_critique_metrics.py +153 -0
- helm/benchmark/metrics/image_generation/psnr_metrics.py +78 -0
- helm/benchmark/metrics/image_generation/q16/__init__.py +0 -0
- helm/benchmark/metrics/image_generation/q16/q16_toxicity_detector.py +90 -0
- helm/benchmark/metrics/image_generation/q16/test_q16.py +18 -0
- helm/benchmark/metrics/image_generation/q16_toxicity_metrics.py +48 -0
- helm/benchmark/metrics/image_generation/skin_tone_metrics.py +164 -0
- helm/benchmark/metrics/image_generation/uiqi_metrics.py +92 -0
- helm/benchmark/metrics/image_generation/watermark/__init__.py +0 -0
- helm/benchmark/metrics/image_generation/watermark/test_watermark_detector.py +16 -0
- helm/benchmark/metrics/image_generation/watermark/watermark_detector.py +87 -0
- helm/benchmark/metrics/image_generation/watermark_metrics.py +48 -0
- helm/benchmark/metrics/instruction_following_critique_metrics.py +3 -1
- helm/benchmark/metrics/language_modeling_metrics.py +99 -0
- helm/benchmark/metrics/machine_translation_metrics.py +89 -0
- helm/benchmark/metrics/metric.py +93 -172
- helm/benchmark/metrics/metric_name.py +0 -1
- helm/benchmark/metrics/metric_service.py +16 -0
- helm/benchmark/metrics/paraphrase_generation_metrics.py +3 -4
- helm/benchmark/metrics/ranking_metrics.py +2 -2
- helm/benchmark/metrics/reference_metric.py +148 -0
- helm/benchmark/metrics/summac/model_summac.py +0 -2
- helm/benchmark/metrics/summarization_metrics.py +2 -2
- helm/benchmark/metrics/test_classification_metrics.py +8 -5
- helm/benchmark/metrics/test_disinformation_metrics.py +78 -0
- helm/benchmark/metrics/{test_basic_metrics.py → test_evaluate_reference_metrics.py} +5 -1
- helm/benchmark/metrics/test_metric.py +2 -2
- helm/benchmark/metrics/tokens/gooseai_token_cost_estimator.py +10 -2
- helm/benchmark/metrics/toxicity_metrics.py +1 -1
- helm/benchmark/metrics/toxicity_utils.py +23 -0
- helm/benchmark/metrics/unitxt_metrics.py +81 -0
- helm/benchmark/metrics/vision_language/__init__.py +0 -0
- helm/benchmark/metrics/vision_language/emd_utils.py +341 -0
- helm/benchmark/metrics/vision_language/image_metrics.py +575 -0
- helm/benchmark/metrics/vision_language/image_utils.py +100 -0
- helm/benchmark/model_deployment_registry.py +74 -0
- helm/benchmark/model_metadata_registry.py +41 -1
- helm/benchmark/multi_gpu_runner.py +133 -0
- helm/benchmark/presentation/create_plots.py +8 -7
- helm/benchmark/presentation/run_display.py +26 -10
- helm/benchmark/presentation/schema.py +15 -40
- helm/benchmark/presentation/summarize.py +119 -79
- helm/benchmark/presentation/table.py +8 -8
- helm/benchmark/presentation/test_contamination.py +2 -2
- helm/benchmark/presentation/test_run_entry.py +1 -2
- helm/benchmark/presentation/test_summarize.py +3 -3
- helm/benchmark/run.py +54 -26
- helm/benchmark/run_expander.py +205 -35
- helm/benchmark/run_spec.py +93 -0
- helm/benchmark/run_spec_factory.py +163 -0
- helm/benchmark/run_specs/__init__.py +0 -0
- helm/benchmark/run_specs/classic_run_specs.py +1510 -0
- helm/benchmark/run_specs/cleva_run_specs.py +277 -0
- helm/benchmark/run_specs/decodingtrust_run_specs.py +314 -0
- helm/benchmark/run_specs/heim_run_specs.py +623 -0
- helm/benchmark/run_specs/instruction_following_run_specs.py +129 -0
- helm/benchmark/run_specs/lite_run_specs.py +307 -0
- helm/benchmark/run_specs/simple_run_specs.py +104 -0
- helm/benchmark/run_specs/unitxt_run_specs.py +42 -0
- helm/benchmark/run_specs/vlm_run_specs.py +757 -0
- helm/benchmark/runner.py +51 -57
- helm/benchmark/runner_config_registry.py +21 -0
- helm/benchmark/scenarios/bbq_scenario.py +1 -1
- helm/benchmark/scenarios/bold_scenario.py +2 -2
- helm/benchmark/scenarios/code_scenario.py +1 -0
- helm/benchmark/scenarios/decodingtrust_adv_demonstration_scenario.py +169 -0
- helm/benchmark/scenarios/decodingtrust_adv_robustness_scenario.py +121 -0
- helm/benchmark/scenarios/decodingtrust_fairness_scenario.py +77 -0
- helm/benchmark/scenarios/decodingtrust_machine_ethics_scenario.py +324 -0
- helm/benchmark/scenarios/decodingtrust_ood_robustness_scenario.py +204 -0
- helm/benchmark/scenarios/decodingtrust_privacy_scenario.py +559 -0
- helm/benchmark/scenarios/decodingtrust_stereotype_bias_scenario.py +67 -0
- helm/benchmark/scenarios/decodingtrust_toxicity_prompts_scenario.py +78 -0
- helm/benchmark/scenarios/dialogue_scenarios.py +0 -1
- helm/benchmark/scenarios/image_generation/__init__.py +0 -0
- helm/benchmark/scenarios/image_generation/common_syntactic_processes_scenario.py +105 -0
- helm/benchmark/scenarios/image_generation/cub200_scenario.py +95 -0
- helm/benchmark/scenarios/image_generation/daily_dalle_scenario.py +124 -0
- helm/benchmark/scenarios/image_generation/demographic_stereotypes_scenario.py +82 -0
- helm/benchmark/scenarios/image_generation/detection_scenario.py +83 -0
- helm/benchmark/scenarios/image_generation/draw_bench_scenario.py +74 -0
- helm/benchmark/scenarios/image_generation/i2p_scenario.py +57 -0
- helm/benchmark/scenarios/image_generation/landing_page_scenario.py +46 -0
- helm/benchmark/scenarios/image_generation/logos_scenario.py +223 -0
- helm/benchmark/scenarios/image_generation/magazine_cover_scenario.py +91 -0
- helm/benchmark/scenarios/image_generation/mental_disorders_scenario.py +46 -0
- helm/benchmark/scenarios/image_generation/mscoco_scenario.py +91 -0
- helm/benchmark/scenarios/image_generation/paint_skills_scenario.py +72 -0
- helm/benchmark/scenarios/image_generation/parti_prompts_scenario.py +94 -0
- helm/benchmark/scenarios/image_generation/radiology_scenario.py +42 -0
- helm/benchmark/scenarios/image_generation/relational_understanding_scenario.py +52 -0
- helm/benchmark/scenarios/image_generation/time_most_significant_historical_figures_scenario.py +124 -0
- helm/benchmark/scenarios/image_generation/winoground_scenario.py +62 -0
- helm/benchmark/scenarios/imdb_scenario.py +0 -1
- helm/benchmark/scenarios/legalbench_scenario.py +6 -2
- helm/benchmark/scenarios/live_qa_scenario.py +94 -0
- helm/benchmark/scenarios/lm_entry_scenario.py +185 -0
- helm/benchmark/scenarios/math_scenario.py +19 -2
- helm/benchmark/scenarios/medication_qa_scenario.py +60 -0
- helm/benchmark/scenarios/numeracy_scenario.py +1 -1
- helm/benchmark/scenarios/opinions_qa_scenario.py +0 -4
- helm/benchmark/scenarios/scenario.py +4 -0
- helm/benchmark/scenarios/simple_scenarios.py +122 -1
- helm/benchmark/scenarios/test_math_scenario.py +6 -0
- helm/benchmark/scenarios/test_scenario.py +6 -3
- helm/benchmark/scenarios/test_simple_scenarios.py +50 -0
- helm/benchmark/scenarios/thai_exam_scenario.py +135 -0
- helm/benchmark/scenarios/unitxt_scenario.py +56 -0
- helm/benchmark/scenarios/verifiability_judgment_scenario.py +3 -1
- helm/benchmark/scenarios/vicuna_scenario.py +1 -1
- helm/benchmark/scenarios/vision_language/a_okvqa_scenario.py +83 -0
- helm/benchmark/scenarios/vision_language/bingo_scenario.py +103 -0
- helm/benchmark/scenarios/vision_language/crossmodal_3600_scenario.py +134 -0
- helm/benchmark/scenarios/vision_language/flickr30k_scenario.py +74 -0
- helm/benchmark/scenarios/vision_language/gqa_scenario.py +91 -0
- helm/benchmark/scenarios/vision_language/hateful_memes_scenario.py +94 -0
- helm/benchmark/scenarios/vision_language/heim_human_eval_scenario.py +113 -0
- helm/benchmark/scenarios/vision_language/image2structure/__init__.py +0 -0
- helm/benchmark/scenarios/vision_language/image2structure/chart2csv_scenario.py +55 -0
- helm/benchmark/scenarios/vision_language/image2structure/image2structure_scenario.py +214 -0
- helm/benchmark/scenarios/vision_language/image2structure/latex_scenario.py +25 -0
- helm/benchmark/scenarios/vision_language/image2structure/musicsheet_scenario.py +20 -0
- helm/benchmark/scenarios/vision_language/image2structure/utils_latex.py +347 -0
- helm/benchmark/scenarios/vision_language/image2structure/webpage/__init__.py +0 -0
- helm/benchmark/scenarios/vision_language/image2structure/webpage/driver.py +84 -0
- helm/benchmark/scenarios/vision_language/image2structure/webpage/jekyll_server.py +182 -0
- helm/benchmark/scenarios/vision_language/image2structure/webpage/utils.py +31 -0
- helm/benchmark/scenarios/vision_language/image2structure/webpage_scenario.py +225 -0
- helm/benchmark/scenarios/vision_language/math_vista_scenario.py +117 -0
- helm/benchmark/scenarios/vision_language/mementos_scenario.py +124 -0
- helm/benchmark/scenarios/vision_language/mm_safety_bench_scenario.py +103 -0
- helm/benchmark/scenarios/vision_language/mme_scenario.py +145 -0
- helm/benchmark/scenarios/vision_language/mmmu_scenario.py +187 -0
- helm/benchmark/scenarios/vision_language/mscoco_captioning_scenario.py +92 -0
- helm/benchmark/scenarios/vision_language/mscoco_categorization_scenario.py +117 -0
- helm/benchmark/scenarios/vision_language/multipanelvqa_scenario.py +169 -0
- helm/benchmark/scenarios/vision_language/originality_scenario.py +35 -0
- helm/benchmark/scenarios/vision_language/pairs_scenario.py +246 -0
- helm/benchmark/scenarios/vision_language/pope_scenario.py +104 -0
- helm/benchmark/scenarios/vision_language/seed_bench_scenario.py +129 -0
- helm/benchmark/scenarios/vision_language/unicorn_scenario.py +108 -0
- helm/benchmark/scenarios/vision_language/viz_wiz_scenario.py +3 -4
- helm/benchmark/scenarios/vision_language/vqa_scenario.py +5 -3
- helm/benchmark/scenarios/wmt_14_scenario.py +1 -1
- helm/benchmark/server.py +24 -1
- helm/benchmark/slurm_runner.py +70 -49
- helm/benchmark/static/benchmarking.js +1 -1
- helm/benchmark/static/schema_classic.yaml +258 -1066
- helm/benchmark/static/schema_image2structure.yaml +304 -0
- helm/benchmark/static/schema_instruction_following.yaml +210 -0
- helm/benchmark/static/schema_lite.yaml +2 -227
- helm/benchmark/static/schema_mmlu.yaml +1507 -0
- helm/benchmark/static/schema_unitxt.yaml +428 -0
- helm/benchmark/static/schema_vhelm_lite.yaml +164 -0
- helm/benchmark/static/schema_vlm.yaml +823 -0
- helm/benchmark/static_build/assets/01-694cb9b7.png +0 -0
- helm/benchmark/static_build/assets/ai21-0eb91ec3.png +0 -0
- helm/benchmark/static_build/assets/aleph-alpha-7ce10034.png +0 -0
- helm/benchmark/static_build/assets/anthropic-70d8bc39.png +0 -0
- helm/benchmark/static_build/assets/bigscience-7f0400c0.png +0 -0
- helm/benchmark/static_build/assets/cohere-3550c6cb.png +0 -0
- helm/benchmark/static_build/assets/crfm-logo-74391ab8.png +0 -0
- helm/benchmark/static_build/assets/eleutherai-b9451114.png +0 -0
- helm/benchmark/static_build/assets/google-06d997ad.png +0 -0
- helm/benchmark/static_build/assets/heim-logo-3e5e3aa4.png +0 -0
- helm/benchmark/static_build/assets/helm-logo-simple-2ed5400b.png +0 -0
- helm/benchmark/static_build/assets/helmhero-28e90f4d.png +0 -0
- helm/benchmark/static_build/assets/index-737eef9e.js +10 -0
- helm/benchmark/static_build/assets/index-878a1094.css +1 -0
- helm/benchmark/static_build/assets/meta-5580e9f1.png +0 -0
- helm/benchmark/static_build/assets/microsoft-f5ee5016.png +0 -0
- helm/benchmark/static_build/assets/mistral-18e1be23.png +0 -0
- helm/benchmark/static_build/assets/nvidia-86fa75c1.png +0 -0
- helm/benchmark/static_build/assets/openai-3f8653e4.png +0 -0
- helm/benchmark/static_build/assets/react-d4a0b69b.js +85 -0
- helm/benchmark/static_build/assets/recharts-6d337683.js +97 -0
- helm/benchmark/static_build/assets/tii-24de195c.png +0 -0
- helm/benchmark/static_build/assets/together-a665a35b.png +0 -0
- helm/benchmark/static_build/assets/tremor-54a99cc4.js +10 -0
- helm/benchmark/static_build/assets/tsinghua-keg-97d4b395.png +0 -0
- helm/benchmark/static_build/assets/vhelm-framework-cde7618a.png +0 -0
- helm/benchmark/static_build/assets/vhelm-model-6d812526.png +0 -0
- helm/benchmark/static_build/assets/yandex-38e09d70.png +0 -0
- helm/benchmark/static_build/config.js +4 -0
- helm/benchmark/static_build/index.html +20 -0
- helm/benchmark/test_data_preprocessor.py +3 -3
- helm/benchmark/test_run_expander.py +1 -1
- helm/benchmark/window_services/ai21_window_service.py +22 -33
- helm/benchmark/window_services/cohere_window_service.py +1 -63
- helm/benchmark/window_services/default_window_service.py +2 -44
- helm/benchmark/window_services/encoder_decoder_window_service.py +0 -11
- helm/benchmark/window_services/ice_window_service.py +0 -34
- helm/benchmark/window_services/image_generation/__init__.py +0 -0
- helm/benchmark/window_services/image_generation/clip_window_service.py +15 -0
- helm/benchmark/window_services/image_generation/lexica_search_window_service.py +9 -0
- helm/benchmark/window_services/image_generation/openai_dalle_window_service.py +9 -0
- helm/benchmark/window_services/image_generation/test_clip_window_service.py +29 -0
- helm/benchmark/window_services/image_generation/test_openai_dalle_window_service.py +30 -0
- helm/benchmark/window_services/local_window_service.py +21 -4
- helm/benchmark/window_services/test_anthropic_window_service.py +2 -1
- helm/benchmark/window_services/test_bloom_window_service.py +2 -1
- helm/benchmark/window_services/test_cohere_window_service.py +2 -1
- helm/benchmark/window_services/test_flan_t5_window_service.py +2 -1
- helm/benchmark/window_services/test_gpt2_window_service.py +2 -2
- helm/benchmark/window_services/test_gpt4_window_service.py +2 -1
- helm/benchmark/window_services/test_gptj_window_service.py +3 -2
- helm/benchmark/window_services/test_gptneox_window_service.py +3 -2
- helm/benchmark/window_services/test_ice_window_service.py +2 -1
- helm/benchmark/window_services/test_openai_window_service.py +2 -1
- helm/benchmark/window_services/test_opt_window_service.py +3 -2
- helm/benchmark/window_services/test_palmyra_window_service.py +2 -1
- helm/benchmark/window_services/test_t0pp_window_service.py +2 -1
- helm/benchmark/window_services/test_t511b_window_service.py +2 -1
- helm/benchmark/window_services/test_ul2_window_service.py +2 -1
- helm/benchmark/window_services/test_utils.py +3 -2
- helm/benchmark/window_services/test_yalm_window_service.py +2 -1
- helm/benchmark/window_services/window_service.py +42 -0
- helm/benchmark/window_services/window_service_factory.py +4 -1
- helm/benchmark/window_services/yalm_window_service.py +0 -27
- helm/clients/__init__.py +0 -0
- helm/{proxy/clients → clients}/ai21_client.py +3 -9
- helm/clients/aleph_alpha_client.py +112 -0
- helm/{proxy/clients → clients}/anthropic_client.py +233 -18
- helm/{proxy/clients → clients}/auto_client.py +59 -31
- helm/clients/bedrock_client.py +128 -0
- helm/clients/bedrock_utils.py +72 -0
- helm/{proxy/clients → clients}/client.py +65 -7
- helm/clients/clip_score_client.py +49 -0
- helm/clients/clip_scorers/__init__.py +0 -0
- helm/clients/clip_scorers/base_clip_scorer.py +18 -0
- helm/clients/clip_scorers/clip_scorer.py +50 -0
- helm/clients/clip_scorers/multilingual_clip_scorer.py +50 -0
- helm/{proxy/clients → clients}/cohere_client.py +4 -11
- helm/clients/gcs_client.py +82 -0
- helm/{proxy/clients → clients}/google_client.py +5 -5
- helm/clients/google_translate_client.py +35 -0
- helm/{proxy/clients → clients}/http_model_client.py +5 -7
- helm/{proxy/clients → clients}/huggingface_client.py +43 -64
- helm/clients/image_generation/__init__.py +0 -0
- helm/clients/image_generation/adobe_vision_client.py +78 -0
- helm/clients/image_generation/aleph_alpha_image_generation_client.py +98 -0
- helm/clients/image_generation/cogview2/__init__.py +0 -0
- helm/clients/image_generation/cogview2/coglm_strategy.py +96 -0
- helm/clients/image_generation/cogview2/coglm_utils.py +82 -0
- helm/clients/image_generation/cogview2/sr_pipeline/__init__.py +15 -0
- helm/clients/image_generation/cogview2/sr_pipeline/direct_sr.py +96 -0
- helm/clients/image_generation/cogview2/sr_pipeline/dsr_model.py +254 -0
- helm/clients/image_generation/cogview2/sr_pipeline/dsr_sampling.py +190 -0
- helm/clients/image_generation/cogview2/sr_pipeline/iterative_sr.py +141 -0
- helm/clients/image_generation/cogview2/sr_pipeline/itersr_model.py +269 -0
- helm/clients/image_generation/cogview2/sr_pipeline/itersr_sampling.py +120 -0
- helm/clients/image_generation/cogview2/sr_pipeline/sr_group.py +42 -0
- helm/clients/image_generation/cogview2_client.py +191 -0
- helm/clients/image_generation/dalle2_client.py +192 -0
- helm/clients/image_generation/dalle3_client.py +108 -0
- helm/clients/image_generation/dalle_mini/__init__.py +3 -0
- helm/clients/image_generation/dalle_mini/data.py +442 -0
- helm/clients/image_generation/dalle_mini/model/__init__.py +5 -0
- helm/clients/image_generation/dalle_mini/model/configuration.py +175 -0
- helm/clients/image_generation/dalle_mini/model/modeling.py +1834 -0
- helm/clients/image_generation/dalle_mini/model/partitions.py +84 -0
- helm/clients/image_generation/dalle_mini/model/processor.py +63 -0
- helm/clients/image_generation/dalle_mini/model/text.py +251 -0
- helm/clients/image_generation/dalle_mini/model/tokenizer.py +9 -0
- helm/clients/image_generation/dalle_mini/model/utils.py +29 -0
- helm/clients/image_generation/dalle_mini/vqgan_jax/__init__.py +1 -0
- helm/clients/image_generation/dalle_mini/vqgan_jax/configuration_vqgan.py +40 -0
- helm/clients/image_generation/dalle_mini/vqgan_jax/convert_pt_model_to_jax.py +107 -0
- helm/clients/image_generation/dalle_mini/vqgan_jax/modeling_flax_vqgan.py +610 -0
- helm/clients/image_generation/dalle_mini_client.py +190 -0
- helm/clients/image_generation/deep_floyd_client.py +78 -0
- helm/clients/image_generation/huggingface_diffusers_client.py +249 -0
- helm/clients/image_generation/image_generation_client_utils.py +9 -0
- helm/clients/image_generation/lexica_client.py +86 -0
- helm/clients/image_generation/mindalle/__init__.py +0 -0
- helm/clients/image_generation/mindalle/models/__init__.py +216 -0
- helm/clients/image_generation/mindalle/models/stage1/__init__.py +0 -0
- helm/clients/image_generation/mindalle/models/stage1/layers.py +312 -0
- helm/clients/image_generation/mindalle/models/stage1/vqgan.py +103 -0
- helm/clients/image_generation/mindalle/models/stage2/__init__.py +0 -0
- helm/clients/image_generation/mindalle/models/stage2/layers.py +144 -0
- helm/clients/image_generation/mindalle/models/stage2/transformer.py +268 -0
- helm/clients/image_generation/mindalle/models/tokenizer.py +30 -0
- helm/clients/image_generation/mindalle/utils/__init__.py +3 -0
- helm/clients/image_generation/mindalle/utils/config.py +129 -0
- helm/clients/image_generation/mindalle/utils/sampling.py +149 -0
- helm/clients/image_generation/mindalle/utils/utils.py +89 -0
- helm/clients/image_generation/mindalle_client.py +115 -0
- helm/clients/image_generation/nudity_check_client.py +64 -0
- helm/clients/image_generation/together_image_generation_client.py +111 -0
- helm/{proxy/clients → clients}/lit_gpt_client.py +4 -4
- helm/{proxy/clients → clients}/megatron_client.py +5 -5
- helm/clients/mistral_client.py +134 -0
- helm/clients/moderation_api_client.py +109 -0
- helm/clients/open_lm_client.py +43 -0
- helm/clients/openai_client.py +301 -0
- helm/{proxy/clients → clients}/palmyra_client.py +6 -8
- helm/{proxy/clients → clients}/perspective_api_client.py +7 -8
- helm/clients/simple_client.py +64 -0
- helm/{proxy/clients → clients}/test_auto_client.py +13 -15
- helm/clients/test_client.py +100 -0
- helm/{proxy/clients → clients}/test_huggingface_client.py +15 -16
- helm/clients/test_simple_client.py +19 -0
- helm/{proxy/clients → clients}/test_together_client.py +20 -8
- helm/{proxy/clients → clients}/together_client.py +104 -73
- helm/clients/vertexai_client.py +400 -0
- helm/clients/vision_language/__init__.py +0 -0
- helm/clients/vision_language/huggingface_vision2seq_client.py +145 -0
- helm/clients/vision_language/huggingface_vlm_client.py +111 -0
- helm/{proxy/clients → clients}/vision_language/idefics_client.py +54 -49
- helm/clients/vision_language/open_flamingo/__init__.py +2 -0
- helm/clients/vision_language/open_flamingo/src/__init__.py +0 -0
- helm/clients/vision_language/open_flamingo/src/factory.py +147 -0
- helm/clients/vision_language/open_flamingo/src/flamingo.py +337 -0
- helm/clients/vision_language/open_flamingo/src/flamingo_lm.py +155 -0
- helm/clients/vision_language/open_flamingo/src/helpers.py +267 -0
- helm/clients/vision_language/open_flamingo/src/utils.py +47 -0
- helm/clients/vision_language/open_flamingo_client.py +155 -0
- helm/clients/vision_language/qwen_vlm_client.py +171 -0
- helm/clients/vllm_client.py +46 -0
- helm/common/cache.py +16 -4
- helm/common/cache_backend_config.py +47 -0
- helm/common/clip_score_request.py +41 -0
- helm/common/file_caches/__init__.py +0 -0
- helm/common/file_caches/file_cache.py +16 -0
- helm/common/file_caches/local_file_cache.py +61 -0
- helm/common/file_caches/test_local_file_cache.py +25 -0
- helm/common/file_upload_request.py +27 -0
- helm/common/general.py +1 -1
- helm/common/image_generation_parameters.py +25 -0
- helm/common/images_utils.py +33 -3
- helm/common/key_value_store.py +35 -4
- helm/common/media_object.py +13 -0
- helm/common/moderations_api_request.py +71 -0
- helm/common/mongo_key_value_store.py +3 -3
- helm/common/multimodal_request_utils.py +31 -0
- helm/common/nudity_check_request.py +29 -0
- helm/common/request.py +15 -17
- helm/common/test_general.py +6 -0
- helm/common/tokenization_request.py +1 -1
- helm/config/model_deployments.yaml +1159 -538
- helm/config/model_metadata.yaml +868 -41
- helm/config/tokenizer_configs.yaml +149 -43
- helm/proxy/accounts.py +31 -4
- helm/proxy/critique/mechanical_turk_critique_importer.py +3 -0
- helm/proxy/critique/model_critique_client.py +8 -6
- helm/proxy/example_queries.py +29 -17
- helm/proxy/server.py +70 -5
- helm/proxy/services/remote_service.py +31 -0
- helm/proxy/services/server_service.py +96 -16
- helm/proxy/services/service.py +30 -0
- helm/proxy/services/test_remote_service.py +4 -3
- helm/proxy/services/test_service.py +0 -12
- helm/proxy/test_accounts.py +32 -0
- helm/proxy/token_counters/auto_token_counter.py +37 -37
- helm/proxy/token_counters/test_auto_token_counter.py +164 -0
- helm/proxy/token_counters/token_counter.py +3 -5
- helm/tokenizers/__init__.py +0 -0
- helm/{proxy/tokenizers → tokenizers}/ai21_tokenizer.py +3 -3
- helm/{proxy/tokenizers → tokenizers}/anthropic_tokenizer.py +1 -1
- helm/{proxy/tokenizers → tokenizers}/auto_tokenizer.py +6 -9
- helm/{proxy/tokenizers → tokenizers}/cohere_tokenizer.py +1 -1
- helm/{proxy/tokenizers → tokenizers}/http_model_tokenizer.py +3 -3
- helm/{proxy/tokenizers → tokenizers}/huggingface_tokenizer.py +7 -26
- helm/tokenizers/simple_tokenizer.py +33 -0
- helm/{proxy/tokenizers → tokenizers}/test_anthropic_tokenizer.py +1 -1
- helm/{proxy/tokenizers → tokenizers}/test_huggingface_tokenizer.py +3 -0
- helm/tokenizers/test_simple_tokenizer.py +33 -0
- helm/{proxy/tokenizers → tokenizers}/vertexai_tokenizer.py +1 -1
- helm/{proxy/tokenizers → tokenizers}/yalm_tokenizer.py +5 -3
- helm/tokenizers/yalm_tokenizer_data/__init__.py +0 -0
- helm/tokenizers/yalm_tokenizer_data/voc_100b.sp +0 -0
- helm/{proxy/tokenizers → tokenizers}/yalm_tokenizer_data/yalm_tokenizer.py +1 -1
- crfm_helm-0.4.0.dist-info/RECORD +0 -397
- helm/benchmark/run_specs.py +0 -2762
- helm/benchmark/test_model_deployment_definition.py +0 -92
- helm/benchmark/test_model_properties.py +0 -1570
- helm/benchmark/vlm_run_specs.py +0 -97
- helm/benchmark/window_services/flan_t5_window_service.py +0 -29
- helm/benchmark/window_services/gpt2_window_service.py +0 -32
- helm/benchmark/window_services/huggingface_window_service.py +0 -60
- helm/benchmark/window_services/t0pp_window_service.py +0 -35
- helm/benchmark/window_services/t511b_window_service.py +0 -30
- helm/benchmark/window_services/test_mt_nlg_window_service.py +0 -48
- helm/benchmark/window_services/ul2_window_service.py +0 -30
- helm/benchmark/window_services/wider_ai21_window_service.py +0 -24
- helm/common/cache_utils.py +0 -14
- helm/proxy/clients/aleph_alpha_client.py +0 -95
- helm/proxy/clients/goose_ai_client.py +0 -99
- helm/proxy/clients/microsoft_client.py +0 -180
- helm/proxy/clients/openai_client.py +0 -206
- helm/proxy/clients/simple_client.py +0 -60
- helm/proxy/clients/test_client.py +0 -49
- helm/proxy/clients/vertexai_client.py +0 -115
- helm/proxy/token_counters/ai21_token_counter.py +0 -20
- helm/proxy/token_counters/cohere_token_counter.py +0 -13
- helm/proxy/token_counters/free_token_counter.py +0 -12
- helm/proxy/token_counters/gooseai_token_counter.py +0 -24
- helm/proxy/token_counters/openai_token_counter.py +0 -22
- helm/proxy/token_counters/test_ai21_token_counter.py +0 -88
- helm/proxy/token_counters/test_openai_token_counter.py +0 -81
- helm/proxy/tokenizers/simple_tokenizer.py +0 -32
- {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/LICENSE +0 -0
- {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/entry_points.txt +0 -0
- {crfm_helm-0.4.0.dist-info → crfm_helm-0.5.1.dist-info}/top_level.txt +0 -0
- /helm/{proxy/clients → benchmark/annotation}/__init__.py +0 -0
- /helm/{proxy/clients/vision_language → benchmark/annotation/image2structure}/__init__.py +0 -0
- /helm/{proxy/tokenizers → benchmark/metrics/image_generation}/__init__.py +0 -0
- /helm/{proxy/tokenizers/yalm_tokenizer_data → benchmark/metrics/image_generation/detectors}/__init__.py +0 -0
- /helm/{proxy/clients → clients}/ai21_utils.py +0 -0
- /helm/{proxy/clients → clients}/cohere_utils.py +0 -0
- /helm/{proxy/clients → clients}/lit_gpt_generate.py +0 -0
- /helm/{proxy/clients → clients}/toxicity_classifier_client.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/aleph_alpha_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/caching_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/ice_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/lit_gpt_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/test_ice_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/test_yalm_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/tiktoken_tokenizer.py +0 -0
- /helm/{proxy/tokenizers → tokenizers}/tokenizer.py +0 -0
- /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
|
+
)
|