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
|
@@ -1,1570 +0,0 @@
|
|
|
1
|
-
"""Temporary test for preserving invariants during the model / tokenizer / window service refactor.
|
|
2
|
-
|
|
3
|
-
Delete this after the refactor is done."""
|
|
4
|
-
|
|
5
|
-
import pytest
|
|
6
|
-
from tempfile import TemporaryDirectory
|
|
7
|
-
from typing import Any
|
|
8
|
-
from helm.benchmark.config_registry import register_builtin_configs_from_helm_package
|
|
9
|
-
from helm.benchmark.model_deployment_registry import (
|
|
10
|
-
ClientSpec,
|
|
11
|
-
ModelDeployment,
|
|
12
|
-
WindowServiceSpec,
|
|
13
|
-
get_model_deployment,
|
|
14
|
-
)
|
|
15
|
-
from helm.benchmark.model_metadata_registry import get_model_metadata, ModelMetadata
|
|
16
|
-
from helm.benchmark.tokenizer_config_registry import TokenizerConfig, TokenizerSpec
|
|
17
|
-
from helm.benchmark.window_services.test_utils import get_tokenizer_service
|
|
18
|
-
|
|
19
|
-
from helm.benchmark.window_services.window_service_factory import WindowServiceFactory
|
|
20
|
-
from helm.proxy.clients.auto_client import AutoClient
|
|
21
|
-
from helm.proxy.tokenizers.auto_tokenizer import AutoTokenizer
|
|
22
|
-
from collections import defaultdict
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
_BUILT_IN_TOKENIZER_CONFIGS = [
|
|
26
|
-
TokenizerConfig(
|
|
27
|
-
name="simple/model1",
|
|
28
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.simple_tokenizer.SimpleTokenizer"),
|
|
29
|
-
end_of_text_token="</s>",
|
|
30
|
-
prefix_token="<s>",
|
|
31
|
-
),
|
|
32
|
-
TokenizerConfig(
|
|
33
|
-
name="neurips/local",
|
|
34
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.http_model_tokenizer.HTTPModelTokenizer"),
|
|
35
|
-
end_of_text_token="<|endoftext|>",
|
|
36
|
-
prefix_token="<|endoftext|>",
|
|
37
|
-
),
|
|
38
|
-
TokenizerConfig(
|
|
39
|
-
name="ai21/j1",
|
|
40
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.ai21_tokenizer.AI21Tokenizer"),
|
|
41
|
-
end_of_text_token=" ",
|
|
42
|
-
prefix_token="",
|
|
43
|
-
),
|
|
44
|
-
TokenizerConfig(
|
|
45
|
-
name="AlephAlpha/luminous-base",
|
|
46
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.aleph_alpha_tokenizer.AlephAlphaTokenizer"),
|
|
47
|
-
end_of_text_token="",
|
|
48
|
-
prefix_token="",
|
|
49
|
-
),
|
|
50
|
-
TokenizerConfig(
|
|
51
|
-
name="AlephAlpha/luminous-extended",
|
|
52
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.aleph_alpha_tokenizer.AlephAlphaTokenizer"),
|
|
53
|
-
end_of_text_token="",
|
|
54
|
-
prefix_token="",
|
|
55
|
-
),
|
|
56
|
-
TokenizerConfig(
|
|
57
|
-
name="AlephAlpha/luminous-supreme",
|
|
58
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.aleph_alpha_tokenizer.AlephAlphaTokenizer"),
|
|
59
|
-
end_of_text_token="",
|
|
60
|
-
prefix_token="",
|
|
61
|
-
),
|
|
62
|
-
TokenizerConfig(
|
|
63
|
-
name="huggingface/gpt2",
|
|
64
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
65
|
-
end_of_text_token="<|endoftext|>",
|
|
66
|
-
prefix_token="<|endoftext|>",
|
|
67
|
-
),
|
|
68
|
-
TokenizerConfig(
|
|
69
|
-
name="microsoft/gpt2",
|
|
70
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
71
|
-
end_of_text_token="<|endoftext|>",
|
|
72
|
-
prefix_token="<<",
|
|
73
|
-
),
|
|
74
|
-
TokenizerConfig(
|
|
75
|
-
name="writer/gpt2",
|
|
76
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
77
|
-
end_of_text_token="",
|
|
78
|
-
prefix_token="",
|
|
79
|
-
),
|
|
80
|
-
TokenizerConfig(
|
|
81
|
-
name="anthropic/claude",
|
|
82
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.anthropic_tokenizer.AnthropicTokenizer"),
|
|
83
|
-
end_of_text_token="<|endoftext|>",
|
|
84
|
-
prefix_token="<|endoftext|>",
|
|
85
|
-
),
|
|
86
|
-
TokenizerConfig(
|
|
87
|
-
name="bigscience/bloom",
|
|
88
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
89
|
-
end_of_text_token="</s>",
|
|
90
|
-
prefix_token="</s>",
|
|
91
|
-
),
|
|
92
|
-
TokenizerConfig(
|
|
93
|
-
name="bigscience/T0pp",
|
|
94
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
95
|
-
end_of_text_token="</s>",
|
|
96
|
-
prefix_token="",
|
|
97
|
-
),
|
|
98
|
-
TokenizerConfig(
|
|
99
|
-
name="cohere/cohere",
|
|
100
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.cohere_tokenizer.CohereTokenizer"),
|
|
101
|
-
end_of_text_token="",
|
|
102
|
-
prefix_token=":",
|
|
103
|
-
),
|
|
104
|
-
TokenizerConfig(
|
|
105
|
-
name="EleutherAI/gpt-j-6B",
|
|
106
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
107
|
-
end_of_text_token="<|endoftext|>",
|
|
108
|
-
prefix_token="<|endoftext|>",
|
|
109
|
-
),
|
|
110
|
-
TokenizerConfig(
|
|
111
|
-
name="EleutherAI/gpt-neox-20b",
|
|
112
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
113
|
-
end_of_text_token="<|endoftext|>",
|
|
114
|
-
prefix_token="<|endoftext|>",
|
|
115
|
-
),
|
|
116
|
-
TokenizerConfig(
|
|
117
|
-
name="hf-internal-testing/llama-tokenizer",
|
|
118
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
119
|
-
end_of_text_token="</s>",
|
|
120
|
-
prefix_token="<s>",
|
|
121
|
-
),
|
|
122
|
-
TokenizerConfig(
|
|
123
|
-
name="meta-llama/Llama-2-7b-hf",
|
|
124
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
125
|
-
end_of_text_token="</s>",
|
|
126
|
-
prefix_token="<s>",
|
|
127
|
-
),
|
|
128
|
-
TokenizerConfig(
|
|
129
|
-
name="mistralai/Mistral-7B-v0.1",
|
|
130
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
131
|
-
end_of_text_token="</s>",
|
|
132
|
-
prefix_token="<s>",
|
|
133
|
-
),
|
|
134
|
-
TokenizerConfig(
|
|
135
|
-
name="tiiuae/falcon-7b",
|
|
136
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
137
|
-
end_of_text_token="<|endoftext|>",
|
|
138
|
-
prefix_token="",
|
|
139
|
-
),
|
|
140
|
-
TokenizerConfig(
|
|
141
|
-
name="bigcode/santacoder",
|
|
142
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
143
|
-
end_of_text_token="<|endoftext|>",
|
|
144
|
-
prefix_token="<|endoftext|>",
|
|
145
|
-
),
|
|
146
|
-
TokenizerConfig(
|
|
147
|
-
name="bigcode/starcoder",
|
|
148
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
149
|
-
end_of_text_token="<|endoftext|>",
|
|
150
|
-
prefix_token="<|endoftext|>",
|
|
151
|
-
),
|
|
152
|
-
TokenizerConfig(
|
|
153
|
-
name="google/t5-11b",
|
|
154
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
155
|
-
end_of_text_token="</s>",
|
|
156
|
-
prefix_token="",
|
|
157
|
-
),
|
|
158
|
-
TokenizerConfig(
|
|
159
|
-
name="google/flan-t5-xxl",
|
|
160
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
161
|
-
end_of_text_token="</s>",
|
|
162
|
-
prefix_token="",
|
|
163
|
-
),
|
|
164
|
-
TokenizerConfig(
|
|
165
|
-
name="google/ul2",
|
|
166
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
167
|
-
end_of_text_token="</s>",
|
|
168
|
-
prefix_token="",
|
|
169
|
-
),
|
|
170
|
-
TokenizerConfig(
|
|
171
|
-
name="google/mt5-base",
|
|
172
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
173
|
-
end_of_text_token="</s>",
|
|
174
|
-
prefix_token="",
|
|
175
|
-
),
|
|
176
|
-
TokenizerConfig(
|
|
177
|
-
name="facebook/opt-66b",
|
|
178
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
179
|
-
end_of_text_token="</s>",
|
|
180
|
-
prefix_token="</s>",
|
|
181
|
-
),
|
|
182
|
-
TokenizerConfig(
|
|
183
|
-
name="openai/cl100k_base",
|
|
184
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.tiktoken_tokenizer.TiktokenTokenizer"),
|
|
185
|
-
end_of_text_token="<|endoftext|>",
|
|
186
|
-
prefix_token="<|endoftext|>",
|
|
187
|
-
),
|
|
188
|
-
TokenizerConfig(
|
|
189
|
-
name="TsinghuaKEG/ice",
|
|
190
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.ice_tokenizer.ICETokenizer"),
|
|
191
|
-
end_of_text_token="</s>",
|
|
192
|
-
prefix_token="",
|
|
193
|
-
),
|
|
194
|
-
TokenizerConfig(
|
|
195
|
-
name="Yandex/yalm",
|
|
196
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.yalm_tokenizer.YaLMTokenizer"),
|
|
197
|
-
end_of_text_token="</s>",
|
|
198
|
-
prefix_token="</s>",
|
|
199
|
-
),
|
|
200
|
-
TokenizerConfig(
|
|
201
|
-
name="lightningai/lit-gpt",
|
|
202
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.lit_gpt_tokenizer.LitGPTTokenizer", args={}),
|
|
203
|
-
end_of_text_token="<|endoftext|>",
|
|
204
|
-
prefix_token="<|endoftext|>",
|
|
205
|
-
),
|
|
206
|
-
TokenizerConfig(
|
|
207
|
-
name="HuggingFaceM4/idefics-9b",
|
|
208
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
209
|
-
end_of_text_token="</s>",
|
|
210
|
-
prefix_token="<s>",
|
|
211
|
-
),
|
|
212
|
-
TokenizerConfig(
|
|
213
|
-
name="HuggingFaceM4/idefics-9b-instruct",
|
|
214
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
215
|
-
end_of_text_token="</s>",
|
|
216
|
-
prefix_token="<s>",
|
|
217
|
-
),
|
|
218
|
-
TokenizerConfig(
|
|
219
|
-
name="HuggingFaceM4/idefics-80b",
|
|
220
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
221
|
-
end_of_text_token="</s>",
|
|
222
|
-
prefix_token="<s>",
|
|
223
|
-
),
|
|
224
|
-
TokenizerConfig(
|
|
225
|
-
name="HuggingFaceM4/idefics-80b-instruct",
|
|
226
|
-
tokenizer_spec=TokenizerSpec(class_name="helm.proxy.tokenizers.huggingface_tokenizer.HuggingFaceTokenizer"),
|
|
227
|
-
end_of_text_token="</s>",
|
|
228
|
-
prefix_token="<s>",
|
|
229
|
-
),
|
|
230
|
-
]
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
_BUILT_IN_MODEL_DEPLOYMENTS = [
|
|
234
|
-
ModelDeployment(
|
|
235
|
-
name="neurips/local",
|
|
236
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.http_model_client.HTTPModelClient"),
|
|
237
|
-
tokenizer_name="neurips/local",
|
|
238
|
-
window_service_spec=WindowServiceSpec(
|
|
239
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
240
|
-
),
|
|
241
|
-
max_sequence_length=2048,
|
|
242
|
-
),
|
|
243
|
-
ModelDeployment(
|
|
244
|
-
name="ai21/j1-jumbo",
|
|
245
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
246
|
-
tokenizer_name="ai21/j1",
|
|
247
|
-
window_service_spec=WindowServiceSpec(
|
|
248
|
-
class_name="helm.benchmark.window_services.ai21_window_service.AI21WindowService"
|
|
249
|
-
),
|
|
250
|
-
max_sequence_length=2047,
|
|
251
|
-
),
|
|
252
|
-
ModelDeployment(
|
|
253
|
-
name="ai21/j1-grande",
|
|
254
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
255
|
-
tokenizer_name="ai21/j1",
|
|
256
|
-
window_service_spec=WindowServiceSpec(
|
|
257
|
-
class_name="helm.benchmark.window_services.ai21_window_service.AI21WindowService"
|
|
258
|
-
),
|
|
259
|
-
max_sequence_length=2047,
|
|
260
|
-
),
|
|
261
|
-
ModelDeployment(
|
|
262
|
-
name="ai21/j1-grande-v2-beta",
|
|
263
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
264
|
-
tokenizer_name="ai21/j1",
|
|
265
|
-
window_service_spec=WindowServiceSpec(
|
|
266
|
-
class_name="helm.benchmark.window_services.ai21_window_service.AI21WindowService"
|
|
267
|
-
),
|
|
268
|
-
max_sequence_length=2047,
|
|
269
|
-
),
|
|
270
|
-
ModelDeployment(
|
|
271
|
-
name="ai21/j1-large",
|
|
272
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
273
|
-
tokenizer_name="ai21/j1",
|
|
274
|
-
window_service_spec=WindowServiceSpec(
|
|
275
|
-
class_name="helm.benchmark.window_services.ai21_window_service.AI21WindowService"
|
|
276
|
-
),
|
|
277
|
-
max_sequence_length=2047,
|
|
278
|
-
),
|
|
279
|
-
ModelDeployment(
|
|
280
|
-
name="ai21/j2-jumbo",
|
|
281
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
282
|
-
tokenizer_name="ai21/j1",
|
|
283
|
-
window_service_spec=WindowServiceSpec(
|
|
284
|
-
class_name="helm.benchmark.window_services.wider_ai21_window_service.AI21Jurassic2JumboWindowService",
|
|
285
|
-
args={},
|
|
286
|
-
),
|
|
287
|
-
max_sequence_length=6000,
|
|
288
|
-
),
|
|
289
|
-
ModelDeployment(
|
|
290
|
-
name="ai21/j2-grande",
|
|
291
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
292
|
-
tokenizer_name="ai21/j1",
|
|
293
|
-
window_service_spec=WindowServiceSpec(
|
|
294
|
-
class_name="helm.benchmark.window_services.ai21_window_service.AI21WindowService"
|
|
295
|
-
),
|
|
296
|
-
max_sequence_length=2047,
|
|
297
|
-
),
|
|
298
|
-
ModelDeployment(
|
|
299
|
-
name="ai21/j2-large",
|
|
300
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.ai21_client.AI21Client"),
|
|
301
|
-
tokenizer_name="ai21/j1",
|
|
302
|
-
window_service_spec=WindowServiceSpec(
|
|
303
|
-
class_name="helm.benchmark.window_services.ai21_window_service.AI21WindowService"
|
|
304
|
-
),
|
|
305
|
-
max_sequence_length=2047,
|
|
306
|
-
),
|
|
307
|
-
ModelDeployment(
|
|
308
|
-
name="AlephAlpha/luminous-base",
|
|
309
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.aleph_alpha_client.AlephAlphaClient"),
|
|
310
|
-
tokenizer_name="AlephAlpha/luminous-base",
|
|
311
|
-
window_service_spec=WindowServiceSpec(
|
|
312
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
313
|
-
),
|
|
314
|
-
max_sequence_length=2048,
|
|
315
|
-
),
|
|
316
|
-
ModelDeployment(
|
|
317
|
-
name="AlephAlpha/luminous-extended",
|
|
318
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.aleph_alpha_client.AlephAlphaClient"),
|
|
319
|
-
tokenizer_name="AlephAlpha/luminous-extended",
|
|
320
|
-
window_service_spec=WindowServiceSpec(
|
|
321
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
322
|
-
),
|
|
323
|
-
max_sequence_length=2048,
|
|
324
|
-
),
|
|
325
|
-
ModelDeployment(
|
|
326
|
-
name="AlephAlpha/luminous-supreme",
|
|
327
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.aleph_alpha_client.AlephAlphaClient"),
|
|
328
|
-
tokenizer_name="AlephAlpha/luminous-supreme",
|
|
329
|
-
window_service_spec=WindowServiceSpec(
|
|
330
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
331
|
-
),
|
|
332
|
-
max_sequence_length=2048,
|
|
333
|
-
),
|
|
334
|
-
ModelDeployment(
|
|
335
|
-
name="anthropic/stanford-online-all-v4-s3",
|
|
336
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.anthropic_client.AnthropicLegacyClient"),
|
|
337
|
-
tokenizer_name="huggingface/gpt2",
|
|
338
|
-
window_service_spec=WindowServiceSpec(
|
|
339
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
340
|
-
),
|
|
341
|
-
max_sequence_length=8192,
|
|
342
|
-
),
|
|
343
|
-
ModelDeployment(
|
|
344
|
-
name="anthropic/claude-2.0",
|
|
345
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.anthropic_client.AnthropicClient"),
|
|
346
|
-
tokenizer_name="anthropic/claude",
|
|
347
|
-
window_service_spec=WindowServiceSpec(
|
|
348
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
349
|
-
),
|
|
350
|
-
max_sequence_length=8000,
|
|
351
|
-
max_sequence_and_generated_tokens_length=9016,
|
|
352
|
-
),
|
|
353
|
-
ModelDeployment(
|
|
354
|
-
name="anthropic/claude-2.1",
|
|
355
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.anthropic_client.AnthropicClient"),
|
|
356
|
-
tokenizer_name="anthropic/claude",
|
|
357
|
-
window_service_spec=WindowServiceSpec(
|
|
358
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
359
|
-
),
|
|
360
|
-
max_sequence_length=8000,
|
|
361
|
-
max_sequence_and_generated_tokens_length=9016,
|
|
362
|
-
),
|
|
363
|
-
ModelDeployment(
|
|
364
|
-
name="anthropic/claude-v1.3",
|
|
365
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.anthropic_client.AnthropicClient"),
|
|
366
|
-
tokenizer_name="anthropic/claude",
|
|
367
|
-
window_service_spec=WindowServiceSpec(
|
|
368
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
369
|
-
),
|
|
370
|
-
max_sequence_length=8000,
|
|
371
|
-
max_sequence_and_generated_tokens_length=9016,
|
|
372
|
-
),
|
|
373
|
-
ModelDeployment(
|
|
374
|
-
name="anthropic/claude-instant-v1",
|
|
375
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.anthropic_client.AnthropicClient"),
|
|
376
|
-
tokenizer_name="anthropic/claude",
|
|
377
|
-
window_service_spec=WindowServiceSpec(
|
|
378
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
379
|
-
),
|
|
380
|
-
max_sequence_length=8000,
|
|
381
|
-
max_sequence_and_generated_tokens_length=9016,
|
|
382
|
-
),
|
|
383
|
-
ModelDeployment(
|
|
384
|
-
name="together/bloom",
|
|
385
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
386
|
-
tokenizer_name="bigscience/bloom",
|
|
387
|
-
window_service_spec=WindowServiceSpec(
|
|
388
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
389
|
-
),
|
|
390
|
-
max_sequence_length=2048,
|
|
391
|
-
max_request_length=2049,
|
|
392
|
-
),
|
|
393
|
-
ModelDeployment(
|
|
394
|
-
name="together/t0pp",
|
|
395
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
396
|
-
tokenizer_name="bigscience/T0pp",
|
|
397
|
-
window_service_spec=WindowServiceSpec(
|
|
398
|
-
class_name="helm.benchmark.window_services.t0pp_window_service.T0ppWindowService"
|
|
399
|
-
),
|
|
400
|
-
max_sequence_length=1024,
|
|
401
|
-
),
|
|
402
|
-
ModelDeployment(
|
|
403
|
-
name="cohere/xlarge-20220609",
|
|
404
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
405
|
-
tokenizer_name="cohere/cohere",
|
|
406
|
-
window_service_spec=WindowServiceSpec(
|
|
407
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereWindowService"
|
|
408
|
-
),
|
|
409
|
-
max_sequence_length=2047,
|
|
410
|
-
max_request_length=2048,
|
|
411
|
-
),
|
|
412
|
-
ModelDeployment(
|
|
413
|
-
name="cohere/xlarge-20221108",
|
|
414
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
415
|
-
tokenizer_name="cohere/cohere",
|
|
416
|
-
window_service_spec=WindowServiceSpec(
|
|
417
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereWindowService"
|
|
418
|
-
),
|
|
419
|
-
max_sequence_length=2047,
|
|
420
|
-
max_request_length=2048,
|
|
421
|
-
),
|
|
422
|
-
ModelDeployment(
|
|
423
|
-
name="cohere/large-20220720",
|
|
424
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
425
|
-
tokenizer_name="cohere/cohere",
|
|
426
|
-
window_service_spec=WindowServiceSpec(
|
|
427
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereWindowService"
|
|
428
|
-
),
|
|
429
|
-
max_sequence_length=2047,
|
|
430
|
-
max_request_length=2048,
|
|
431
|
-
),
|
|
432
|
-
ModelDeployment(
|
|
433
|
-
name="cohere/medium-20220720",
|
|
434
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
435
|
-
tokenizer_name="cohere/cohere",
|
|
436
|
-
window_service_spec=WindowServiceSpec(
|
|
437
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereWindowService"
|
|
438
|
-
),
|
|
439
|
-
max_sequence_length=2047,
|
|
440
|
-
max_request_length=2048,
|
|
441
|
-
),
|
|
442
|
-
ModelDeployment(
|
|
443
|
-
name="cohere/medium-20221108",
|
|
444
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
445
|
-
tokenizer_name="cohere/cohere",
|
|
446
|
-
window_service_spec=WindowServiceSpec(
|
|
447
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereWindowService"
|
|
448
|
-
),
|
|
449
|
-
max_sequence_length=2047,
|
|
450
|
-
max_request_length=2048,
|
|
451
|
-
),
|
|
452
|
-
ModelDeployment(
|
|
453
|
-
name="cohere/small-20220720",
|
|
454
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
455
|
-
tokenizer_name="cohere/cohere",
|
|
456
|
-
window_service_spec=WindowServiceSpec(
|
|
457
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereWindowService"
|
|
458
|
-
),
|
|
459
|
-
max_sequence_length=2047,
|
|
460
|
-
max_request_length=2048,
|
|
461
|
-
),
|
|
462
|
-
ModelDeployment(
|
|
463
|
-
name="cohere/command-medium-beta",
|
|
464
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
465
|
-
tokenizer_name="cohere/cohere",
|
|
466
|
-
window_service_spec=WindowServiceSpec(
|
|
467
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereCommandWindowService"
|
|
468
|
-
),
|
|
469
|
-
max_sequence_length=2019,
|
|
470
|
-
max_request_length=2020,
|
|
471
|
-
),
|
|
472
|
-
ModelDeployment(
|
|
473
|
-
name="cohere/command-xlarge-beta",
|
|
474
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
475
|
-
tokenizer_name="cohere/cohere",
|
|
476
|
-
window_service_spec=WindowServiceSpec(
|
|
477
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereCommandWindowService"
|
|
478
|
-
),
|
|
479
|
-
max_sequence_length=2019,
|
|
480
|
-
max_request_length=2020,
|
|
481
|
-
),
|
|
482
|
-
ModelDeployment(
|
|
483
|
-
name="cohere/command",
|
|
484
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
485
|
-
tokenizer_name="cohere/cohere",
|
|
486
|
-
window_service_spec=WindowServiceSpec(
|
|
487
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereCommandWindowService"
|
|
488
|
-
),
|
|
489
|
-
max_sequence_length=2019,
|
|
490
|
-
max_request_length=2020,
|
|
491
|
-
),
|
|
492
|
-
ModelDeployment(
|
|
493
|
-
name="cohere/command-light",
|
|
494
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.cohere_client.CohereClient"),
|
|
495
|
-
tokenizer_name="cohere/cohere",
|
|
496
|
-
window_service_spec=WindowServiceSpec(
|
|
497
|
-
class_name="helm.benchmark.window_services.cohere_window_service.CohereCommandWindowService"
|
|
498
|
-
),
|
|
499
|
-
max_sequence_length=2019,
|
|
500
|
-
max_request_length=2020,
|
|
501
|
-
),
|
|
502
|
-
ModelDeployment(
|
|
503
|
-
name="together/gpt-j-6b",
|
|
504
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
505
|
-
tokenizer_name="EleutherAI/gpt-j-6B",
|
|
506
|
-
window_service_spec=WindowServiceSpec(
|
|
507
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
508
|
-
),
|
|
509
|
-
max_sequence_length=2048,
|
|
510
|
-
max_request_length=2049,
|
|
511
|
-
),
|
|
512
|
-
ModelDeployment(
|
|
513
|
-
name="together/gpt-neox-20b",
|
|
514
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
515
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
516
|
-
window_service_spec=WindowServiceSpec(
|
|
517
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
518
|
-
),
|
|
519
|
-
max_sequence_length=2048,
|
|
520
|
-
max_request_length=2049,
|
|
521
|
-
),
|
|
522
|
-
ModelDeployment(
|
|
523
|
-
name="together/pythia-1b-v0",
|
|
524
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
525
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
526
|
-
window_service_spec=WindowServiceSpec(
|
|
527
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
528
|
-
),
|
|
529
|
-
max_sequence_length=2048,
|
|
530
|
-
max_request_length=2049,
|
|
531
|
-
),
|
|
532
|
-
ModelDeployment(
|
|
533
|
-
name="together/pythia-2.8b-v0",
|
|
534
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
535
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
536
|
-
window_service_spec=WindowServiceSpec(
|
|
537
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
538
|
-
),
|
|
539
|
-
max_sequence_length=2048,
|
|
540
|
-
max_request_length=2049,
|
|
541
|
-
),
|
|
542
|
-
ModelDeployment(
|
|
543
|
-
name="together/pythia-6.9b",
|
|
544
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
545
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
546
|
-
window_service_spec=WindowServiceSpec(
|
|
547
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
548
|
-
),
|
|
549
|
-
max_sequence_length=2048,
|
|
550
|
-
max_request_length=2049,
|
|
551
|
-
),
|
|
552
|
-
ModelDeployment(
|
|
553
|
-
name="together/pythia-12b-v0",
|
|
554
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
555
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
556
|
-
window_service_spec=WindowServiceSpec(
|
|
557
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
558
|
-
),
|
|
559
|
-
max_sequence_length=2048,
|
|
560
|
-
max_request_length=2049,
|
|
561
|
-
),
|
|
562
|
-
ModelDeployment(
|
|
563
|
-
name="together/llama-7b",
|
|
564
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
565
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
566
|
-
window_service_spec=WindowServiceSpec(
|
|
567
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
568
|
-
),
|
|
569
|
-
max_sequence_length=2047,
|
|
570
|
-
),
|
|
571
|
-
ModelDeployment(
|
|
572
|
-
name="together/llama-13b",
|
|
573
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
574
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
575
|
-
window_service_spec=WindowServiceSpec(
|
|
576
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
577
|
-
),
|
|
578
|
-
max_sequence_length=2047,
|
|
579
|
-
),
|
|
580
|
-
ModelDeployment(
|
|
581
|
-
name="together/llama-30b",
|
|
582
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
583
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
584
|
-
window_service_spec=WindowServiceSpec(
|
|
585
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
586
|
-
),
|
|
587
|
-
max_sequence_length=2047,
|
|
588
|
-
),
|
|
589
|
-
ModelDeployment(
|
|
590
|
-
name="together/llama-65b",
|
|
591
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
592
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
593
|
-
window_service_spec=WindowServiceSpec(
|
|
594
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
595
|
-
),
|
|
596
|
-
max_sequence_length=2047,
|
|
597
|
-
),
|
|
598
|
-
ModelDeployment(
|
|
599
|
-
name="together/llama-2-7b",
|
|
600
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
601
|
-
tokenizer_name="meta-llama/Llama-2-7b-hf",
|
|
602
|
-
window_service_spec=WindowServiceSpec(
|
|
603
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
604
|
-
),
|
|
605
|
-
max_sequence_length=4094,
|
|
606
|
-
),
|
|
607
|
-
ModelDeployment(
|
|
608
|
-
name="together/llama-2-13b",
|
|
609
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
610
|
-
tokenizer_name="meta-llama/Llama-2-7b-hf",
|
|
611
|
-
window_service_spec=WindowServiceSpec(
|
|
612
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
613
|
-
),
|
|
614
|
-
max_sequence_length=4094,
|
|
615
|
-
),
|
|
616
|
-
ModelDeployment(
|
|
617
|
-
name="together/llama-2-70b",
|
|
618
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
619
|
-
tokenizer_name="meta-llama/Llama-2-7b-hf",
|
|
620
|
-
window_service_spec=WindowServiceSpec(
|
|
621
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
622
|
-
),
|
|
623
|
-
max_sequence_length=4094,
|
|
624
|
-
),
|
|
625
|
-
ModelDeployment(
|
|
626
|
-
name="together/alpaca-7b",
|
|
627
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
628
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
629
|
-
window_service_spec=WindowServiceSpec(
|
|
630
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
631
|
-
),
|
|
632
|
-
max_sequence_length=2048,
|
|
633
|
-
),
|
|
634
|
-
ModelDeployment(
|
|
635
|
-
name="together/vicuna-7b-v1.3",
|
|
636
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
637
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
638
|
-
window_service_spec=WindowServiceSpec(
|
|
639
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
640
|
-
),
|
|
641
|
-
max_sequence_length=2048,
|
|
642
|
-
),
|
|
643
|
-
ModelDeployment(
|
|
644
|
-
name="together/vicuna-13b-v1.3",
|
|
645
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
646
|
-
tokenizer_name="hf-internal-testing/llama-tokenizer",
|
|
647
|
-
window_service_spec=WindowServiceSpec(
|
|
648
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
649
|
-
),
|
|
650
|
-
max_sequence_length=2048,
|
|
651
|
-
),
|
|
652
|
-
ModelDeployment(
|
|
653
|
-
name="together/mistral-7b-v0.1",
|
|
654
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
655
|
-
tokenizer_name="mistralai/Mistral-7B-v0.1",
|
|
656
|
-
window_service_spec=WindowServiceSpec(
|
|
657
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
658
|
-
),
|
|
659
|
-
max_sequence_length=4095,
|
|
660
|
-
),
|
|
661
|
-
ModelDeployment(
|
|
662
|
-
name="together/mpt-7b",
|
|
663
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
664
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
665
|
-
window_service_spec=WindowServiceSpec(
|
|
666
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
667
|
-
),
|
|
668
|
-
max_sequence_length=2048,
|
|
669
|
-
max_request_length=2049,
|
|
670
|
-
),
|
|
671
|
-
ModelDeployment(
|
|
672
|
-
name="together/mpt-instruct-7b",
|
|
673
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
674
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
675
|
-
window_service_spec=WindowServiceSpec(
|
|
676
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
677
|
-
),
|
|
678
|
-
max_sequence_length=2048,
|
|
679
|
-
max_request_length=2049,
|
|
680
|
-
),
|
|
681
|
-
ModelDeployment(
|
|
682
|
-
name="together/mpt-30b",
|
|
683
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
684
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
685
|
-
window_service_spec=WindowServiceSpec(
|
|
686
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
687
|
-
),
|
|
688
|
-
max_sequence_length=2048,
|
|
689
|
-
max_request_length=2049,
|
|
690
|
-
),
|
|
691
|
-
ModelDeployment(
|
|
692
|
-
name="together/mpt-instruct-30b",
|
|
693
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
694
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
695
|
-
window_service_spec=WindowServiceSpec(
|
|
696
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
697
|
-
),
|
|
698
|
-
max_sequence_length=2048,
|
|
699
|
-
max_request_length=2049,
|
|
700
|
-
),
|
|
701
|
-
ModelDeployment(
|
|
702
|
-
name="together/falcon-7b",
|
|
703
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
704
|
-
tokenizer_name="tiiuae/falcon-7b",
|
|
705
|
-
window_service_spec=WindowServiceSpec(
|
|
706
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
707
|
-
),
|
|
708
|
-
max_sequence_length=2047,
|
|
709
|
-
),
|
|
710
|
-
ModelDeployment(
|
|
711
|
-
name="together/falcon-7b-instruct",
|
|
712
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
713
|
-
tokenizer_name="tiiuae/falcon-7b",
|
|
714
|
-
window_service_spec=WindowServiceSpec(
|
|
715
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
716
|
-
),
|
|
717
|
-
max_sequence_length=2047,
|
|
718
|
-
),
|
|
719
|
-
ModelDeployment(
|
|
720
|
-
name="together/falcon-40b",
|
|
721
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
722
|
-
tokenizer_name="tiiuae/falcon-7b",
|
|
723
|
-
window_service_spec=WindowServiceSpec(
|
|
724
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
725
|
-
),
|
|
726
|
-
max_sequence_length=2047,
|
|
727
|
-
),
|
|
728
|
-
ModelDeployment(
|
|
729
|
-
name="together/falcon-40b-instruct",
|
|
730
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
731
|
-
tokenizer_name="tiiuae/falcon-7b",
|
|
732
|
-
window_service_spec=WindowServiceSpec(
|
|
733
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
734
|
-
),
|
|
735
|
-
max_sequence_length=2047,
|
|
736
|
-
),
|
|
737
|
-
ModelDeployment(
|
|
738
|
-
name="gooseai/gpt-neo-20b",
|
|
739
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.goose_ai_client.GooseAIClient"),
|
|
740
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
741
|
-
window_service_spec=WindowServiceSpec(
|
|
742
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
743
|
-
),
|
|
744
|
-
max_sequence_length=2048,
|
|
745
|
-
max_request_length=2049,
|
|
746
|
-
),
|
|
747
|
-
ModelDeployment(
|
|
748
|
-
name="gooseai/gpt-j-6b",
|
|
749
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.goose_ai_client.GooseAIClient"),
|
|
750
|
-
tokenizer_name="EleutherAI/gpt-j-6B",
|
|
751
|
-
window_service_spec=WindowServiceSpec(
|
|
752
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
753
|
-
),
|
|
754
|
-
max_sequence_length=2048,
|
|
755
|
-
max_request_length=2049,
|
|
756
|
-
),
|
|
757
|
-
ModelDeployment(
|
|
758
|
-
name="huggingface/gpt2",
|
|
759
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.huggingface_client.HuggingFaceClient"),
|
|
760
|
-
tokenizer_name="huggingface/gpt2",
|
|
761
|
-
window_service_spec=WindowServiceSpec(
|
|
762
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
763
|
-
),
|
|
764
|
-
max_sequence_length=1024,
|
|
765
|
-
max_request_length=1025,
|
|
766
|
-
),
|
|
767
|
-
ModelDeployment(
|
|
768
|
-
name="huggingface/gpt-j-6b",
|
|
769
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.huggingface_client.HuggingFaceClient"),
|
|
770
|
-
tokenizer_name="EleutherAI/gpt-j-6B",
|
|
771
|
-
window_service_spec=WindowServiceSpec(
|
|
772
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
773
|
-
),
|
|
774
|
-
max_sequence_length=2048,
|
|
775
|
-
max_request_length=2049,
|
|
776
|
-
),
|
|
777
|
-
ModelDeployment(
|
|
778
|
-
name="huggingface/santacoder",
|
|
779
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.huggingface_client.HuggingFaceClient"),
|
|
780
|
-
tokenizer_name="bigcode/santacoder",
|
|
781
|
-
window_service_spec=WindowServiceSpec(
|
|
782
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
783
|
-
),
|
|
784
|
-
max_sequence_length=2048,
|
|
785
|
-
),
|
|
786
|
-
ModelDeployment(
|
|
787
|
-
name="huggingface/starcoder",
|
|
788
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.huggingface_client.HuggingFaceClient"),
|
|
789
|
-
tokenizer_name="bigcode/starcoder",
|
|
790
|
-
window_service_spec=WindowServiceSpec(
|
|
791
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
792
|
-
),
|
|
793
|
-
max_sequence_length=8192,
|
|
794
|
-
),
|
|
795
|
-
ModelDeployment(
|
|
796
|
-
name="together/t5-11b",
|
|
797
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
798
|
-
tokenizer_name="google/t5-11b",
|
|
799
|
-
window_service_spec=WindowServiceSpec(
|
|
800
|
-
class_name="helm.benchmark.window_services.t511b_window_service.T511bWindowService"
|
|
801
|
-
),
|
|
802
|
-
max_sequence_length=511,
|
|
803
|
-
),
|
|
804
|
-
ModelDeployment(
|
|
805
|
-
name="together/flan-t5-xxl",
|
|
806
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
807
|
-
tokenizer_name="google/flan-t5-xxl",
|
|
808
|
-
window_service_spec=WindowServiceSpec(
|
|
809
|
-
class_name="helm.benchmark.window_services.flan_t5_window_service.FlanT5WindowService"
|
|
810
|
-
),
|
|
811
|
-
max_sequence_length=511,
|
|
812
|
-
),
|
|
813
|
-
ModelDeployment(
|
|
814
|
-
name="together/ul2",
|
|
815
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
816
|
-
tokenizer_name="google/ul2",
|
|
817
|
-
window_service_spec=WindowServiceSpec(
|
|
818
|
-
class_name="helm.benchmark.window_services.ul2_window_service.UL2WindowService"
|
|
819
|
-
),
|
|
820
|
-
max_sequence_length=511,
|
|
821
|
-
),
|
|
822
|
-
ModelDeployment(
|
|
823
|
-
name="google/text-bison@001",
|
|
824
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vertexai_client.VertexAIClient"),
|
|
825
|
-
tokenizer_name="google/mt5-base",
|
|
826
|
-
window_service_spec=WindowServiceSpec(
|
|
827
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
828
|
-
),
|
|
829
|
-
max_sequence_length=6000,
|
|
830
|
-
max_sequence_and_generated_tokens_length=7000,
|
|
831
|
-
),
|
|
832
|
-
ModelDeployment(
|
|
833
|
-
name="google/text-bison-32k",
|
|
834
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vertexai_client.VertexAIClient"),
|
|
835
|
-
tokenizer_name="google/mt5-base",
|
|
836
|
-
window_service_spec=WindowServiceSpec(
|
|
837
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
838
|
-
),
|
|
839
|
-
max_sequence_length=32000,
|
|
840
|
-
max_sequence_and_generated_tokens_length=32000,
|
|
841
|
-
),
|
|
842
|
-
ModelDeployment(
|
|
843
|
-
name="google/text-unicorn@001",
|
|
844
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vertexai_client.VertexAIClient"),
|
|
845
|
-
tokenizer_name="google/mt5-base",
|
|
846
|
-
window_service_spec=WindowServiceSpec(
|
|
847
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
848
|
-
),
|
|
849
|
-
max_sequence_length=6000,
|
|
850
|
-
max_sequence_and_generated_tokens_length=7000,
|
|
851
|
-
),
|
|
852
|
-
ModelDeployment(
|
|
853
|
-
name="google/code-bison@001",
|
|
854
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vertexai_client.VertexAIClient"),
|
|
855
|
-
tokenizer_name="google/mt5-base",
|
|
856
|
-
window_service_spec=WindowServiceSpec(
|
|
857
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
858
|
-
),
|
|
859
|
-
max_sequence_length=6000,
|
|
860
|
-
max_sequence_and_generated_tokens_length=7000,
|
|
861
|
-
),
|
|
862
|
-
ModelDeployment(
|
|
863
|
-
name="google/code-bison-32k",
|
|
864
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vertexai_client.VertexAIClient"),
|
|
865
|
-
tokenizer_name="google/mt5-base",
|
|
866
|
-
window_service_spec=WindowServiceSpec(
|
|
867
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
868
|
-
),
|
|
869
|
-
max_sequence_length=32000,
|
|
870
|
-
max_sequence_and_generated_tokens_length=32000,
|
|
871
|
-
),
|
|
872
|
-
ModelDeployment(
|
|
873
|
-
name="together/h3-2.7b",
|
|
874
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
875
|
-
tokenizer_name="huggingface/gpt2",
|
|
876
|
-
window_service_spec=WindowServiceSpec(
|
|
877
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
878
|
-
),
|
|
879
|
-
max_sequence_length=1024,
|
|
880
|
-
max_request_length=1025,
|
|
881
|
-
),
|
|
882
|
-
ModelDeployment(
|
|
883
|
-
name="together/opt-175b",
|
|
884
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
885
|
-
tokenizer_name="facebook/opt-66b",
|
|
886
|
-
window_service_spec=WindowServiceSpec(
|
|
887
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
888
|
-
),
|
|
889
|
-
max_sequence_length=2048,
|
|
890
|
-
max_request_length=2049,
|
|
891
|
-
),
|
|
892
|
-
ModelDeployment(
|
|
893
|
-
name="together/opt-66b",
|
|
894
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
895
|
-
tokenizer_name="facebook/opt-66b",
|
|
896
|
-
window_service_spec=WindowServiceSpec(
|
|
897
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
898
|
-
),
|
|
899
|
-
max_sequence_length=2048,
|
|
900
|
-
max_request_length=2049,
|
|
901
|
-
),
|
|
902
|
-
ModelDeployment(
|
|
903
|
-
name="together/opt-6.7b",
|
|
904
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
905
|
-
tokenizer_name="facebook/opt-66b",
|
|
906
|
-
window_service_spec=WindowServiceSpec(
|
|
907
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
908
|
-
),
|
|
909
|
-
max_sequence_length=2048,
|
|
910
|
-
max_request_length=2049,
|
|
911
|
-
),
|
|
912
|
-
ModelDeployment(
|
|
913
|
-
name="together/opt-1.3b",
|
|
914
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
915
|
-
tokenizer_name="facebook/opt-66b",
|
|
916
|
-
window_service_spec=WindowServiceSpec(
|
|
917
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
918
|
-
),
|
|
919
|
-
max_sequence_length=2048,
|
|
920
|
-
max_request_length=2049,
|
|
921
|
-
),
|
|
922
|
-
ModelDeployment(
|
|
923
|
-
name="microsoft/TNLGv2_530B",
|
|
924
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.microsoft_client.MicrosoftClient"),
|
|
925
|
-
tokenizer_name="microsoft/gpt2",
|
|
926
|
-
window_service_spec=WindowServiceSpec(
|
|
927
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
928
|
-
),
|
|
929
|
-
max_sequence_length=2047,
|
|
930
|
-
max_request_length=2048,
|
|
931
|
-
),
|
|
932
|
-
ModelDeployment(
|
|
933
|
-
name="microsoft/TNLGv2_7B",
|
|
934
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.microsoft_client.MicrosoftClient"),
|
|
935
|
-
tokenizer_name="microsoft/gpt2",
|
|
936
|
-
window_service_spec=WindowServiceSpec(
|
|
937
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
938
|
-
),
|
|
939
|
-
max_sequence_length=2047,
|
|
940
|
-
max_request_length=2048,
|
|
941
|
-
),
|
|
942
|
-
ModelDeployment(
|
|
943
|
-
name="openai/davinci",
|
|
944
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
945
|
-
tokenizer_name="huggingface/gpt2",
|
|
946
|
-
window_service_spec=WindowServiceSpec(
|
|
947
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
948
|
-
),
|
|
949
|
-
max_sequence_length=2048,
|
|
950
|
-
max_request_length=2049,
|
|
951
|
-
),
|
|
952
|
-
ModelDeployment(
|
|
953
|
-
name="openai/curie",
|
|
954
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
955
|
-
tokenizer_name="huggingface/gpt2",
|
|
956
|
-
window_service_spec=WindowServiceSpec(
|
|
957
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
958
|
-
),
|
|
959
|
-
max_sequence_length=2048,
|
|
960
|
-
max_request_length=2049,
|
|
961
|
-
),
|
|
962
|
-
ModelDeployment(
|
|
963
|
-
name="openai/babbage",
|
|
964
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
965
|
-
tokenizer_name="huggingface/gpt2",
|
|
966
|
-
window_service_spec=WindowServiceSpec(
|
|
967
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
968
|
-
),
|
|
969
|
-
max_sequence_length=2048,
|
|
970
|
-
max_request_length=2049,
|
|
971
|
-
),
|
|
972
|
-
ModelDeployment(
|
|
973
|
-
name="openai/ada",
|
|
974
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
975
|
-
tokenizer_name="huggingface/gpt2",
|
|
976
|
-
window_service_spec=WindowServiceSpec(
|
|
977
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
978
|
-
),
|
|
979
|
-
max_sequence_length=2048,
|
|
980
|
-
max_request_length=2049,
|
|
981
|
-
),
|
|
982
|
-
ModelDeployment(
|
|
983
|
-
name="openai/text-davinci-003",
|
|
984
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
985
|
-
tokenizer_name="huggingface/gpt2",
|
|
986
|
-
window_service_spec=WindowServiceSpec(
|
|
987
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
988
|
-
),
|
|
989
|
-
max_sequence_length=4000,
|
|
990
|
-
max_request_length=4001,
|
|
991
|
-
),
|
|
992
|
-
ModelDeployment(
|
|
993
|
-
name="openai/text-davinci-002",
|
|
994
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
995
|
-
tokenizer_name="huggingface/gpt2",
|
|
996
|
-
window_service_spec=WindowServiceSpec(
|
|
997
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
998
|
-
),
|
|
999
|
-
max_sequence_length=4000,
|
|
1000
|
-
max_request_length=4001,
|
|
1001
|
-
),
|
|
1002
|
-
ModelDeployment(
|
|
1003
|
-
name="openai/text-davinci-001",
|
|
1004
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1005
|
-
tokenizer_name="huggingface/gpt2",
|
|
1006
|
-
window_service_spec=WindowServiceSpec(
|
|
1007
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1008
|
-
),
|
|
1009
|
-
max_sequence_length=2048,
|
|
1010
|
-
max_request_length=2049,
|
|
1011
|
-
),
|
|
1012
|
-
ModelDeployment(
|
|
1013
|
-
name="openai/text-curie-001",
|
|
1014
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1015
|
-
tokenizer_name="huggingface/gpt2",
|
|
1016
|
-
window_service_spec=WindowServiceSpec(
|
|
1017
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1018
|
-
),
|
|
1019
|
-
max_sequence_length=2048,
|
|
1020
|
-
max_request_length=2049,
|
|
1021
|
-
),
|
|
1022
|
-
ModelDeployment(
|
|
1023
|
-
name="openai/text-babbage-001",
|
|
1024
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1025
|
-
tokenizer_name="huggingface/gpt2",
|
|
1026
|
-
window_service_spec=WindowServiceSpec(
|
|
1027
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1028
|
-
),
|
|
1029
|
-
max_sequence_length=2048,
|
|
1030
|
-
max_request_length=2049,
|
|
1031
|
-
),
|
|
1032
|
-
ModelDeployment(
|
|
1033
|
-
name="openai/text-ada-001",
|
|
1034
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1035
|
-
tokenizer_name="huggingface/gpt2",
|
|
1036
|
-
window_service_spec=WindowServiceSpec(
|
|
1037
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1038
|
-
),
|
|
1039
|
-
max_sequence_length=2048,
|
|
1040
|
-
max_request_length=2049,
|
|
1041
|
-
),
|
|
1042
|
-
ModelDeployment(
|
|
1043
|
-
name="openai/code-davinci-002",
|
|
1044
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1045
|
-
tokenizer_name="huggingface/gpt2",
|
|
1046
|
-
window_service_spec=WindowServiceSpec(
|
|
1047
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1048
|
-
),
|
|
1049
|
-
max_sequence_length=4000,
|
|
1050
|
-
max_request_length=4001,
|
|
1051
|
-
),
|
|
1052
|
-
ModelDeployment(
|
|
1053
|
-
name="openai/code-davinci-001",
|
|
1054
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1055
|
-
tokenizer_name="huggingface/gpt2",
|
|
1056
|
-
window_service_spec=WindowServiceSpec(
|
|
1057
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1058
|
-
),
|
|
1059
|
-
max_sequence_length=2048,
|
|
1060
|
-
max_request_length=2049,
|
|
1061
|
-
),
|
|
1062
|
-
ModelDeployment(
|
|
1063
|
-
name="openai/code-cushman-001",
|
|
1064
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1065
|
-
tokenizer_name="huggingface/gpt2",
|
|
1066
|
-
window_service_spec=WindowServiceSpec(
|
|
1067
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1068
|
-
),
|
|
1069
|
-
max_sequence_length=2048,
|
|
1070
|
-
max_request_length=2049,
|
|
1071
|
-
),
|
|
1072
|
-
ModelDeployment(
|
|
1073
|
-
name="openai/gpt-4-1106-preview",
|
|
1074
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1075
|
-
tokenizer_name="openai/cl100k_base",
|
|
1076
|
-
window_service_spec=WindowServiceSpec(
|
|
1077
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1078
|
-
),
|
|
1079
|
-
max_sequence_length=128000,
|
|
1080
|
-
max_request_length=128001,
|
|
1081
|
-
),
|
|
1082
|
-
ModelDeployment(
|
|
1083
|
-
name="openai/gpt-4-0314",
|
|
1084
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1085
|
-
tokenizer_name="openai/cl100k_base",
|
|
1086
|
-
window_service_spec=WindowServiceSpec(
|
|
1087
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1088
|
-
),
|
|
1089
|
-
max_sequence_length=8192,
|
|
1090
|
-
max_request_length=8193,
|
|
1091
|
-
),
|
|
1092
|
-
ModelDeployment(
|
|
1093
|
-
name="openai/gpt-4-32k-0314",
|
|
1094
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1095
|
-
tokenizer_name="openai/cl100k_base",
|
|
1096
|
-
window_service_spec=WindowServiceSpec(
|
|
1097
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1098
|
-
),
|
|
1099
|
-
max_sequence_length=32768,
|
|
1100
|
-
max_request_length=32769,
|
|
1101
|
-
),
|
|
1102
|
-
ModelDeployment(
|
|
1103
|
-
name="openai/gpt-4-0613",
|
|
1104
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1105
|
-
tokenizer_name="openai/cl100k_base",
|
|
1106
|
-
window_service_spec=WindowServiceSpec(
|
|
1107
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1108
|
-
),
|
|
1109
|
-
max_sequence_length=8192,
|
|
1110
|
-
max_request_length=8193,
|
|
1111
|
-
),
|
|
1112
|
-
ModelDeployment(
|
|
1113
|
-
name="openai/gpt-4-32k-0613",
|
|
1114
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1115
|
-
tokenizer_name="openai/cl100k_base",
|
|
1116
|
-
window_service_spec=WindowServiceSpec(
|
|
1117
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1118
|
-
),
|
|
1119
|
-
max_sequence_length=32768,
|
|
1120
|
-
max_request_length=32769,
|
|
1121
|
-
),
|
|
1122
|
-
ModelDeployment(
|
|
1123
|
-
name="openai/gpt-3.5-turbo-0301",
|
|
1124
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1125
|
-
tokenizer_name="openai/cl100k_base",
|
|
1126
|
-
window_service_spec=WindowServiceSpec(
|
|
1127
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1128
|
-
),
|
|
1129
|
-
max_sequence_length=4000,
|
|
1130
|
-
max_request_length=4001,
|
|
1131
|
-
),
|
|
1132
|
-
ModelDeployment(
|
|
1133
|
-
name="openai/gpt-3.5-turbo-0613",
|
|
1134
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1135
|
-
tokenizer_name="openai/cl100k_base",
|
|
1136
|
-
window_service_spec=WindowServiceSpec(
|
|
1137
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1138
|
-
),
|
|
1139
|
-
max_sequence_length=4000,
|
|
1140
|
-
max_request_length=4001,
|
|
1141
|
-
),
|
|
1142
|
-
ModelDeployment(
|
|
1143
|
-
name="openai/gpt-3.5-turbo-16k-0613",
|
|
1144
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1145
|
-
tokenizer_name="openai/cl100k_base",
|
|
1146
|
-
window_service_spec=WindowServiceSpec(
|
|
1147
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1148
|
-
),
|
|
1149
|
-
max_sequence_length=16000,
|
|
1150
|
-
max_request_length=16001,
|
|
1151
|
-
),
|
|
1152
|
-
ModelDeployment(
|
|
1153
|
-
name="openai/text-similarity-davinci-001",
|
|
1154
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1155
|
-
tokenizer_name="huggingface/gpt2",
|
|
1156
|
-
window_service_spec=WindowServiceSpec(
|
|
1157
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1158
|
-
),
|
|
1159
|
-
max_sequence_length=2048,
|
|
1160
|
-
max_request_length=2049,
|
|
1161
|
-
),
|
|
1162
|
-
ModelDeployment(
|
|
1163
|
-
name="openai/text-similarity-curie-001",
|
|
1164
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1165
|
-
tokenizer_name="huggingface/gpt2",
|
|
1166
|
-
window_service_spec=WindowServiceSpec(
|
|
1167
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1168
|
-
),
|
|
1169
|
-
max_sequence_length=2048,
|
|
1170
|
-
max_request_length=2049,
|
|
1171
|
-
),
|
|
1172
|
-
ModelDeployment(
|
|
1173
|
-
name="openai/text-similarity-babbage-001",
|
|
1174
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1175
|
-
tokenizer_name="huggingface/gpt2",
|
|
1176
|
-
window_service_spec=WindowServiceSpec(
|
|
1177
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1178
|
-
),
|
|
1179
|
-
max_sequence_length=2048,
|
|
1180
|
-
max_request_length=2049,
|
|
1181
|
-
),
|
|
1182
|
-
ModelDeployment(
|
|
1183
|
-
name="openai/text-similarity-ada-001",
|
|
1184
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1185
|
-
tokenizer_name="huggingface/gpt2",
|
|
1186
|
-
window_service_spec=WindowServiceSpec(
|
|
1187
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1188
|
-
),
|
|
1189
|
-
max_sequence_length=2048,
|
|
1190
|
-
max_request_length=2049,
|
|
1191
|
-
),
|
|
1192
|
-
ModelDeployment(
|
|
1193
|
-
name="openai/text-embedding-ada-002",
|
|
1194
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.openai_client.OpenAIClient"),
|
|
1195
|
-
tokenizer_name="huggingface/gpt2",
|
|
1196
|
-
window_service_spec=WindowServiceSpec(
|
|
1197
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1198
|
-
),
|
|
1199
|
-
max_sequence_length=2048,
|
|
1200
|
-
max_request_length=2049,
|
|
1201
|
-
),
|
|
1202
|
-
ModelDeployment(
|
|
1203
|
-
name="together/gpt-jt-6b-v1",
|
|
1204
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1205
|
-
tokenizer_name="EleutherAI/gpt-j-6B",
|
|
1206
|
-
window_service_spec=WindowServiceSpec(
|
|
1207
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1208
|
-
),
|
|
1209
|
-
max_sequence_length=2048,
|
|
1210
|
-
max_request_length=2049,
|
|
1211
|
-
),
|
|
1212
|
-
ModelDeployment(
|
|
1213
|
-
name="together/gpt-neoxt-chat-base-20b",
|
|
1214
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1215
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1216
|
-
window_service_spec=WindowServiceSpec(
|
|
1217
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1218
|
-
),
|
|
1219
|
-
max_sequence_length=2048,
|
|
1220
|
-
max_request_length=2049,
|
|
1221
|
-
),
|
|
1222
|
-
ModelDeployment(
|
|
1223
|
-
name="together/redpajama-incite-base-3b-v1",
|
|
1224
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1225
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1226
|
-
window_service_spec=WindowServiceSpec(
|
|
1227
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1228
|
-
),
|
|
1229
|
-
max_sequence_length=2048,
|
|
1230
|
-
max_request_length=2049,
|
|
1231
|
-
),
|
|
1232
|
-
ModelDeployment(
|
|
1233
|
-
name="together/redpajama-incite-instruct-3b-v1",
|
|
1234
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1235
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1236
|
-
window_service_spec=WindowServiceSpec(
|
|
1237
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1238
|
-
),
|
|
1239
|
-
max_sequence_length=2048,
|
|
1240
|
-
max_request_length=2049,
|
|
1241
|
-
),
|
|
1242
|
-
ModelDeployment(
|
|
1243
|
-
name="together/redpajama-incite-base-7b",
|
|
1244
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1245
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1246
|
-
window_service_spec=WindowServiceSpec(
|
|
1247
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1248
|
-
),
|
|
1249
|
-
max_sequence_length=2048,
|
|
1250
|
-
max_request_length=2049,
|
|
1251
|
-
),
|
|
1252
|
-
ModelDeployment(
|
|
1253
|
-
name="together/redpajama-incite-instruct-7b",
|
|
1254
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1255
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1256
|
-
window_service_spec=WindowServiceSpec(
|
|
1257
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1258
|
-
),
|
|
1259
|
-
max_sequence_length=2048,
|
|
1260
|
-
max_request_length=2049,
|
|
1261
|
-
),
|
|
1262
|
-
ModelDeployment(
|
|
1263
|
-
name="together/glm",
|
|
1264
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1265
|
-
tokenizer_name="TsinghuaKEG/ice",
|
|
1266
|
-
window_service_spec=WindowServiceSpec(
|
|
1267
|
-
class_name="helm.benchmark.window_services.ice_window_service.ICEWindowService"
|
|
1268
|
-
),
|
|
1269
|
-
max_sequence_length=2048,
|
|
1270
|
-
max_request_length=2049,
|
|
1271
|
-
),
|
|
1272
|
-
ModelDeployment(
|
|
1273
|
-
name="writer/palmyra-base",
|
|
1274
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.palmyra_client.PalmyraClient"),
|
|
1275
|
-
tokenizer_name="writer/gpt2",
|
|
1276
|
-
window_service_spec=WindowServiceSpec(
|
|
1277
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1278
|
-
),
|
|
1279
|
-
max_sequence_length=2048,
|
|
1280
|
-
max_sequence_and_generated_tokens_length=2048,
|
|
1281
|
-
),
|
|
1282
|
-
ModelDeployment(
|
|
1283
|
-
name="writer/palmyra-large",
|
|
1284
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.palmyra_client.PalmyraClient"),
|
|
1285
|
-
tokenizer_name="writer/gpt2",
|
|
1286
|
-
window_service_spec=WindowServiceSpec(
|
|
1287
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1288
|
-
),
|
|
1289
|
-
max_sequence_length=2048,
|
|
1290
|
-
max_sequence_and_generated_tokens_length=2048,
|
|
1291
|
-
),
|
|
1292
|
-
ModelDeployment(
|
|
1293
|
-
name="writer/palmyra-instruct-30",
|
|
1294
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.palmyra_client.PalmyraClient"),
|
|
1295
|
-
tokenizer_name="writer/gpt2",
|
|
1296
|
-
window_service_spec=WindowServiceSpec(
|
|
1297
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1298
|
-
),
|
|
1299
|
-
max_sequence_length=2048,
|
|
1300
|
-
max_sequence_and_generated_tokens_length=2048,
|
|
1301
|
-
),
|
|
1302
|
-
ModelDeployment(
|
|
1303
|
-
name="writer/palmyra-e",
|
|
1304
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.palmyra_client.PalmyraClient"),
|
|
1305
|
-
tokenizer_name="writer/gpt2",
|
|
1306
|
-
window_service_spec=WindowServiceSpec(
|
|
1307
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1308
|
-
),
|
|
1309
|
-
max_sequence_length=2048,
|
|
1310
|
-
max_sequence_and_generated_tokens_length=2048,
|
|
1311
|
-
),
|
|
1312
|
-
ModelDeployment(
|
|
1313
|
-
name="writer/silk-road",
|
|
1314
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.palmyra_client.PalmyraClient"),
|
|
1315
|
-
tokenizer_name="writer/gpt2",
|
|
1316
|
-
window_service_spec=WindowServiceSpec(
|
|
1317
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1318
|
-
),
|
|
1319
|
-
max_sequence_length=8192,
|
|
1320
|
-
max_sequence_and_generated_tokens_length=8192,
|
|
1321
|
-
),
|
|
1322
|
-
ModelDeployment(
|
|
1323
|
-
name="writer/palmyra-x",
|
|
1324
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.palmyra_client.PalmyraClient"),
|
|
1325
|
-
tokenizer_name="writer/gpt2",
|
|
1326
|
-
window_service_spec=WindowServiceSpec(
|
|
1327
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1328
|
-
),
|
|
1329
|
-
max_sequence_length=8192,
|
|
1330
|
-
max_sequence_and_generated_tokens_length=8192,
|
|
1331
|
-
),
|
|
1332
|
-
ModelDeployment(
|
|
1333
|
-
name="together/yalm",
|
|
1334
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1335
|
-
tokenizer_name="Yandex/yalm",
|
|
1336
|
-
window_service_spec=WindowServiceSpec(
|
|
1337
|
-
class_name="helm.benchmark.window_services.yalm_window_service.YaLMWindowService"
|
|
1338
|
-
),
|
|
1339
|
-
max_sequence_length=2048,
|
|
1340
|
-
max_request_length=2049,
|
|
1341
|
-
),
|
|
1342
|
-
ModelDeployment(
|
|
1343
|
-
name="nvidia/megatron-gpt2",
|
|
1344
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.megatron_client.MegatronClient"),
|
|
1345
|
-
tokenizer_name="huggingface/gpt2",
|
|
1346
|
-
window_service_spec=WindowServiceSpec(
|
|
1347
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1348
|
-
),
|
|
1349
|
-
max_sequence_length=1024,
|
|
1350
|
-
),
|
|
1351
|
-
ModelDeployment(
|
|
1352
|
-
name="together/dolly-v2-3b",
|
|
1353
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1354
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1355
|
-
window_service_spec=WindowServiceSpec(
|
|
1356
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1357
|
-
),
|
|
1358
|
-
max_sequence_length=2048,
|
|
1359
|
-
max_request_length=2049,
|
|
1360
|
-
),
|
|
1361
|
-
ModelDeployment(
|
|
1362
|
-
name="together/dolly-v2-7b",
|
|
1363
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1364
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1365
|
-
window_service_spec=WindowServiceSpec(
|
|
1366
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1367
|
-
),
|
|
1368
|
-
max_sequence_length=2048,
|
|
1369
|
-
max_request_length=2049,
|
|
1370
|
-
),
|
|
1371
|
-
ModelDeployment(
|
|
1372
|
-
name="together/dolly-v2-12b",
|
|
1373
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1374
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1375
|
-
window_service_spec=WindowServiceSpec(
|
|
1376
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1377
|
-
),
|
|
1378
|
-
max_sequence_length=2048,
|
|
1379
|
-
max_request_length=2049,
|
|
1380
|
-
),
|
|
1381
|
-
ModelDeployment(
|
|
1382
|
-
name="together/stablelm-base-alpha-3b",
|
|
1383
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1384
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1385
|
-
window_service_spec=WindowServiceSpec(
|
|
1386
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1387
|
-
),
|
|
1388
|
-
max_sequence_length=4096,
|
|
1389
|
-
max_request_length=4097,
|
|
1390
|
-
),
|
|
1391
|
-
ModelDeployment(
|
|
1392
|
-
name="together/stablelm-base-alpha-7b",
|
|
1393
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.together_client.TogetherClient"),
|
|
1394
|
-
tokenizer_name="EleutherAI/gpt-neox-20b",
|
|
1395
|
-
window_service_spec=WindowServiceSpec(
|
|
1396
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1397
|
-
),
|
|
1398
|
-
max_sequence_length=4096,
|
|
1399
|
-
max_request_length=4097,
|
|
1400
|
-
),
|
|
1401
|
-
ModelDeployment(
|
|
1402
|
-
name="lightningai/lit-gpt",
|
|
1403
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.lit_gpt_client.LitGPTClient", args={}),
|
|
1404
|
-
model_name=None,
|
|
1405
|
-
tokenizer_name="lightningai/lit-gpt",
|
|
1406
|
-
window_service_spec=WindowServiceSpec(
|
|
1407
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService", args={}
|
|
1408
|
-
),
|
|
1409
|
-
max_sequence_length=2048,
|
|
1410
|
-
max_request_length=None,
|
|
1411
|
-
max_sequence_and_generated_tokens_length=None,
|
|
1412
|
-
),
|
|
1413
|
-
ModelDeployment(
|
|
1414
|
-
name="HuggingFaceM4/idefics-9b",
|
|
1415
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vision_language.idefics_client.IDEFICSClient"),
|
|
1416
|
-
tokenizer_name="HuggingFaceM4/idefics-9b",
|
|
1417
|
-
window_service_spec=WindowServiceSpec(
|
|
1418
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1419
|
-
),
|
|
1420
|
-
max_sequence_length=2048,
|
|
1421
|
-
),
|
|
1422
|
-
ModelDeployment(
|
|
1423
|
-
name="HuggingFaceM4/idefics-9b-instruct",
|
|
1424
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vision_language.idefics_client.IDEFICSClient"),
|
|
1425
|
-
tokenizer_name="HuggingFaceM4/idefics-9b-instruct",
|
|
1426
|
-
window_service_spec=WindowServiceSpec(
|
|
1427
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1428
|
-
),
|
|
1429
|
-
max_sequence_length=2048,
|
|
1430
|
-
),
|
|
1431
|
-
ModelDeployment(
|
|
1432
|
-
name="HuggingFaceM4/idefics-80b",
|
|
1433
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vision_language.idefics_client.IDEFICSClient"),
|
|
1434
|
-
tokenizer_name="HuggingFaceM4/idefics-80b",
|
|
1435
|
-
window_service_spec=WindowServiceSpec(
|
|
1436
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1437
|
-
),
|
|
1438
|
-
max_sequence_length=2048,
|
|
1439
|
-
),
|
|
1440
|
-
ModelDeployment(
|
|
1441
|
-
name="HuggingFaceM4/idefics-80b-instruct",
|
|
1442
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.vision_language.idefics_client.IDEFICSClient"),
|
|
1443
|
-
tokenizer_name="HuggingFaceM4/idefics-80b-instruct",
|
|
1444
|
-
window_service_spec=WindowServiceSpec(
|
|
1445
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1446
|
-
),
|
|
1447
|
-
max_sequence_length=2048,
|
|
1448
|
-
),
|
|
1449
|
-
ModelDeployment(
|
|
1450
|
-
name="simple/model1",
|
|
1451
|
-
client_spec=ClientSpec(class_name="helm.proxy.clients.simple_client.SimpleClient"),
|
|
1452
|
-
tokenizer_name="simple/model1",
|
|
1453
|
-
window_service_spec=WindowServiceSpec(
|
|
1454
|
-
class_name="helm.benchmark.window_services.default_window_service.DefaultWindowService"
|
|
1455
|
-
),
|
|
1456
|
-
max_sequence_length=2048,
|
|
1457
|
-
),
|
|
1458
|
-
]
|
|
1459
|
-
|
|
1460
|
-
|
|
1461
|
-
_INT_MAX: int = 2**31 - 1
|
|
1462
|
-
|
|
1463
|
-
|
|
1464
|
-
def _full_class_name(obj: Any) -> str:
|
|
1465
|
-
return f"{obj.__class__.__module__}.{obj.__class__.__name__}"
|
|
1466
|
-
|
|
1467
|
-
|
|
1468
|
-
# HACK: This looks like it should be done in a setup_class()
|
|
1469
|
-
# for the test below but apparently pytest first check the parametrize
|
|
1470
|
-
# before running the setup_class().
|
|
1471
|
-
# Therefore ALL_MODEL_DEPLOYMENTS is empty and no test would be run,
|
|
1472
|
-
# so we need to do this here.
|
|
1473
|
-
register_builtin_configs_from_helm_package()
|
|
1474
|
-
|
|
1475
|
-
|
|
1476
|
-
class TestModelProperties:
|
|
1477
|
-
@pytest.mark.parametrize("deployment_name", [deployment.name for deployment in _BUILT_IN_MODEL_DEPLOYMENTS])
|
|
1478
|
-
def test_models_has_window_service(self, deployment_name: str):
|
|
1479
|
-
auto_client = AutoClient(defaultdict(str), "", "")
|
|
1480
|
-
auto_tokenizer = AutoTokenizer(defaultdict(str), "", "")
|
|
1481
|
-
model_deployments = {
|
|
1482
|
-
model_deployment.name: model_deployment for model_deployment in _BUILT_IN_MODEL_DEPLOYMENTS
|
|
1483
|
-
}
|
|
1484
|
-
tokenizer_configs = {
|
|
1485
|
-
tokenizer_config.name: tokenizer_config for tokenizer_config in _BUILT_IN_TOKENIZER_CONFIGS
|
|
1486
|
-
}
|
|
1487
|
-
with TemporaryDirectory() as tmpdir:
|
|
1488
|
-
tokenizer_service = get_tokenizer_service(tmpdir)
|
|
1489
|
-
deployment: ModelDeployment = get_model_deployment(deployment_name)
|
|
1490
|
-
model: ModelMetadata = get_model_metadata(deployment.model_name or deployment_name)
|
|
1491
|
-
# Can't test lit-gpt client because it requires manual dependencies
|
|
1492
|
-
if "lit-gpt" in model.name:
|
|
1493
|
-
return
|
|
1494
|
-
|
|
1495
|
-
# Can't test Llama 2 because it requires Hugging Face credentials
|
|
1496
|
-
if "llama-2-" in model.name:
|
|
1497
|
-
return
|
|
1498
|
-
|
|
1499
|
-
# Can't test Vertex AI because it requires Google credentials
|
|
1500
|
-
if "text-bison" in model.name or "text-unicorn" in model.name:
|
|
1501
|
-
return
|
|
1502
|
-
|
|
1503
|
-
client = auto_client._get_client(deployment_name)
|
|
1504
|
-
window_service = WindowServiceFactory.get_window_service(deployment_name, tokenizer_service)
|
|
1505
|
-
tokenizer_name = window_service.tokenizer_name
|
|
1506
|
-
tokenizer = auto_tokenizer._get_tokenizer(tokenizer_name)
|
|
1507
|
-
|
|
1508
|
-
client_class_name = _full_class_name(client)
|
|
1509
|
-
tokenizer_class_name = _full_class_name(tokenizer)
|
|
1510
|
-
window_service_class_name = _full_class_name(window_service)
|
|
1511
|
-
|
|
1512
|
-
prefix_token = window_service.prefix_token
|
|
1513
|
-
end_of_text_token = window_service.end_of_text_token
|
|
1514
|
-
|
|
1515
|
-
max_sequence_length = window_service.max_sequence_length
|
|
1516
|
-
max_request_length = (
|
|
1517
|
-
window_service.max_request_length
|
|
1518
|
-
if window_service.max_request_length != window_service.max_sequence_length
|
|
1519
|
-
else None
|
|
1520
|
-
)
|
|
1521
|
-
max_sequence_and_generated_tokens_length = (
|
|
1522
|
-
window_service.max_sequence_and_generated_tokens_length
|
|
1523
|
-
if window_service.max_sequence_and_generated_tokens_length != _INT_MAX
|
|
1524
|
-
else None
|
|
1525
|
-
)
|
|
1526
|
-
|
|
1527
|
-
model_deployment = ModelDeployment(
|
|
1528
|
-
name=deployment_name,
|
|
1529
|
-
client_spec=ClientSpec(class_name=client_class_name),
|
|
1530
|
-
tokenizer_name=tokenizer_name,
|
|
1531
|
-
window_service_spec=WindowServiceSpec(class_name=window_service_class_name),
|
|
1532
|
-
max_sequence_length=max_sequence_length,
|
|
1533
|
-
max_request_length=max_request_length,
|
|
1534
|
-
max_sequence_and_generated_tokens_length=max_sequence_and_generated_tokens_length,
|
|
1535
|
-
)
|
|
1536
|
-
tokenizer_config = TokenizerConfig(
|
|
1537
|
-
name=tokenizer_name,
|
|
1538
|
-
tokenizer_spec=TokenizerSpec(class_name=tokenizer_class_name),
|
|
1539
|
-
end_of_text_token=end_of_text_token,
|
|
1540
|
-
prefix_token=prefix_token,
|
|
1541
|
-
)
|
|
1542
|
-
# NOTE: To generate the _BUILT_IN_MODEL_DEPLOYMENT and _BUILT_IN_TOKENIZER_CONFIGS lists above,
|
|
1543
|
-
# print tokenizer_config and model_deployment here.
|
|
1544
|
-
|
|
1545
|
-
# Cannot directly compare model_deployment and deployment because they will have different values
|
|
1546
|
-
# for deprecated for example as it is not specified in this file.
|
|
1547
|
-
# In stead compare all the fields defined here.
|
|
1548
|
-
test_model_deployment: ModelDeployment = model_deployments[deployment_name]
|
|
1549
|
-
assert model_deployment.name == test_model_deployment.name
|
|
1550
|
-
assert model_deployment.client_spec.class_name == test_model_deployment.client_spec.class_name
|
|
1551
|
-
assert model_deployment.tokenizer_name == test_model_deployment.tokenizer_name
|
|
1552
|
-
if model_deployment.window_service_spec:
|
|
1553
|
-
assert test_model_deployment.window_service_spec
|
|
1554
|
-
assert (
|
|
1555
|
-
model_deployment.window_service_spec.class_name
|
|
1556
|
-
== test_model_deployment.window_service_spec.class_name
|
|
1557
|
-
)
|
|
1558
|
-
else:
|
|
1559
|
-
assert not test_model_deployment.window_service_spec
|
|
1560
|
-
assert model_deployment.max_sequence_length == test_model_deployment.max_sequence_length
|
|
1561
|
-
assert model_deployment.max_request_length == test_model_deployment.max_request_length
|
|
1562
|
-
assert (
|
|
1563
|
-
model_deployment.max_sequence_and_generated_tokens_length
|
|
1564
|
-
== test_model_deployment.max_sequence_and_generated_tokens_length
|
|
1565
|
-
)
|
|
1566
|
-
# DefaultWindowService overrides the huggingface/gpt2 tokenizer with different special tokens,
|
|
1567
|
-
# so there are currently two tokenizers named huggingface/gpt2
|
|
1568
|
-
# TODO: Give DefaultWindowService's tokenizer a different name e.g. writer/palmyra
|
|
1569
|
-
if tokenizer_name != "huggingface/gpt2":
|
|
1570
|
-
assert tokenizer_configs[tokenizer_name] == tokenizer_config
|