50 lines
1.4 KiB
Plaintext
50 lines
1.4 KiB
Plaintext
# testing
|
|
pytest
|
|
tensorizer>=2.9.0
|
|
pytest-forked
|
|
pytest-asyncio
|
|
pytest-rerunfailures
|
|
pytest-shard
|
|
pytest-timeout
|
|
|
|
# testing utils
|
|
awscli
|
|
backoff # required for phi4mm test
|
|
einops # required for MPT, qwen-vl and Mamba
|
|
httpx
|
|
librosa # required for audio tests
|
|
vector_quantize_pytorch # required for minicpmo_26 test
|
|
vocos # required for minicpmo_26 test
|
|
peft
|
|
pqdm
|
|
ray[cgraph]>=2.43.0, !=2.44.* # Ray Compiled Graph, required by pipeline parallelism tests
|
|
sentence-transformers # required for embedding tests
|
|
soundfile # required for audio tests
|
|
jiwer # required for audio tests
|
|
timm # required for internvl test
|
|
torch==2.6.0
|
|
torchaudio==2.6.0
|
|
torchvision==0.21.0
|
|
transformers_stream_generator # required for qwen-vl test
|
|
matplotlib # required for qwen-vl test
|
|
mistral_common[opencv] >= 1.5.4 # required for pixtral test
|
|
num2words # required for smolvlm test
|
|
opencv-python-headless >= 4.11.0 # required for video test
|
|
datamodel_code_generator # required for minicpm3 test
|
|
lm-eval[api]==0.4.8 # required for model evaluation test
|
|
transformers==4.51.1
|
|
huggingface-hub[hf_xet]>=0.30.0 # Required for Xet downloads.
|
|
# quantization
|
|
bitsandbytes>=0.45.3
|
|
buildkite-test-collector==0.1.9
|
|
|
|
genai_perf==0.0.8
|
|
tritonclient==2.51.0
|
|
|
|
numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
|
|
numba == 0.61.2; python_version > '3.9'
|
|
numpy
|
|
runai-model-streamer==0.11.0
|
|
runai-model-streamer-s3==0.11.0
|
|
fastsafetensors>=0.1.10
|