wehub-resource-sync
91 lines
2.1 KiB
Python
91 lines
2.1 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm.benchmarks.datasets.datasets import (
|
|
DEFAULT_NUM_PROMPTS,
|
|
AIMODataset,
|
|
ASRDataset,
|
|
BenchmarkDataset,
|
|
BFCLDataset,
|
|
BlazeditDataset,
|
|
BurstGPTDataset,
|
|
ConversationDataset,
|
|
CustomAudioDataset,
|
|
CustomDataset,
|
|
CustomImageDataset,
|
|
HuggingFaceDataset,
|
|
InstructCoderDataset,
|
|
MLPerfDataset,
|
|
MMStarDataset,
|
|
MMVUDataset,
|
|
MTBenchDataset,
|
|
MultiModalConversationDataset,
|
|
NextEditPredictionDataset,
|
|
PrefixRepetitionRandomDataset,
|
|
RandomDataset,
|
|
RandomDatasetForReranking,
|
|
RandomMultiModalDataset,
|
|
SampleRequest,
|
|
ShareGPTDataset,
|
|
SonnetDataset,
|
|
SpecBench,
|
|
VisionArenaDataset,
|
|
add_dataset_parser,
|
|
add_random_dataset_base_args,
|
|
add_random_multimodal_dataset_args,
|
|
gen_prompt_decode_to_target_len,
|
|
get_samples,
|
|
is_valid_sequence,
|
|
lora_path_on_disk,
|
|
lora_tokenizer_cache,
|
|
process_audio,
|
|
process_image,
|
|
process_video,
|
|
zeta_prompt,
|
|
)
|
|
from vllm.benchmarks.datasets.utils import RangeRatio
|
|
|
|
__all__ = [
|
|
"DEFAULT_NUM_PROMPTS",
|
|
"AIMODataset",
|
|
"ASRDataset",
|
|
"BenchmarkDataset",
|
|
"BFCLDataset",
|
|
"BlazeditDataset",
|
|
"BurstGPTDataset",
|
|
"ConversationDataset",
|
|
"CustomDataset",
|
|
"CustomAudioDataset",
|
|
"CustomImageDataset",
|
|
"HuggingFaceDataset",
|
|
"InstructCoderDataset",
|
|
"MLPerfDataset",
|
|
"MMStarDataset",
|
|
"MMVUDataset",
|
|
"MTBenchDataset",
|
|
"MultiModalConversationDataset",
|
|
"NextEditPredictionDataset",
|
|
"PrefixRepetitionRandomDataset",
|
|
"RandomDataset",
|
|
"RandomDatasetForReranking",
|
|
"RandomMultiModalDataset",
|
|
"SampleRequest",
|
|
"ShareGPTDataset",
|
|
"SonnetDataset",
|
|
"SpecBench",
|
|
"VisionArenaDataset",
|
|
"add_dataset_parser",
|
|
"add_random_dataset_base_args",
|
|
"add_random_multimodal_dataset_args",
|
|
"gen_prompt_decode_to_target_len",
|
|
"get_samples",
|
|
"is_valid_sequence",
|
|
"lora_path_on_disk",
|
|
"lora_tokenizer_cache",
|
|
"process_audio",
|
|
"process_image",
|
|
"process_video",
|
|
"RangeRatio",
|
|
"zeta_prompt",
|
|
]
|