38 lines
865 B
Plaintext
38 lines
865 B
Plaintext
|
# 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 tests
|
||
|
|
opencv-python # required for video tests
|
||
|
|
peft
|
||
|
|
requests
|
||
|
|
ray[adag]==2.35
|
||
|
|
sentence-transformers # required for embedding
|
||
|
|
soundfile # required for audio test
|
||
|
|
timm # required for internvl test
|
||
|
|
torch==2.5.0
|
||
|
|
transformers_stream_generator # required for qwen-vl test
|
||
|
|
matplotlib # required for qwen-vl test
|
||
|
|
datamodel_code_generator # required for minicpm3 test
|
||
|
|
lm-eval[api]==0.4.4 # required for model evaluation 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
|
||
|
|
|
||
|
|
numpy < 2.0.0
|