| 1 | # testing |
| 2 | pytest |
| 3 | tensorizer==2.10.1 |
| 4 | pytest-forked |
| 5 | pytest-asyncio |
| 6 | pytest-rerunfailures |
| 7 | pytest-shard |
| 8 | pytest-timeout |
| 9 | |
| 10 | # testing utils |
| 11 | backoff # required for phi4mm test |
| 12 | blobfile # required for kimi-vl test |
| 13 | einops # required for qwen-vl and Mamba |
| 14 | httpx |
| 15 | librosa # required for audio tests |
| 16 | vocos # required for minicpmo_26 test |
| 17 | peft |
| 18 | pqdm |
| 19 | ray[cgraph,default]>=2.56.0,<2.57 # Ray Compiled Graph, required by pipeline parallelism tests; sharded-RDT weight transfer needs Ray Direct Transport (>=2.56) |
| 20 | sentence-transformers>=5.2.0 # required for embedding tests |
| 21 | soundfile # required for audio tests |
| 22 | jiwer # required for audio tests |
| 23 | timm # required for internvl test |
| 24 | transformers_stream_generator # required for qwen-vl test |
| 25 | matplotlib # required for qwen-vl test |
| 26 | mistral_common[image,audio] >= 1.11.6 # required for voxtral test |
| 27 | num2words # required for smolvlm test |
| 28 | opencv-python-headless >= 4.13.0 # required for video test |
| 29 | datamodel_code_generator>=0.64.0 # required for minicpm3 test |
| 30 | lm-eval[api]>=0.4.12 # required for model evaluation test |
| 31 | mteb[bm25s]>=2, <3 # required for mteb test |
| 32 | transformers==5.17.0 |
| 33 | schemathesis>=4.0.0 # Required for openai schema test. |
| 34 | # quantization |
| 35 | buildkite-test-collector==0.1.9 |
| 36 | |
| 37 | |
| 38 | genai_perf>=0.0.8 |
| 39 | tritonclient>=2.51.0 |
| 40 | |
| 41 | numba == 0.65.0 # Required for N-gram speculative decoding |
| 42 | numpy |
| 43 | runai-model-streamer[s3,gcs,azure]==0.15.7 |
| 44 | fastsafetensors>=0.3.3 |
| 45 | instanttensor>=0.1.9 |
| 46 | pydantic>=2.12 # 2.11 leads to error on python 3.13 |