crfm-helm 0.5.4__py3-none-any.whl → 0.5.5__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.5.5.dist-info/METADATA +413 -0
- crfm_helm-0.5.5.dist-info/RECORD +894 -0
- {crfm_helm-0.5.4.dist-info → crfm_helm-0.5.5.dist-info}/WHEEL +1 -1
- helm/benchmark/adaptation/adapter_spec.py +13 -1
- helm/benchmark/adaptation/adapters/adapter_factory.py +15 -1
- helm/benchmark/adaptation/adapters/binary_ranking_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/chat_adapter.py +49 -0
- helm/benchmark/adaptation/adapters/ehr_instruction_adapter.py +108 -0
- helm/benchmark/adaptation/adapters/generation_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/in_context_learning_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/language_modeling_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/multimodal/generation_multimodal_adapter.py +4 -2
- helm/benchmark/adaptation/adapters/multimodal/in_context_learning_multimodal_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/multimodal/multiple_choice_joint_multimodal_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/multimodal/test_in_context_learning_multimodal_adapter.py +4 -2
- helm/benchmark/adaptation/adapters/multimodal/test_multimodal_prompt.py +1 -1
- helm/benchmark/adaptation/adapters/multiple_choice_calibrated_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/multiple_choice_joint_adapter.py +2 -2
- helm/benchmark/adaptation/adapters/multiple_choice_joint_chain_of_thought_adapter.py +87 -0
- helm/benchmark/adaptation/adapters/multiple_choice_separate_adapter.py +1 -1
- helm/benchmark/adaptation/adapters/test_generation_adapter.py +3 -3
- helm/benchmark/adaptation/adapters/test_language_modeling_adapter.py +2 -2
- helm/benchmark/adaptation/adapters/test_multiple_choice_joint_adapter.py +2 -2
- helm/benchmark/adaptation/common_adapter_specs.py +69 -4
- helm/benchmark/adaptation/prompt.py +1 -1
- helm/benchmark/annotation/aci_bench_annotator.py +95 -0
- helm/benchmark/annotation/air_bench_annotator.py +20 -5
- helm/benchmark/annotation/annotator.py +5 -0
- helm/benchmark/annotation/annotator_factory.py +3 -20
- helm/benchmark/annotation/autobencher_capabilities_annotator.py +107 -0
- helm/benchmark/annotation/autobencher_safety_annotator.py +98 -0
- helm/benchmark/annotation/bigcodebench_annotator.py +108 -0
- helm/benchmark/annotation/bird_sql_annotator.py +58 -0
- helm/benchmark/annotation/chw_care_plan_annotator.py +98 -0
- helm/benchmark/annotation/czech_bank_qa_annotator.py +78 -0
- helm/benchmark/annotation/dischargeme_annotator.py +107 -0
- helm/benchmark/annotation/ehr_sql_annotator.py +87 -0
- helm/benchmark/annotation/helpdesk_call_summarization_annotator.py +131 -0
- helm/benchmark/annotation/image2struct/image_compiler_annotator.py +6 -1
- helm/benchmark/annotation/live_qa_annotator.py +1 -1
- helm/benchmark/annotation/med_dialog_annotator.py +99 -0
- helm/benchmark/annotation/medalign_annotator.py +100 -0
- helm/benchmark/annotation/medi_qa_annotator.py +98 -0
- helm/benchmark/annotation/medication_qa_annotator.py +87 -63
- helm/benchmark/annotation/mental_health_annotator.py +98 -0
- helm/benchmark/annotation/mimic_rrs_annotator.py +100 -0
- helm/benchmark/annotation/model_as_judge.py +218 -6
- helm/benchmark/annotation/mtsamples_procedures_annotator.py +98 -0
- helm/benchmark/annotation/mtsamples_replicate_annotator.py +101 -0
- helm/benchmark/annotation/omni_math/gpt_evaluation_template.txt +152 -0
- helm/benchmark/annotation/omni_math/gpt_evaluation_zero_shot_template.txt +36 -0
- helm/benchmark/annotation/omni_math_annotator.py +132 -0
- helm/benchmark/annotation/spider_annotator.py +18 -0
- helm/benchmark/annotation/starr_patient_instructions_annotator.py +98 -0
- helm/benchmark/annotation/wildbench/eval_template.pairwise.v2.md +75 -0
- helm/benchmark/annotation/wildbench/eval_template.score.v2.md +66 -0
- helm/benchmark/annotation/wildbench_annotator.py +119 -0
- helm/benchmark/annotation_executor.py +35 -15
- helm/benchmark/augmentations/cleva_perturbation.py +9 -8
- helm/benchmark/augmentations/contraction_expansion_perturbation.py +2 -2
- helm/benchmark/augmentations/contrast_sets_perturbation.py +2 -2
- helm/benchmark/augmentations/dialect_perturbation.py +4 -5
- helm/benchmark/augmentations/extra_space_perturbation.py +2 -2
- helm/benchmark/augmentations/filler_words_perturbation.py +2 -2
- helm/benchmark/augmentations/gender_perturbation.py +2 -2
- helm/benchmark/augmentations/lowercase_perturbation.py +2 -2
- helm/benchmark/augmentations/mild_mix_perturbation.py +6 -6
- helm/benchmark/augmentations/misspelling_perturbation.py +2 -2
- helm/benchmark/augmentations/person_name_perturbation.py +4 -5
- helm/benchmark/augmentations/perturbation.py +1 -1
- helm/benchmark/augmentations/space_perturbation.py +2 -2
- helm/benchmark/augmentations/suffix_perturbation.py +2 -2
- helm/benchmark/augmentations/synonym_perturbation.py +4 -3
- helm/benchmark/augmentations/test_perturbation.py +16 -13
- helm/benchmark/augmentations/translate_perturbation.py +2 -2
- helm/benchmark/augmentations/typos_perturbation.py +2 -2
- helm/benchmark/data_preprocessor.py +2 -2
- helm/benchmark/huggingface_registration.py +2 -7
- helm/benchmark/metrics/aci_bench_metrics.py +34 -0
- helm/benchmark/metrics/basic_metrics.py +6 -6
- helm/benchmark/metrics/bbq_metrics.py +2 -2
- helm/benchmark/metrics/bias_metrics.py +12 -3
- helm/benchmark/metrics/bigcodebench_metrics.py +25 -0
- helm/benchmark/metrics/bird_sql_metrics.py +28 -0
- helm/benchmark/metrics/chw_care_plan_metrics.py +34 -0
- helm/benchmark/metrics/classification_metrics.py +76 -12
- helm/benchmark/metrics/cleva_harms_metrics.py +8 -7
- helm/benchmark/metrics/code_metrics.py +5 -5
- helm/benchmark/metrics/comet_metric.py +125 -0
- helm/benchmark/metrics/common_metric_specs.py +9 -2
- helm/benchmark/metrics/conv_fin_qa_calc_metrics.py +72 -0
- helm/benchmark/metrics/copyright_metrics.py +4 -4
- helm/benchmark/metrics/czech_bank_qa_metrics.py +29 -0
- helm/benchmark/metrics/decodingtrust_fairness_metrics.py +2 -2
- helm/benchmark/metrics/decodingtrust_privacy_metrics.py +2 -2
- helm/benchmark/metrics/decodingtrust_stereotype_bias_metrics.py +2 -2
- helm/benchmark/metrics/dischargeme_metrics.py +34 -0
- helm/benchmark/metrics/disinformation_metrics.py +4 -4
- helm/benchmark/metrics/dry_run_metrics.py +5 -5
- helm/benchmark/metrics/efficiency_metrics.py +3 -3
- helm/benchmark/metrics/ehr_sql_metrics.py +103 -0
- helm/benchmark/metrics/evaluate_instances_metric.py +3 -3
- helm/benchmark/metrics/evaluate_reference_metrics.py +144 -16
- helm/benchmark/metrics/gpqa_chain_of_thought_metric.py +103 -0
- helm/benchmark/metrics/gpt4_audio_critique_metrics.py +167 -0
- helm/benchmark/metrics/helpdesk_call_summarization_metrics.py +36 -0
- helm/benchmark/metrics/ifeval/__init__.py +0 -0
- helm/benchmark/metrics/ifeval/instructions.py +1574 -0
- helm/benchmark/metrics/ifeval/instructions_registry.py +182 -0
- helm/benchmark/metrics/ifeval/instructions_registry.pyi +3 -0
- helm/benchmark/metrics/ifeval/instructions_util.py +153 -0
- helm/benchmark/metrics/ifeval_metrics.py +55 -0
- helm/benchmark/metrics/image_generation/aesthetics_metrics.py +1 -1
- helm/benchmark/metrics/image_generation/detection_metrics.py +1 -1
- helm/benchmark/metrics/image_generation/detectors/vitdet.py +1 -1
- helm/benchmark/metrics/image_generation/fractal_dimension/test_fractal_dimension_util.py +1 -1
- helm/benchmark/metrics/image_generation/fractal_dimension_metric.py +1 -1
- helm/benchmark/metrics/image_generation/nsfw_metrics.py +1 -1
- helm/benchmark/metrics/image_generation/q16/test_q16.py +3 -1
- helm/benchmark/metrics/image_generation/q16_toxicity_metrics.py +1 -1
- helm/benchmark/metrics/image_generation/skin_tone_metrics.py +2 -2
- helm/benchmark/metrics/image_generation/watermark/test_watermark_detector.py +1 -1
- helm/benchmark/metrics/image_generation/watermark_metrics.py +1 -1
- helm/benchmark/metrics/instruction_following_critique_metrics.py +4 -4
- helm/benchmark/metrics/language_modeling_metrics.py +4 -4
- helm/benchmark/metrics/machine_translation_metrics.py +2 -2
- helm/benchmark/metrics/med_dialog_metrics.py +34 -0
- helm/benchmark/metrics/medalign_metrics.py +34 -0
- helm/benchmark/metrics/medcalc_bench_metrics.py +124 -0
- helm/benchmark/metrics/medec_metrics.py +101 -0
- helm/benchmark/metrics/medi_qa_metrics.py +34 -0
- helm/benchmark/metrics/medication_qa_metrics.py +15 -4
- helm/benchmark/metrics/mental_health_metrics.py +34 -0
- helm/benchmark/metrics/metric.py +3 -3
- helm/benchmark/metrics/mimic_rrs_metrics.py +34 -0
- helm/benchmark/metrics/mimiciv_billing_code_metrics.py +96 -0
- helm/benchmark/metrics/mtsamples_procedures_metrics.py +34 -0
- helm/benchmark/metrics/mtsamples_replicate_metrics.py +34 -0
- helm/benchmark/metrics/nltk_helper.py +32 -0
- helm/benchmark/metrics/numeracy_metrics.py +4 -4
- helm/benchmark/metrics/omni_math_metrics.py +32 -0
- helm/benchmark/metrics/output_processing_metric.py +60 -0
- helm/benchmark/metrics/output_processors.py +15 -0
- helm/benchmark/metrics/paraphrase_generation_metrics.py +2 -2
- helm/benchmark/metrics/ranking_metrics.py +3 -3
- helm/benchmark/metrics/reference_metric.py +3 -3
- helm/benchmark/metrics/{bhasa_metrics.py → seahelm_metrics.py} +3 -3
- helm/benchmark/metrics/seahelm_metrics_specs.py +10 -0
- helm/benchmark/metrics/spider_metrics.py +7 -0
- helm/benchmark/metrics/starr_patient_instructions_metrics.py +34 -0
- helm/benchmark/metrics/statistic.py +1 -1
- helm/benchmark/metrics/summac/model_summac.py +1 -1
- helm/benchmark/metrics/summarization_critique_metrics.py +4 -4
- helm/benchmark/metrics/summarization_metrics.py +19 -9
- helm/benchmark/metrics/test_bias_metrics.py +5 -1
- helm/benchmark/metrics/test_classification_metrics.py +140 -68
- helm/benchmark/metrics/test_evaluate_reference_metrics.py +15 -0
- helm/benchmark/metrics/test_metric.py +1 -1
- helm/benchmark/metrics/test_statistic.py +2 -2
- helm/benchmark/metrics/tokens/ai21_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/tokens/auto_token_cost_estimator.py +6 -6
- helm/benchmark/metrics/tokens/cohere_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/tokens/free_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/tokens/gooseai_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/tokens/openai_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/tokens/test_ai21_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/tokens/test_openai_token_cost_estimator.py +1 -1
- helm/benchmark/metrics/toxicity_metrics.py +4 -4
- helm/benchmark/metrics/unitxt_metrics.py +4 -1
- helm/benchmark/metrics/vision_language/image_metrics.py +1 -1
- helm/benchmark/metrics/wildbench_metrics.py +34 -0
- helm/benchmark/model_metadata_registry.py +16 -0
- helm/benchmark/presentation/summarize.py +23 -10
- helm/benchmark/presentation/torr_robustness_summarizer.py +178 -0
- helm/benchmark/reeval_run.py +203 -0
- helm/benchmark/reeval_runner.py +355 -0
- helm/benchmark/run.py +8 -17
- helm/benchmark/run_expander.py +78 -8
- helm/benchmark/run_spec_factory.py +12 -0
- helm/benchmark/run_specs/air_bench_run_specs.py +21 -3
- helm/benchmark/run_specs/audio_run_specs.py +613 -0
- helm/benchmark/run_specs/call_center_run_specs.py +49 -0
- helm/benchmark/run_specs/capabilities_run_specs.py +308 -0
- helm/benchmark/run_specs/classic_run_specs.py +1 -69
- helm/benchmark/run_specs/enem_challenge_specs.py +31 -0
- helm/benchmark/run_specs/enterprise_run_specs.py +260 -0
- helm/benchmark/run_specs/experimental_run_specs.py +112 -3
- helm/benchmark/run_specs/imdb_ptbr_run_specs.py +30 -0
- helm/benchmark/run_specs/lite_run_specs.py +2 -2
- helm/benchmark/run_specs/long_context_run_specs.py +89 -0
- helm/benchmark/run_specs/medhelm_run_specs.py +1155 -0
- helm/benchmark/run_specs/mmlu_clinical_afr_run_specs.py +49 -0
- helm/benchmark/run_specs/oab_exams_specs.py +32 -0
- helm/benchmark/run_specs/safety_run_specs.py +37 -0
- helm/benchmark/run_specs/{bhasa_run_specs.py → seahelm_run_specs.py} +44 -44
- helm/benchmark/run_specs/sql_run_specs.py +54 -0
- helm/benchmark/run_specs/tweetsentbr_run_specs.py +32 -0
- helm/benchmark/run_specs/unitxt_run_specs.py +14 -5
- helm/benchmark/run_specs/vlm_run_specs.py +75 -2
- helm/benchmark/run_specs/winogrande_afr_run_specs.py +47 -0
- helm/benchmark/scenarios/aci_bench_scenario.py +120 -0
- helm/benchmark/scenarios/air_bench_scenario.py +6 -1
- helm/benchmark/scenarios/anthropic_hh_rlhf_scenario.py +5 -3
- helm/benchmark/scenarios/anthropic_red_team_scenario.py +1 -1
- helm/benchmark/scenarios/audio_language/__init__.py +0 -0
- helm/benchmark/scenarios/audio_language/air_bench_chat_scenario.py +128 -0
- helm/benchmark/scenarios/audio_language/air_bench_foundation_scenario.py +154 -0
- helm/benchmark/scenarios/audio_language/ami_scenario.py +96 -0
- helm/benchmark/scenarios/audio_language/audio_mnist_scenario.py +62 -0
- helm/benchmark/scenarios/audio_language/audio_pairs_scenario.py +62 -0
- helm/benchmark/scenarios/audio_language/audiocaps_scenario.py +59 -0
- helm/benchmark/scenarios/audio_language/casual_conversations2_scenario.py +152 -0
- helm/benchmark/scenarios/audio_language/common_voice_15_scenario.py +99 -0
- helm/benchmark/scenarios/audio_language/covost2_scenario.py +163 -0
- helm/benchmark/scenarios/audio_language/fleurs_fairness_scenario.py +83 -0
- helm/benchmark/scenarios/audio_language/fleurs_scenario.py +312 -0
- helm/benchmark/scenarios/audio_language/iemocap_audio_scenario.py +83 -0
- helm/benchmark/scenarios/audio_language/librispeech_fairness_scenario.py +96 -0
- helm/benchmark/scenarios/audio_language/librispeech_scenario.py +80 -0
- helm/benchmark/scenarios/audio_language/meld_audio_scenario.py +113 -0
- helm/benchmark/scenarios/audio_language/multilingual_librispeech_scenario.py +80 -0
- helm/benchmark/scenarios/audio_language/mustard_scenario.py +142 -0
- helm/benchmark/scenarios/audio_language/mutox_scenario.py +254 -0
- helm/benchmark/scenarios/audio_language/parade_scenario.py +97 -0
- helm/benchmark/scenarios/audio_language/speech_robust_bench_scenario.py +124 -0
- helm/benchmark/scenarios/audio_language/vocal_sound_scenario.py +69 -0
- helm/benchmark/scenarios/audio_language/voice_jailbreak_attacks_scenario.py +87 -0
- helm/benchmark/scenarios/audio_language/voxceleb2_scenario.py +106 -0
- helm/benchmark/scenarios/autobencher_capabilities_scenario.py +68 -0
- helm/benchmark/scenarios/autobencher_safety_scenario.py +51 -0
- helm/benchmark/scenarios/babi_qa_scenario.py +1 -1
- helm/benchmark/scenarios/banking77_scenario.py +6 -1
- helm/benchmark/scenarios/bbq_scenario.py +1 -1
- helm/benchmark/scenarios/big_bench_scenario.py +11 -1
- helm/benchmark/scenarios/bigcodebench_scenario.py +58 -0
- helm/benchmark/scenarios/bird_sql_scenario.py +94 -0
- helm/benchmark/scenarios/bird_sql_scenario_helper.py +118 -0
- helm/benchmark/scenarios/blimp_scenario.py +1 -1
- helm/benchmark/scenarios/bold_scenario.py +1 -1
- helm/benchmark/scenarios/boolq_scenario.py +1 -1
- helm/benchmark/scenarios/casehold_scenario.py +79 -0
- helm/benchmark/scenarios/chw_care_plan_scenario.py +105 -0
- helm/benchmark/scenarios/civil_comments_scenario.py +1 -1
- helm/benchmark/scenarios/clear_scenario.py +153 -0
- helm/benchmark/scenarios/cleva_scenario.py +2 -2
- helm/benchmark/scenarios/code_scenario.py +17 -4
- helm/benchmark/scenarios/commonsense_scenario.py +1 -1
- helm/benchmark/scenarios/conv_fin_qa_calc_scenario.py +97 -0
- helm/benchmark/scenarios/copyright_scenario.py +1 -1
- helm/benchmark/scenarios/covid_dialog_scenario.py +10 -1
- helm/benchmark/scenarios/cti_to_mitre_scenario.py +240 -0
- helm/benchmark/scenarios/custom_mcqa_scenario.py +1 -1
- helm/benchmark/scenarios/czech_bank_qa_scenario.py +130 -0
- helm/benchmark/scenarios/decodingtrust_adv_demonstration_scenario.py +1 -1
- helm/benchmark/scenarios/decodingtrust_privacy_scenario.py +1 -1
- helm/benchmark/scenarios/decodingtrust_stereotype_bias_scenario.py +1 -1
- helm/benchmark/scenarios/decodingtrust_toxicity_prompts_scenario.py +1 -1
- helm/benchmark/scenarios/dialogue_scenarios.py +13 -2
- helm/benchmark/scenarios/dischargeme_scenario.py +157 -0
- helm/benchmark/scenarios/disinformation_scenario.py +10 -1
- helm/benchmark/scenarios/dyck_language_scenario.py +10 -1
- helm/benchmark/scenarios/echr_judgment_classification_scenario.py +113 -0
- helm/benchmark/scenarios/ehr_sql_scenario.py +131 -0
- helm/benchmark/scenarios/ehrshot_scenario.py +1546 -0
- helm/benchmark/scenarios/enem_challenge_scenario.py +58 -0
- helm/benchmark/scenarios/entity_data_imputation_scenario.py +11 -1
- helm/benchmark/scenarios/entity_matching_scenario.py +12 -2
- helm/benchmark/scenarios/financial_phrasebank_scenario.py +94 -0
- helm/benchmark/scenarios/gold_commodity_news_scenario.py +124 -0
- helm/benchmark/scenarios/gpqa_scenario.py +80 -0
- helm/benchmark/scenarios/grammar_scenario.py +2 -2
- helm/benchmark/scenarios/gsm_scenario.py +10 -1
- helm/benchmark/scenarios/harm_bench_gcg_transfer_scenario.py +50 -0
- helm/benchmark/scenarios/harm_bench_scenario.py +1 -1
- helm/benchmark/scenarios/headqa_scenario.py +131 -0
- helm/benchmark/scenarios/helpdesk_call_summarization_scenario.py +37 -0
- helm/benchmark/scenarios/ice_scenario.py +8 -4
- helm/benchmark/scenarios/ifeval_scenario.py +53 -0
- helm/benchmark/scenarios/imdb_ptbr_scenario.py +60 -0
- helm/benchmark/scenarios/imdb_scenario.py +11 -2
- helm/benchmark/scenarios/infinite_bench_sum_scenario.py +82 -0
- helm/benchmark/scenarios/interactive_qa_mmlu_scenario.py +2 -2
- helm/benchmark/scenarios/koala_scenario.py +1 -1
- helm/benchmark/scenarios/legal_contract_summarization_scenario.py +129 -0
- helm/benchmark/scenarios/legal_opinion_sentiment_classification_scenario.py +77 -0
- helm/benchmark/scenarios/legal_summarization_scenario.py +11 -1
- helm/benchmark/scenarios/legal_support_scenario.py +11 -1
- helm/benchmark/scenarios/legalbench_scenario.py +22 -3
- helm/benchmark/scenarios/lex_glue_scenario.py +12 -2
- helm/benchmark/scenarios/lextreme_scenario.py +11 -1
- helm/benchmark/scenarios/live_qa_scenario.py +1 -1
- helm/benchmark/scenarios/lm_entry_scenario.py +1 -1
- helm/benchmark/scenarios/lsat_qa_scenario.py +1 -1
- helm/benchmark/scenarios/math_scenario.py +9 -1
- helm/benchmark/scenarios/me_q_sum_scenario.py +10 -1
- helm/benchmark/scenarios/med_dialog_scenario.py +22 -24
- helm/benchmark/scenarios/med_mcqa_scenario.py +10 -1
- helm/benchmark/scenarios/med_paragraph_simplification_scenario.py +10 -1
- helm/benchmark/scenarios/med_qa_scenario.py +10 -1
- helm/benchmark/scenarios/medalign_scenario.py +88 -0
- helm/benchmark/scenarios/medalign_scenario_helper.py +429 -0
- helm/benchmark/scenarios/medbullets_scenario.py +140 -0
- helm/benchmark/scenarios/medcalc_bench_scenario.py +125 -0
- helm/benchmark/scenarios/medec_scenario.py +120 -0
- helm/benchmark/scenarios/medhallu_scenario.py +66 -0
- helm/benchmark/scenarios/medi_qa_scenario.py +105 -0
- helm/benchmark/scenarios/medication_qa_scenario.py +2 -2
- helm/benchmark/scenarios/mental_health_scenario.py +112 -0
- helm/benchmark/scenarios/mimic_bhc_scenario.py +98 -0
- helm/benchmark/scenarios/mimic_rrs_scenario.py +89 -0
- helm/benchmark/scenarios/mimiciv_billing_code_scenario.py +71 -0
- helm/benchmark/scenarios/mmlu_clinical_afr_scenario.py +74 -0
- helm/benchmark/scenarios/mmlu_pro_scenario.py +95 -0
- helm/benchmark/scenarios/mmlu_scenario.py +11 -1
- helm/benchmark/scenarios/msmarco_scenario.py +1 -1
- helm/benchmark/scenarios/mtsamples_procedures_scenario.py +141 -0
- helm/benchmark/scenarios/mtsamples_replicate_scenario.py +141 -0
- helm/benchmark/scenarios/n2c2_ct_matching_scenario.py +271 -0
- helm/benchmark/scenarios/narrativeqa_scenario.py +1 -1
- helm/benchmark/scenarios/natural_qa_scenario.py +1 -1
- helm/benchmark/scenarios/newsqa_scenario.py +1 -1
- helm/benchmark/scenarios/numeracy_scenario.py +10 -1
- helm/benchmark/scenarios/oab_exams_scenario.py +57 -0
- helm/benchmark/scenarios/omni_math_scenario.py +53 -0
- helm/benchmark/scenarios/open_assistant_scenario.py +11 -2
- helm/benchmark/scenarios/opinions_qa_scenario.py +1 -1
- helm/benchmark/scenarios/pubmed_qa_scenario.py +54 -43
- helm/benchmark/scenarios/quac_scenario.py +10 -1
- helm/benchmark/scenarios/race_based_med_scenario.py +142 -0
- helm/benchmark/scenarios/raft_scenario.py +17 -2
- helm/benchmark/scenarios/real_toxicity_prompts_scenario.py +1 -1
- helm/benchmark/scenarios/ruler_qa_scenario_helper.py +171 -0
- helm/benchmark/scenarios/ruler_qa_scenarios.py +88 -0
- helm/benchmark/scenarios/scenario.py +9 -1
- helm/benchmark/scenarios/{bhasa_scenario.py → seahelm_scenario.py} +7 -2
- helm/benchmark/scenarios/self_instruct_scenario.py +1 -1
- helm/benchmark/scenarios/shc_bmt_scenario.py +69 -0
- helm/benchmark/scenarios/shc_cdi_scenario.py +70 -0
- helm/benchmark/scenarios/shc_conf_scenario.py +70 -0
- helm/benchmark/scenarios/shc_ent_scenario.py +72 -0
- helm/benchmark/scenarios/shc_gip_scenario.py +66 -0
- helm/benchmark/scenarios/shc_ptbm_scenario.py +76 -0
- helm/benchmark/scenarios/shc_sei_scenario.py +89 -0
- helm/benchmark/scenarios/shc_sequoia_scenario.py +69 -0
- helm/benchmark/scenarios/simple_safety_tests_scenario.py +1 -1
- helm/benchmark/scenarios/spider_scenario.py +91 -0
- helm/benchmark/scenarios/starr_patient_instructions_scenario.py +90 -0
- helm/benchmark/scenarios/summarization_scenario.py +11 -1
- helm/benchmark/scenarios/sumosum_scenario.py +157 -0
- helm/benchmark/scenarios/synthetic_efficiency_scenario.py +1 -1
- helm/benchmark/scenarios/synthetic_reasoning_natural_scenario.py +11 -1
- helm/benchmark/scenarios/synthetic_reasoning_scenario.py +11 -1
- helm/benchmark/scenarios/test_bigcodebench_scenario.py +26 -0
- helm/benchmark/scenarios/test_czech_bank_qa_scenario.py +18 -0
- helm/benchmark/scenarios/test_enem_challenge_scenario.py +53 -0
- helm/benchmark/scenarios/test_ewok_scenario.py +6 -2
- helm/benchmark/scenarios/test_gold_commodity_news_scenario.py +18 -0
- helm/benchmark/scenarios/test_gpqa_scenario.py +44 -0
- helm/benchmark/scenarios/test_ifeval_scenario.py +36 -0
- helm/benchmark/scenarios/test_imdb_ptbr_scenario.py +27 -0
- helm/benchmark/scenarios/test_infinite_bench_sum_scenario.py +46 -0
- helm/benchmark/scenarios/test_math_scenario.py +1 -0
- helm/benchmark/scenarios/test_mmlu_clinical_afr_scenario.py +21 -0
- helm/benchmark/scenarios/test_mmlu_pro_scenario.py +53 -0
- helm/benchmark/scenarios/test_oab_exams_scenario.py +51 -0
- helm/benchmark/scenarios/test_omni_math_scenario.py +27 -0
- helm/benchmark/scenarios/test_tweetsentbr_scenario.py +24 -0
- helm/benchmark/scenarios/test_wildbench_scenario.py +15 -0
- helm/benchmark/scenarios/test_winogrande_afr_scenario.py +19 -0
- helm/benchmark/scenarios/thai_exam_scenario.py +10 -1
- helm/benchmark/scenarios/the_pile_scenario.py +1 -1
- helm/benchmark/scenarios/truthful_qa_scenario.py +10 -1
- helm/benchmark/scenarios/tweetsentbr_scenario.py +66 -0
- helm/benchmark/scenarios/twitter_aae_scenario.py +1 -1
- helm/benchmark/scenarios/unitxt_scenario.py +8 -2
- helm/benchmark/scenarios/verifiability_judgment_scenario.py +1 -1
- helm/benchmark/scenarios/vicuna_scenario.py +1 -1
- helm/benchmark/scenarios/vision_language/blink_scenario.py +140 -0
- helm/benchmark/scenarios/vision_language/mm_star_scenario.py +95 -0
- helm/benchmark/scenarios/vision_language/vqa_rad_scenario.py +88 -0
- helm/benchmark/scenarios/wikifact_scenario.py +11 -1
- helm/benchmark/scenarios/wikitext_103_scenario.py +1 -1
- helm/benchmark/scenarios/wildbench_scenario.py +83 -0
- helm/benchmark/scenarios/winogrande_afr_scenario.py +78 -0
- helm/benchmark/scenarios/wmt_14_scenario.py +14 -2
- helm/benchmark/scenarios/xstest_scenario.py +1 -1
- helm/benchmark/server.py +11 -0
- helm/benchmark/slurm_runner.py +1 -1
- helm/benchmark/static/schema_audio.yaml +752 -0
- helm/benchmark/static/schema_autobencher.yaml +150 -0
- helm/benchmark/static/schema_call_center.yaml +97 -60
- helm/benchmark/static/schema_capabilities.yaml +254 -0
- helm/benchmark/static/schema_czech_bank.yaml +148 -0
- helm/benchmark/static/schema_enem_challenge.yaml +146 -0
- helm/benchmark/static/schema_enterprise.yaml +298 -0
- helm/benchmark/static/schema_finance.yaml +14 -12
- helm/benchmark/static/schema_heim.yaml +1389 -0
- helm/benchmark/static/{schema_medical.yaml → schema_long_context.yaml} +67 -82
- helm/benchmark/static/schema_medhelm.yaml +1081 -0
- helm/benchmark/static/schema_mmlu_winogrande_afr.yaml +1045 -0
- helm/benchmark/static/schema_safety.yaml +18 -1
- helm/benchmark/static/{schema_bhasa.yaml → schema_seahelm.yaml} +30 -16
- helm/benchmark/static/schema_social_audio.yaml +224 -0
- helm/benchmark/static/schema_sql.yaml +171 -0
- helm/benchmark/static/{schema_tables.yaml → schema_torr.yaml} +169 -36
- helm/benchmark/static/schema_tweetsentbr.yaml +146 -0
- helm/benchmark/static/schema_vhelm.yaml +109 -36
- helm/benchmark/static_build/assets/helm-safety-2907a7b6.png +0 -0
- helm/benchmark/static_build/assets/index-262903c1.js +10 -0
- helm/benchmark/static_build/assets/index-42060d71.css +1 -0
- helm/benchmark/static_build/assets/medhelm-overview-3ddfcd65.png +0 -0
- helm/benchmark/static_build/assets/{react-d4a0b69b.js → react-f82877fd.js} +1 -1
- helm/benchmark/static_build/assets/{recharts-6d337683.js → recharts-4037aff0.js} +1 -1
- helm/benchmark/static_build/assets/{tremor-54a99cc4.js → tremor-9cefc3c5.js} +1 -1
- helm/benchmark/static_build/config.js +1 -1
- helm/benchmark/static_build/index.html +5 -5
- helm/benchmark/window_services/default_window_service.py +1 -1
- helm/benchmark/window_services/encoder_decoder_window_service.py +1 -1
- helm/benchmark/window_services/ice_window_service.py +1 -1
- helm/benchmark/window_services/image_generation/lexica_search_window_service.py +1 -1
- helm/benchmark/window_services/image_generation/openai_dalle_window_service.py +1 -1
- helm/benchmark/window_services/local_window_service.py +2 -2
- helm/benchmark/window_services/test_anthropic_window_service.py +3 -3
- helm/benchmark/window_services/test_bloom_window_service.py +3 -3
- helm/benchmark/window_services/test_gpt2_window_service.py +7 -2
- helm/benchmark/window_services/test_gpt4_window_service.py +8 -3
- helm/benchmark/window_services/test_gptj_window_service.py +8 -3
- helm/benchmark/window_services/test_gptneox_window_service.py +3 -3
- helm/benchmark/window_services/test_openai_window_service.py +8 -3
- helm/benchmark/window_services/test_opt_window_service.py +3 -3
- helm/benchmark/window_services/test_palmyra_window_service.py +3 -3
- helm/benchmark/window_services/test_t0pp_window_service.py +3 -3
- helm/benchmark/window_services/test_t511b_window_service.py +3 -3
- helm/benchmark/window_services/test_ul2_window_service.py +3 -3
- helm/benchmark/window_services/test_utils.py +1 -1
- helm/benchmark/window_services/test_yalm_window_service.py +3 -3
- helm/benchmark/window_services/yalm_window_service.py +1 -1
- helm/clients/ai21_client.py +3 -3
- helm/clients/aleph_alpha_client.py +1 -1
- helm/clients/audio_language/__init__.py +0 -0
- helm/clients/audio_language/diva_llama_client.py +118 -0
- helm/clients/audio_language/llama_omni_client.py +198 -0
- helm/clients/audio_language/qwen2_audiolm_client.py +188 -0
- helm/clients/audio_language/qwen_audiolm_client.py +150 -0
- helm/clients/auto_client.py +4 -2
- helm/clients/azure_openai_client.py +55 -0
- helm/clients/bedrock_client.py +201 -7
- helm/clients/bedrock_utils.py +33 -0
- helm/clients/clip_scorers/clip_scorer.py +1 -1
- helm/clients/clip_scorers/multilingual_clip_scorer.py +1 -1
- helm/clients/cohere_client.py +3 -3
- helm/clients/google_client.py +1 -1
- helm/clients/http_model_client.py +1 -1
- helm/clients/huggingface_client.py +10 -18
- helm/clients/ibm_client.py +267 -0
- helm/clients/image_generation/adobe_vision_client.py +1 -1
- helm/clients/image_generation/aleph_alpha_image_generation_client.py +1 -1
- helm/clients/image_generation/cogview2/sr_pipeline/__init__.py +3 -3
- helm/clients/image_generation/cogview2/sr_pipeline/direct_sr.py +5 -2
- helm/clients/image_generation/cogview2/sr_pipeline/iterative_sr.py +5 -2
- helm/clients/image_generation/cogview2/sr_pipeline/sr_group.py +2 -2
- helm/clients/image_generation/cogview2_client.py +1 -1
- helm/clients/image_generation/dalle2_client.py +1 -1
- helm/clients/image_generation/dalle3_client.py +2 -2
- helm/clients/image_generation/dalle_mini/__init__.py +1 -1
- helm/clients/image_generation/dalle_mini/data.py +1 -1
- helm/clients/image_generation/dalle_mini/model/__init__.py +5 -5
- helm/clients/image_generation/dalle_mini/model/configuration.py +1 -1
- helm/clients/image_generation/dalle_mini/model/modeling.py +2 -2
- helm/clients/image_generation/dalle_mini/model/processor.py +4 -4
- helm/clients/image_generation/dalle_mini/model/tokenizer.py +1 -1
- helm/clients/image_generation/dalle_mini/vqgan_jax/__init__.py +1 -1
- helm/clients/image_generation/dalle_mini/vqgan_jax/convert_pt_model_to_jax.py +2 -2
- helm/clients/image_generation/dalle_mini/vqgan_jax/modeling_flax_vqgan.py +1 -1
- helm/clients/image_generation/dalle_mini_client.py +1 -1
- helm/clients/image_generation/deep_floyd_client.py +1 -1
- helm/clients/image_generation/huggingface_diffusers_client.py +1 -1
- helm/clients/image_generation/lexica_client.py +1 -1
- helm/clients/image_generation/mindalle/models/__init__.py +6 -6
- helm/clients/image_generation/mindalle/models/stage1/vqgan.py +1 -1
- helm/clients/image_generation/mindalle/models/stage2/transformer.py +1 -1
- helm/clients/image_generation/mindalle/utils/__init__.py +3 -3
- helm/clients/image_generation/mindalle_client.py +1 -1
- helm/clients/image_generation/together_image_generation_client.py +1 -1
- helm/clients/lit_gpt_client.py +2 -2
- helm/clients/mistral_client.py +62 -18
- helm/clients/nvidia_nim_client.py +0 -3
- helm/clients/openai_client.py +241 -22
- helm/clients/palmyra_client.py +1 -4
- helm/clients/reka_client.py +1 -1
- helm/clients/stanfordhealthcare_azure_openai_client.py +58 -0
- helm/clients/stanfordhealthcare_claude_client.py +31 -0
- helm/clients/stanfordhealthcare_google_client.py +43 -0
- helm/clients/stanfordhealthcare_http_model_client.py +93 -0
- helm/clients/stanfordhealthcare_openai_client.py +62 -0
- helm/clients/stanfordhealthcare_shc_openai_client.py +42 -0
- helm/clients/test_client.py +1 -1
- helm/clients/test_together_client.py +6 -1
- helm/clients/together_client.py +47 -7
- helm/clients/upstage_client.py +23 -0
- helm/clients/vertexai_client.py +39 -13
- helm/clients/vision_language/open_flamingo/__init__.py +2 -2
- helm/clients/vision_language/open_flamingo/src/factory.py +3 -3
- helm/clients/vision_language/open_flamingo/src/flamingo.py +2 -2
- helm/clients/vision_language/open_flamingo/src/flamingo_lm.py +2 -2
- helm/clients/vision_language/qwen2_vlm_client.py +175 -0
- helm/clients/vllm_client.py +4 -6
- helm/clients/yi_client.py +0 -3
- helm/common/audio_utils.py +111 -0
- helm/common/file_caches/local_file_cache.py +1 -1
- helm/common/file_caches/test_local_file_cache.py +1 -1
- helm/common/images_utils.py +2 -2
- helm/common/media_object.py +2 -2
- helm/common/multimodal_request_utils.py +26 -0
- helm/common/reeval_parameters.py +12 -0
- helm/common/request.py +6 -2
- helm/common/response_format.py +18 -0
- helm/common/test_media_object.py +1 -1
- helm/config/model_deployments.yaml +1112 -19
- helm/config/model_metadata.yaml +985 -44
- helm/config/tokenizer_configs.yaml +379 -3
- helm/proxy/cli.py +2 -2
- helm/proxy/example_queries.py +1 -1
- helm/proxy/server.py +11 -4
- helm/proxy/services/remote_service.py +1 -1
- helm/proxy/services/server_service.py +1 -1
- helm/proxy/services/test_remote_service.py +2 -2
- helm/proxy/services/test_service.py +1 -1
- helm/proxy/static/general.js +122 -0
- helm/proxy/static/help.html +99 -0
- helm/proxy/static/index.css +57 -0
- helm/proxy/static/index.html +40 -0
- helm/proxy/static/index.js +456 -0
- helm/proxy/static/info-icon.png +0 -0
- helm/proxy/test_retry.py +1 -1
- helm/proxy/token_counters/auto_token_counter.py +1 -1
- helm/tokenizers/aleph_alpha_tokenizer.py +1 -1
- helm/tokenizers/caching_tokenizer.py +2 -30
- helm/tokenizers/http_model_tokenizer.py +1 -1
- helm/tokenizers/huggingface_tokenizer.py +2 -2
- helm/tokenizers/lit_gpt_tokenizer.py +1 -1
- helm/tokenizers/test_anthropic_tokenizer.py +6 -2
- helm/tokenizers/test_huggingface_tokenizer.py +1 -1
- helm/tokenizers/test_yalm_tokenizer.py +1 -1
- helm/tokenizers/tiktoken_tokenizer.py +1 -1
- helm/tokenizers/tokenizer.py +3 -1
- helm/tokenizers/yalm_tokenizer.py +3 -3
- helm/tokenizers/yalm_tokenizer_data/test_yalm_tokenizer.py +1 -1
- crfm_helm-0.5.4.dist-info/METADATA +0 -350
- crfm_helm-0.5.4.dist-info/RECORD +0 -697
- helm/benchmark/metrics/bhasa_metrics_specs.py +0 -10
- helm/benchmark/static_build/assets/01-694cb9b7.png +0 -0
- helm/benchmark/static_build/assets/accenture-6f97eeda.png +0 -0
- helm/benchmark/static_build/assets/ai21-0eb91ec3.png +0 -0
- helm/benchmark/static_build/assets/aisingapore-6dfc9acf.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/cresta-9e22b983.png +0 -0
- helm/benchmark/static_build/assets/cuhk-8c5631e9.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/index-05c76bb1.css +0 -1
- helm/benchmark/static_build/assets/index-3ee38b3d.js +0 -10
- 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/scb10x-204bd786.png +0 -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/tsinghua-keg-97d4b395.png +0 -0
- helm/benchmark/static_build/assets/wellsfargo-a86a6c4a.png +0 -0
- helm/benchmark/static_build/assets/yandex-38e09d70.png +0 -0
- helm/tokenizers/anthropic_tokenizer.py +0 -52
- {crfm_helm-0.5.4.dist-info → crfm_helm-0.5.5.dist-info}/entry_points.txt +0 -0
- {crfm_helm-0.5.4.dist-info → crfm_helm-0.5.5.dist-info/licenses}/LICENSE +0 -0
- {crfm_helm-0.5.4.dist-info → crfm_helm-0.5.5.dist-info}/top_level.txt +0 -0
|
@@ -86,7 +86,9 @@ tokenizer_configs:
|
|
|
86
86
|
# Anthropic
|
|
87
87
|
- name: anthropic/claude
|
|
88
88
|
tokenizer_spec:
|
|
89
|
-
class_name: "helm.tokenizers.
|
|
89
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
90
|
+
args:
|
|
91
|
+
pretrained_model_name_or_path: Xenova/claude-tokenizer
|
|
90
92
|
end_of_text_token: "<|endoftext|>"
|
|
91
93
|
prefix_token: "<|endoftext|>"
|
|
92
94
|
|
|
@@ -169,12 +171,25 @@ tokenizer_configs:
|
|
|
169
171
|
end_of_text_token: "<|end▁of▁sentence|>"
|
|
170
172
|
prefix_token: "<|begin▁of▁sentence|>"
|
|
171
173
|
|
|
174
|
+
- name: deepseek-ai/deepseek-v3
|
|
175
|
+
tokenizer_spec:
|
|
176
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
177
|
+
end_of_text_token: "<|end▁of▁sentence|>"
|
|
178
|
+
prefix_token: "<|begin▁of▁sentence|>"
|
|
179
|
+
|
|
180
|
+
- name: deepseek-ai/deepseek-r1
|
|
181
|
+
tokenizer_spec:
|
|
182
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
183
|
+
end_of_text_token: "<|end▁of▁sentence|>"
|
|
184
|
+
prefix_token: "<|begin▁of▁sentence|>"
|
|
185
|
+
|
|
172
186
|
# EleutherAI
|
|
173
187
|
- name: EleutherAI/gpt-j-6B
|
|
174
188
|
tokenizer_spec:
|
|
175
189
|
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
176
190
|
end_of_text_token: "<|endoftext|>"
|
|
177
191
|
prefix_token: "<|endoftext|>"
|
|
192
|
+
|
|
178
193
|
- name: EleutherAI/gpt-neox-20b
|
|
179
194
|
tokenizer_spec:
|
|
180
195
|
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
@@ -333,6 +348,14 @@ tokenizer_configs:
|
|
|
333
348
|
prefix_token: "<|begin_of_text|>"
|
|
334
349
|
end_of_text_token: "<|end_of_text|>"
|
|
335
350
|
|
|
351
|
+
- name: meta/llama-3.1-8b-instruct
|
|
352
|
+
tokenizer_spec:
|
|
353
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
354
|
+
args:
|
|
355
|
+
pretrained_model_name_or_path: meta-llama/Meta-Llama-3.1-8B-Instruct
|
|
356
|
+
prefix_token: "<|begin_of_text|>"
|
|
357
|
+
end_of_text_token: "<|eot_id|>"
|
|
358
|
+
|
|
336
359
|
- name: meta/llama-3.2-3b-instruct
|
|
337
360
|
tokenizer_spec:
|
|
338
361
|
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
@@ -340,6 +363,22 @@ tokenizer_configs:
|
|
|
340
363
|
pretrained_model_name_or_path: meta-llama/Llama-3.2-3B-Instruct
|
|
341
364
|
prefix_token: "<|begin_of_text|>"
|
|
342
365
|
end_of_text_token: "<|eot_id|>"
|
|
366
|
+
|
|
367
|
+
- name: meta/llama-3.2-1b-instruct
|
|
368
|
+
tokenizer_spec:
|
|
369
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
370
|
+
args:
|
|
371
|
+
pretrained_model_name_or_path: meta-llama/Llama-3.2-1B-Instruct
|
|
372
|
+
prefix_token: "<|begin_of_text|>"
|
|
373
|
+
end_of_text_token: "<|eot_id|>"
|
|
374
|
+
|
|
375
|
+
- name: meta/llama-3.1-8b-instruct
|
|
376
|
+
tokenizer_spec:
|
|
377
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
378
|
+
args:
|
|
379
|
+
pretrained_model_name_or_path: meta-llama/Llama-3.1-8B-Instruct
|
|
380
|
+
prefix_token: "<|begin_of_text|>"
|
|
381
|
+
end_of_text_token: "<|eot_id|>"
|
|
343
382
|
|
|
344
383
|
- name: meta/llama-3.2-11b-vision-instruct
|
|
345
384
|
tokenizer_spec:
|
|
@@ -349,6 +388,14 @@ tokenizer_configs:
|
|
|
349
388
|
prefix_token: "<|begin_of_text|>"
|
|
350
389
|
end_of_text_token: "<|eot_id|>"
|
|
351
390
|
|
|
391
|
+
- name: meta/llama-3.3-70b-instruct
|
|
392
|
+
tokenizer_spec:
|
|
393
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
394
|
+
args:
|
|
395
|
+
pretrained_model_name_or_path: meta-llama/Llama-3.3-70B-Instruct
|
|
396
|
+
prefix_token: "<|begin_of_text|>"
|
|
397
|
+
end_of_text_token: "<|eot_id|>"
|
|
398
|
+
|
|
352
399
|
# 01-ai
|
|
353
400
|
- name: 01-ai/Yi-6B
|
|
354
401
|
tokenizer_spec:
|
|
@@ -406,6 +453,18 @@ tokenizer_configs:
|
|
|
406
453
|
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
407
454
|
end_of_text_token: "<|endoftext|>"
|
|
408
455
|
prefix_token: "<s>"
|
|
456
|
+
|
|
457
|
+
- name: microsoft/phi-3.5-mini-instruct
|
|
458
|
+
tokenizer_spec:
|
|
459
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
460
|
+
end_of_text_token: "<|endoftext|>"
|
|
461
|
+
prefix_token: "<s>"
|
|
462
|
+
|
|
463
|
+
- name: microsoft/phi-3.5-mini-instruct
|
|
464
|
+
tokenizer_spec:
|
|
465
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
466
|
+
end_of_text_token: "<|endoftext|>"
|
|
467
|
+
prefix_token: "<s>"
|
|
409
468
|
|
|
410
469
|
# Mistralai
|
|
411
470
|
- name: mistralai/Mistral-7B-v0.1
|
|
@@ -444,6 +503,37 @@ tokenizer_configs:
|
|
|
444
503
|
end_of_text_token: "</s>"
|
|
445
504
|
prefix_token: "<s>"
|
|
446
505
|
|
|
506
|
+
- name: mistralai/Mistral-Large-Instruct-2411
|
|
507
|
+
tokenizer_spec:
|
|
508
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
509
|
+
end_of_text_token: "</s>"
|
|
510
|
+
prefix_token: "<s>"
|
|
511
|
+
|
|
512
|
+
- name: mistralai/Ministral-8B-Instruct-2410
|
|
513
|
+
tokenizer_spec:
|
|
514
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
515
|
+
end_of_text_token: "</s>"
|
|
516
|
+
prefix_token: "<s>"
|
|
517
|
+
|
|
518
|
+
- name: mistralai/Mistral-Small-24B-Instruct-2501
|
|
519
|
+
tokenizer_spec:
|
|
520
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
521
|
+
end_of_text_token: "</s>"
|
|
522
|
+
prefix_token: "<s>"
|
|
523
|
+
|
|
524
|
+
# Nectec
|
|
525
|
+
- name: nectec/OpenThaiLLM-Prebuilt-7B
|
|
526
|
+
tokenizer_spec:
|
|
527
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
528
|
+
end_of_text_token: "<|im_end|>"
|
|
529
|
+
prefix_token: ""
|
|
530
|
+
|
|
531
|
+
- name: nectec/Pathumma-llm-text-1.0.0
|
|
532
|
+
tokenizer_spec:
|
|
533
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
534
|
+
end_of_text_token: "<|im_end|>"
|
|
535
|
+
prefix_token: "<|im_start|>"
|
|
536
|
+
|
|
447
537
|
# Neurips
|
|
448
538
|
- name: neurips/local
|
|
449
539
|
tokenizer_spec:
|
|
@@ -461,6 +551,14 @@ tokenizer_configs:
|
|
|
461
551
|
end_of_text_token: "<|endoftext|>"
|
|
462
552
|
prefix_token: "<|endoftext|>"
|
|
463
553
|
|
|
554
|
+
- name: nvidia/llama-3.1-nemotron-70b-instruct
|
|
555
|
+
tokenizer_spec:
|
|
556
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
557
|
+
args:
|
|
558
|
+
pretrained_model_name_or_path: nvidia/Llama-3.1-Nemotron-70B-Instruct-HF
|
|
559
|
+
end_of_text_token: "<|eot_id|>"
|
|
560
|
+
prefix_token: "<|begin_of_text|>"
|
|
561
|
+
|
|
464
562
|
# OpenAI
|
|
465
563
|
- name: openai/cl100k_base
|
|
466
564
|
tokenizer_spec:
|
|
@@ -510,8 +608,22 @@ tokenizer_configs:
|
|
|
510
608
|
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
511
609
|
args:
|
|
512
610
|
pretrained_model_name_or_path: Qwen/Qwen2-72B-Instruct
|
|
513
|
-
end_of_text_token: <|im_end|>"
|
|
514
|
-
prefix_token: "<|im_start|>
|
|
611
|
+
end_of_text_token: "<|im_end|>"
|
|
612
|
+
prefix_token: "<|im_start|>"
|
|
613
|
+
|
|
614
|
+
- name: qwen/qwen2.5-7b-instruct
|
|
615
|
+
tokenizer_spec:
|
|
616
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
617
|
+
args:
|
|
618
|
+
pretrained_model_name_or_path: Qwen/Qwen2.5-7B-Instruct
|
|
619
|
+
end_of_text_token: "<|im_end|>"
|
|
620
|
+
prefix_token: "<|im_start|>"
|
|
621
|
+
|
|
622
|
+
- name: qwen/qwq-32b-preview
|
|
623
|
+
tokenizer_spec:
|
|
624
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
625
|
+
end_of_text_token: "<|im_end|>"
|
|
626
|
+
prefix_token: ""
|
|
515
627
|
|
|
516
628
|
- name: qwen/qwen-vl
|
|
517
629
|
tokenizer_spec:
|
|
@@ -533,6 +645,25 @@ tokenizer_configs:
|
|
|
533
645
|
end_of_text_token: "<|endoftext|>"
|
|
534
646
|
prefix_token: ""
|
|
535
647
|
|
|
648
|
+
- name: qwen/qwen-audio-chat
|
|
649
|
+
tokenizer_spec:
|
|
650
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
651
|
+
args:
|
|
652
|
+
pretrained_model_name_or_path: Qwen/Qwen-Audio-Chat
|
|
653
|
+
trust_remote_code: true
|
|
654
|
+
# Source: https://github.com/QwenLM/Qwen-Audio
|
|
655
|
+
end_of_text_token: "<|endoftext|>"
|
|
656
|
+
prefix_token: ""
|
|
657
|
+
|
|
658
|
+
- name: qwen/qwen2-audio-instruct
|
|
659
|
+
tokenizer_spec:
|
|
660
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
661
|
+
args:
|
|
662
|
+
pretrained_model_name_or_path: Qwen/Qwen2-Audio-7B-Instruct
|
|
663
|
+
trust_remote_code: false
|
|
664
|
+
end_of_text_token: "<|endoftext|>"
|
|
665
|
+
prefix_token: ""
|
|
666
|
+
|
|
536
667
|
# SambaLingo
|
|
537
668
|
- name: sambanova/sambalingo-thai-base
|
|
538
669
|
tokenizer_spec:
|
|
@@ -573,6 +704,15 @@ tokenizer_configs:
|
|
|
573
704
|
end_of_text_token: "</s>"
|
|
574
705
|
prefix_token: "<s>"
|
|
575
706
|
|
|
707
|
+
# Upstage
|
|
708
|
+
- name: upstage/solar-pro-preview-instruct
|
|
709
|
+
tokenizer_spec:
|
|
710
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
711
|
+
args:
|
|
712
|
+
trust_remote_code: true
|
|
713
|
+
end_of_text_token: "<|im_end|>"
|
|
714
|
+
prefix_token: "<|startoftext|>"
|
|
715
|
+
|
|
576
716
|
# Writer
|
|
577
717
|
- name: writer/gpt2
|
|
578
718
|
tokenizer_spec:
|
|
@@ -588,3 +728,239 @@ tokenizer_configs:
|
|
|
588
728
|
class_name: "helm.tokenizers.yalm_tokenizer.YaLMTokenizer"
|
|
589
729
|
end_of_text_token: "</s>"
|
|
590
730
|
prefix_token: "</s>"
|
|
731
|
+
|
|
732
|
+
# Diva Llama
|
|
733
|
+
- name: stanford/diva-llama
|
|
734
|
+
tokenizer_spec:
|
|
735
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
736
|
+
args:
|
|
737
|
+
pretrained_model_name_or_path: WillHeld/DiVA-llama-3-v0-8b
|
|
738
|
+
trust_remote_code: true
|
|
739
|
+
prefix_token: "<|begin_of_text|>"
|
|
740
|
+
end_of_text_token: "<|eot_id|>"
|
|
741
|
+
|
|
742
|
+
# LLaMA-Omni
|
|
743
|
+
- name: ictnlp/llama-3.1-8b-omni
|
|
744
|
+
tokenizer_spec:
|
|
745
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
746
|
+
args:
|
|
747
|
+
pretrained_model_name_or_path: ICTNLP/Llama-3.1-8B-Omni
|
|
748
|
+
trust_remote_code: false
|
|
749
|
+
end_of_text_token: "<|eot_id|>"
|
|
750
|
+
prefix_token: "<|begin_of_text|>"
|
|
751
|
+
|
|
752
|
+
# IBM - Granite 3.0
|
|
753
|
+
- name: ibm-granite/granite-3.0-2b-base
|
|
754
|
+
tokenizer_spec:
|
|
755
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
756
|
+
args:
|
|
757
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-2b-base
|
|
758
|
+
end_of_text_token: ""
|
|
759
|
+
prefix_token: ""
|
|
760
|
+
|
|
761
|
+
- name: ibm-granite/granite-3.0-2b-instruct
|
|
762
|
+
tokenizer_spec:
|
|
763
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
764
|
+
args:
|
|
765
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-2b-instruct
|
|
766
|
+
end_of_text_token: ""
|
|
767
|
+
prefix_token: ""
|
|
768
|
+
|
|
769
|
+
- name: ibm-granite/granite-3.0-8b-instruct
|
|
770
|
+
tokenizer_spec:
|
|
771
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
772
|
+
args:
|
|
773
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-8b-instruct
|
|
774
|
+
end_of_text_token: ""
|
|
775
|
+
prefix_token: ""
|
|
776
|
+
|
|
777
|
+
- name: ibm-granite/granite-3.0-8b-base
|
|
778
|
+
tokenizer_spec:
|
|
779
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
780
|
+
args:
|
|
781
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-8b-base
|
|
782
|
+
end_of_text_token: ""
|
|
783
|
+
prefix_token: ""
|
|
784
|
+
|
|
785
|
+
- name: ibm-granite/granite-3.0-3b-a800m-instruct
|
|
786
|
+
tokenizer_spec:
|
|
787
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
788
|
+
args:
|
|
789
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-3b-a800m-instruct
|
|
790
|
+
end_of_text_token: ""
|
|
791
|
+
prefix_token: ""
|
|
792
|
+
|
|
793
|
+
- name: ibm-granite/granite-3.0-3b-a800m-base
|
|
794
|
+
tokenizer_spec:
|
|
795
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
796
|
+
args:
|
|
797
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-3b-a800m-base
|
|
798
|
+
end_of_text_token: ""
|
|
799
|
+
prefix_token: ""
|
|
800
|
+
|
|
801
|
+
- name: ibm-granite/granite-3.0-1b-a400m-instruct
|
|
802
|
+
tokenizer_spec:
|
|
803
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
804
|
+
args:
|
|
805
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-1b-a400m-instruct
|
|
806
|
+
end_of_text_token: ""
|
|
807
|
+
prefix_token: ""
|
|
808
|
+
|
|
809
|
+
- name: ibm-granite/granite-3.0-1b-a400m-base
|
|
810
|
+
tokenizer_spec:
|
|
811
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
812
|
+
args:
|
|
813
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.0-1b-a400m-base
|
|
814
|
+
end_of_text_token: ""
|
|
815
|
+
prefix_token: ""
|
|
816
|
+
|
|
817
|
+
- name: maritaca-ai/sabia-7b
|
|
818
|
+
tokenizer_spec:
|
|
819
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
820
|
+
args:
|
|
821
|
+
pretrained_model_name_or_path: maritaca-ai/sabia-7b
|
|
822
|
+
end_of_text_token: "</s>"
|
|
823
|
+
prefix_token: "<s>"
|
|
824
|
+
|
|
825
|
+
# Granite-3.1-8b-base
|
|
826
|
+
- name: ibm-granite/granite-3.1-8b-base
|
|
827
|
+
tokenizer_spec:
|
|
828
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
829
|
+
args:
|
|
830
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-8b-base
|
|
831
|
+
prefix_token: ""
|
|
832
|
+
end_of_text_token: "<|endoftext|>"
|
|
833
|
+
|
|
834
|
+
# Granite-3.1-8b-instruct
|
|
835
|
+
- name: ibm-granite/granite-3.1-8b-instruct
|
|
836
|
+
tokenizer_spec:
|
|
837
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
838
|
+
args:
|
|
839
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-8b-instruct
|
|
840
|
+
prefix_token: ""
|
|
841
|
+
end_of_text_token: "<|endoftext|>"
|
|
842
|
+
|
|
843
|
+
# Granite-3.1-2b-instruct
|
|
844
|
+
- name: ibm-granite/granite-3.1-2b-instruct
|
|
845
|
+
tokenizer_spec:
|
|
846
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
847
|
+
args:
|
|
848
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-2b-instruct
|
|
849
|
+
prefix_token: ""
|
|
850
|
+
end_of_text_token: ""
|
|
851
|
+
|
|
852
|
+
# Granite-3.1-2b-base
|
|
853
|
+
- name: ibm-granite/granite-3.1-2b-base
|
|
854
|
+
tokenizer_spec:
|
|
855
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
856
|
+
args:
|
|
857
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-2b-base
|
|
858
|
+
prefix_token: ""
|
|
859
|
+
end_of_text_token: ""
|
|
860
|
+
|
|
861
|
+
# Granite-3.1-3b-a800m-instruct
|
|
862
|
+
- name: ibm-granite/granite-3.1-3b-a800m-instruct
|
|
863
|
+
tokenizer_spec:
|
|
864
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
865
|
+
args:
|
|
866
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-3b-a800m-instruct
|
|
867
|
+
prefix_token: ""
|
|
868
|
+
end_of_text_token: ""
|
|
869
|
+
|
|
870
|
+
# Granite-3.1-3b-a800m-base
|
|
871
|
+
- name: ibm-granite/granite-3.1-3b-a800m-base
|
|
872
|
+
tokenizer_spec:
|
|
873
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
874
|
+
args:
|
|
875
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-3b-a800m-base
|
|
876
|
+
prefix_token: ""
|
|
877
|
+
end_of_text_token: ""
|
|
878
|
+
|
|
879
|
+
# Granite-3.1-1b-a400m-instruct
|
|
880
|
+
- name: ibm-granite/granite-3.1-1b-a400m-instruct
|
|
881
|
+
tokenizer_spec:
|
|
882
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
883
|
+
args:
|
|
884
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-1b-a400m-instruct
|
|
885
|
+
prefix_token: ""
|
|
886
|
+
end_of_text_token: ""
|
|
887
|
+
|
|
888
|
+
# Granite-3.1-1b-a400m-base
|
|
889
|
+
- name: ibm-granite/granite-3.1-1b-a400m-base
|
|
890
|
+
tokenizer_spec:
|
|
891
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
892
|
+
args:
|
|
893
|
+
pretrained_model_name_or_path: ibm-granite/granite-3.1-1b-a400m-base
|
|
894
|
+
prefix_token: ""
|
|
895
|
+
end_of_text_token: ""
|
|
896
|
+
|
|
897
|
+
- name: ibm-granite/granite-20b-code-instruct-8k
|
|
898
|
+
tokenizer_spec:
|
|
899
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
900
|
+
args:
|
|
901
|
+
pretrained_model_name_or_path: ibm-granite/granite-20b-code-instruct-8k
|
|
902
|
+
prefix_token: ""
|
|
903
|
+
end_of_text_token: ""
|
|
904
|
+
|
|
905
|
+
- name: ibm-granite/granite-3b-code-instruct-128k
|
|
906
|
+
tokenizer_spec:
|
|
907
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
908
|
+
args:
|
|
909
|
+
pretrained_model_name_or_path: ibm-granite/granite-3b-code-instruct-128k
|
|
910
|
+
prefix_token: ""
|
|
911
|
+
end_of_text_token: ""
|
|
912
|
+
|
|
913
|
+
|
|
914
|
+
|
|
915
|
+
- name: ibm-granite/granite-34b-code-instruct-8k
|
|
916
|
+
tokenizer_spec:
|
|
917
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
918
|
+
args:
|
|
919
|
+
pretrained_model_name_or_path: ibm-granite/granite-34b-code-instruct-8k
|
|
920
|
+
prefix_token: ""
|
|
921
|
+
end_of_text_token: ""
|
|
922
|
+
|
|
923
|
+
- name: ibm-granite/granite-8b-code-instruct-128k
|
|
924
|
+
tokenizer_spec:
|
|
925
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
926
|
+
args:
|
|
927
|
+
pretrained_model_name_or_path: ibm-granite/granite-8b-code-instruct-128k
|
|
928
|
+
prefix_token: ""
|
|
929
|
+
end_of_text_token: ""
|
|
930
|
+
|
|
931
|
+
|
|
932
|
+
- name: ibm-granite/granite-guardian-3.1-2b
|
|
933
|
+
tokenizer_spec:
|
|
934
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
935
|
+
args:
|
|
936
|
+
pretrained_model_name_or_path: ibm-granite/granite-guardian-3.1-2b
|
|
937
|
+
prefix_token: ""
|
|
938
|
+
end_of_text_token: ""
|
|
939
|
+
|
|
940
|
+
- name: ibm-granite/granite-guardian-3.1-8b
|
|
941
|
+
tokenizer_spec:
|
|
942
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
943
|
+
args:
|
|
944
|
+
pretrained_model_name_or_path: ibm-granite/granite-guardian-3.1-8b
|
|
945
|
+
prefix_token: ""
|
|
946
|
+
end_of_text_token: ""
|
|
947
|
+
|
|
948
|
+
|
|
949
|
+
|
|
950
|
+
# DeepSeek-R1-Distill-Llama-3.1-8b
|
|
951
|
+
- name: deepseek-ai/DeepSeek-R1-Distill-Llama-8B
|
|
952
|
+
tokenizer_spec:
|
|
953
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
954
|
+
args:
|
|
955
|
+
pretrained_model_name_or_path: deepseek-ai/DeepSeek-R1-Distill-Llama-8B
|
|
956
|
+
end_of_text_token: "<|end▁of▁sentence|>"
|
|
957
|
+
prefix_token: "<|begin▁of▁sentence|>"
|
|
958
|
+
|
|
959
|
+
# deepseek-ai/deepseek-coder-6.7b-instruct
|
|
960
|
+
- name: deepseek-ai/deepseek-coder-6.7b-instruct
|
|
961
|
+
tokenizer_spec:
|
|
962
|
+
class_name: "helm.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"
|
|
963
|
+
args:
|
|
964
|
+
pretrained_model_name_or_path: deepseek-ai/deepseek-coder-6.7b-instruct
|
|
965
|
+
end_of_text_token: "<|end▁of▁sentence|>"
|
|
966
|
+
prefix_token: "<|begin▁of▁sentence|>"
|
helm/proxy/cli.py
CHANGED
|
@@ -23,8 +23,8 @@ import sys
|
|
|
23
23
|
|
|
24
24
|
from helm.common.hierarchical_logger import hlog
|
|
25
25
|
from helm.common.authentication import Authentication
|
|
26
|
-
from .accounts import Usage, Account
|
|
27
|
-
from .services.remote_service import RemoteService, add_service_args, create_authentication
|
|
26
|
+
from helm.proxy.accounts import Usage, Account
|
|
27
|
+
from helm.proxy.services.remote_service import RemoteService, add_service_args, create_authentication
|
|
28
28
|
|
|
29
29
|
GRANULARITIES = ["daily", "monthly", "total"]
|
|
30
30
|
UNLIMITED_QUOTA = "unlimited"
|
helm/proxy/example_queries.py
CHANGED
helm/proxy/server.py
CHANGED
|
@@ -6,6 +6,7 @@ Look at `index.js` to see how the functionality is invoked.
|
|
|
6
6
|
from urllib.parse import unquote_plus
|
|
7
7
|
import argparse
|
|
8
8
|
import dataclasses
|
|
9
|
+
import importlib_resources as resources
|
|
9
10
|
import json
|
|
10
11
|
import os
|
|
11
12
|
import sys
|
|
@@ -29,9 +30,9 @@ from helm.common.perspective_api_request import PerspectiveAPIRequest
|
|
|
29
30
|
from helm.common.moderations_api_request import ModerationAPIRequest
|
|
30
31
|
from helm.common.tokenization_request import TokenizationRequest, DecodeRequest
|
|
31
32
|
from helm.proxy.services.service import CACHE_DIR
|
|
32
|
-
from .accounts import Account
|
|
33
|
-
from .services.server_service import ServerService
|
|
34
|
-
from .query import Query
|
|
33
|
+
from helm.proxy.accounts import Account
|
|
34
|
+
from helm.proxy.services.server_service import ServerService
|
|
35
|
+
from helm.proxy.query import Query
|
|
35
36
|
|
|
36
37
|
try:
|
|
37
38
|
import gunicorn # noqa
|
|
@@ -86,7 +87,7 @@ def handle_root():
|
|
|
86
87
|
|
|
87
88
|
@app.get("/static/<filename:path>")
|
|
88
89
|
def handle_static_filename(filename):
|
|
89
|
-
resp = bottle.static_file(filename, root=
|
|
90
|
+
resp = bottle.static_file(filename, root=app.config["helm.staticpath"])
|
|
90
91
|
resp.add_header("Cache-Control", "no-store, must-revalidate ")
|
|
91
92
|
return resp
|
|
92
93
|
|
|
@@ -284,6 +285,12 @@ def main():
|
|
|
284
285
|
ensure_directory_exists(sqlite_cache_path)
|
|
285
286
|
cache_backend_config = SqliteCacheBackendConfig(sqlite_cache_path)
|
|
286
287
|
|
|
288
|
+
static_package_name = "helm.proxy.static"
|
|
289
|
+
resource_path = resources.files(static_package_name).joinpath("index.html")
|
|
290
|
+
with resources.as_file(resource_path) as resource_filename:
|
|
291
|
+
static_path = str(resource_filename.parent)
|
|
292
|
+
app.config["helm.staticpath"] = static_path
|
|
293
|
+
|
|
287
294
|
service = ServerService(base_path=args.base_path, cache_backend_config=cache_backend_config)
|
|
288
295
|
|
|
289
296
|
gunicorn_args = {
|
|
@@ -24,7 +24,7 @@ from helm.common.request import Request, RequestResult
|
|
|
24
24
|
from dacite import from_dict
|
|
25
25
|
from helm.proxy.accounts import Account
|
|
26
26
|
from helm.proxy.query import Query, QueryResult
|
|
27
|
-
from .service import Service, GeneralInfo
|
|
27
|
+
from helm.proxy.services.service import Service, GeneralInfo
|
|
28
28
|
|
|
29
29
|
|
|
30
30
|
class RemoteServiceError(Exception):
|
|
@@ -35,7 +35,7 @@ from helm.proxy.query import Query, QueryResult
|
|
|
35
35
|
from helm.proxy.retry import retry_request
|
|
36
36
|
from helm.proxy.token_counters.auto_token_counter import AutoTokenCounter
|
|
37
37
|
from helm.tokenizers.auto_tokenizer import AutoTokenizer
|
|
38
|
-
from .service import (
|
|
38
|
+
from helm.proxy.services.service import (
|
|
39
39
|
Service,
|
|
40
40
|
CACHE_DIR,
|
|
41
41
|
ACCOUNTS_FILE,
|
|
@@ -18,8 +18,8 @@ from helm.common.authentication import Authentication
|
|
|
18
18
|
from helm.common.request import Request, RequestResult
|
|
19
19
|
from helm.common.tokenization_request import TokenizationRequest, TokenizationRequestResult
|
|
20
20
|
from helm.proxy.accounts import Account, set_default_quotas
|
|
21
|
-
from .remote_service import RemoteService
|
|
22
|
-
from .service import ACCOUNTS_FILE
|
|
21
|
+
from helm.proxy.services.remote_service import RemoteService
|
|
22
|
+
from helm.proxy.services.service import ACCOUNTS_FILE
|
|
23
23
|
|
|
24
24
|
|
|
25
25
|
@dataclass(frozen=True)
|
|
@@ -8,7 +8,7 @@ from helm.common.authentication import Authentication
|
|
|
8
8
|
from helm.common.request import Request
|
|
9
9
|
from helm.proxy.accounts import AuthenticationError, Accounts
|
|
10
10
|
from helm.proxy.query import Query
|
|
11
|
-
from .server_service import ServerService
|
|
11
|
+
from helm.proxy.services.server_service import ServerService
|
|
12
12
|
|
|
13
13
|
|
|
14
14
|
def get_authentication():
|
|
@@ -0,0 +1,122 @@
|
|
|
1
|
+
function assert(condition, message) {
|
|
2
|
+
if (!condition) {
|
|
3
|
+
throw message || "Assertion failed";
|
|
4
|
+
}
|
|
5
|
+
}
|
|
6
|
+
|
|
7
|
+
function encodeUrlParams(params) {
|
|
8
|
+
let s = "";
|
|
9
|
+
for (let k in params) {
|
|
10
|
+
if (params[k] != null) {
|
|
11
|
+
s += (s === "" ? "?" : "&") + k + "=" + encodeURIComponent(params[k]);
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
return s;
|
|
15
|
+
}
|
|
16
|
+
|
|
17
|
+
function decodeUrlParams(str) {
|
|
18
|
+
const params = {};
|
|
19
|
+
if (str === "") return params;
|
|
20
|
+
const items = str.substring(1).split(/&/);
|
|
21
|
+
for (let i = 0; i < items.length; i++) {
|
|
22
|
+
const pair = items[i].split(/=/);
|
|
23
|
+
params[pair[0]] = decodeURIComponent(pair[1]);
|
|
24
|
+
}
|
|
25
|
+
return params;
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
function updateBrowserLocation(params) {
|
|
29
|
+
// Update the address bar
|
|
30
|
+
window.history.pushState(
|
|
31
|
+
{},
|
|
32
|
+
"",
|
|
33
|
+
window.location.pathname + encodeUrlParams(params),
|
|
34
|
+
);
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
function createCookie(key, value, days) {
|
|
38
|
+
let expires = "";
|
|
39
|
+
if (days) {
|
|
40
|
+
const date = new Date();
|
|
41
|
+
date.setTime(date.getTime() + days * 24 * 60 * 60 * 1000);
|
|
42
|
+
expires = "; expires=" + date.toUTCString();
|
|
43
|
+
}
|
|
44
|
+
document.cookie = key + "=" + value + expires + "; path=/";
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
function readCookie(key) {
|
|
48
|
+
let tokens = document.cookie.split(";");
|
|
49
|
+
for (let i = 0; i < tokens.length; i++) {
|
|
50
|
+
const [k, v] = tokens[i].trim().split("=", 2);
|
|
51
|
+
if (key === k) return v;
|
|
52
|
+
}
|
|
53
|
+
return null;
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
function eraseCookie(key) {
|
|
57
|
+
createCookie(key, "", -1);
|
|
58
|
+
}
|
|
59
|
+
|
|
60
|
+
function renderTimestamp(timestamp) {
|
|
61
|
+
if (!timestamp) return null;
|
|
62
|
+
const d = new Date(timestamp * 1000);
|
|
63
|
+
return d.toLocaleString();
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
function renderDict(data) {
|
|
67
|
+
return JSON.stringify(data).substring(0, 10000);
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
function loadScript(src, onload, onerror) {
|
|
71
|
+
// Using jquery doesn't work, so do it in with our bare hands.
|
|
72
|
+
const s = document.createElement("script");
|
|
73
|
+
s.src = src;
|
|
74
|
+
s.onload = onload;
|
|
75
|
+
s.onerror = onerror;
|
|
76
|
+
document.head.appendChild(s);
|
|
77
|
+
}
|
|
78
|
+
|
|
79
|
+
function getRandomString() {
|
|
80
|
+
const vocab =
|
|
81
|
+
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
|
|
82
|
+
let text = "";
|
|
83
|
+
for (let i = 0; i < 6; i++)
|
|
84
|
+
text += vocab.charAt(Math.floor(Math.random() * vocab.length));
|
|
85
|
+
return text;
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
function round(x, n) {
|
|
89
|
+
const base = Math.pow(10, n);
|
|
90
|
+
return Math.round(x * base) / base;
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
function multilineHtml(s) {
|
|
94
|
+
return s.replace(/\n/g, "<br>");
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
function renderError(e) {
|
|
98
|
+
return $("<div>").addClass("alert alert-danger").append(multilineHtml(e));
|
|
99
|
+
}
|
|
100
|
+
|
|
101
|
+
function helpIcon(help, link) {
|
|
102
|
+
// Show a ?
|
|
103
|
+
return $("<a>", { href: link, target: "blank_", class: "help-icon" }).append(
|
|
104
|
+
$("<img>", { src: "info-icon.png", width: 15, title: help }),
|
|
105
|
+
);
|
|
106
|
+
}
|
|
107
|
+
|
|
108
|
+
const markdownConverter = new showdown.Converter({ optionKey: "value" });
|
|
109
|
+
function renderMarkdown(markdown) {
|
|
110
|
+
return markdown && markdownConverter.makeHtml(markdown);
|
|
111
|
+
}
|
|
112
|
+
|
|
113
|
+
function refreshHashLocation() {
|
|
114
|
+
// If we request a hash location (URL contains #foo), the problem is #foo
|
|
115
|
+
// might not exist (since it's generated). Call this function to jump to the
|
|
116
|
+
// hash location once all the anchors are generated.
|
|
117
|
+
if (location.hash) {
|
|
118
|
+
const hash = location.hash;
|
|
119
|
+
location.hash = "";
|
|
120
|
+
location.hash = hash;
|
|
121
|
+
}
|
|
122
|
+
}
|