| # testing | |
| pytest | |
| tensorizer>=2.9.0 | |
| pytest-forked | |
| pytest-asyncio | |
| pytest-rerunfailures | |
| pytest-shard | |
| # testing utils | |
| awscli | |
| einops # required for MPT, qwen-vl and Mamba | |
| httpx | |
| librosa # required for audio test | |
| opencv-python # required for video test | |
| peft | |
| requests | |
| ray[adag]==2.35 | |
| sentence-transformers # required for embedding | |
| soundfile # required for audio test | |
| compressed-tensors==0.4.0 # required for compressed-tensors | |
| timm # required for internvl test | |
| transformers_stream_generator # required for qwen-vl test | |
| matplotlib # required for qwen-vl test | |
| datamodel_code_generator # required for minicpm3 test | |
| # TODO: Add this after fully implementing llava(mantis) | |
| # git+https://github.com/TIGER-AI-Lab/Mantis.git # required for llava(mantis) test | |
| # Benchmarking | |
| aiohttp | |
| # quantization | |
| bitsandbytes>=0.44.0 | |
| buildkite-test-collector==0.1.8 | |