mirror of
https://github.com/vllm-project/vllm.git
synced 2025-12-06 06:53:12 +08:00
60 lines
2.0 KiB
Plaintext
60 lines
2.0 KiB
Plaintext
# testing
|
|
pytest
|
|
tensorizer==2.10.1
|
|
pytest-forked
|
|
pytest-asyncio
|
|
pytest-rerunfailures
|
|
pytest-shard
|
|
pytest-timeout
|
|
pytest-cov
|
|
|
|
# testing utils
|
|
backoff # required for phi4mm test
|
|
blobfile # required for kimi-vl test
|
|
einops # required for MPT, qwen-vl
|
|
httpx
|
|
librosa # required for audio tests
|
|
vector_quantize_pytorch # required for minicpmo_26 test
|
|
vocos # required for minicpmo_26 test
|
|
peft>=0.15.0 # required for phi-4-mm test
|
|
pqdm
|
|
ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests
|
|
sentence-transformers # required for embedding tests
|
|
soundfile # required for audio tests
|
|
jiwer # required for audio tests
|
|
tblib # for pickling test exceptions
|
|
timm >=1.0.17 # required for internvl and gemma3n-mm test
|
|
torch==2.9.0
|
|
torchaudio==2.9.0
|
|
torchvision==0.24.0
|
|
transformers_stream_generator # required for qwen-vl test
|
|
matplotlib # required for qwen-vl test
|
|
mistral_common[image,audio] >= 1.8.5 # required for voxtral test
|
|
num2words # required for smolvlm test
|
|
open_clip_torch==2.32.0 # Required for nemotron_vl test
|
|
opencv-python-headless >= 4.11.0 # required for video test
|
|
datamodel_code_generator # required for minicpm3 test
|
|
# TODO: Use lm-eval[api]==0.4.10 once released
|
|
lm-eval[api] @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d # required for model evaluation test
|
|
mteb[bm25s]>=2, <3 # required for mteb test
|
|
transformers==4.57.3
|
|
tokenizers==0.22.0
|
|
schemathesis>=3.39.15 # Required for openai schema test.
|
|
# quantization
|
|
bitsandbytes==0.46.1
|
|
buildkite-test-collector==0.1.9
|
|
|
|
|
|
genai_perf==0.0.8
|
|
tritonclient==2.51.0
|
|
|
|
arctic-inference == 0.1.1 # Required for suffix decoding test
|
|
numba == 0.61.2 # Required for N-gram speculative decoding
|
|
numpy
|
|
runai-model-streamer[s3,gcs]==0.15.3
|
|
fastsafetensors>=0.1.10
|
|
pydantic>=2.12 # 2.11 leads to error on python 3.13
|
|
decord==0.6.0
|
|
terratorch @ git+https://github.com/IBM/terratorch.git@1.1.rc3 # required for PrithviMAE test
|
|
gpt-oss >= 0.0.7; python_version > '3.11'
|