2024-08-02 13:55:40 -07:00
|
|
|
# Needed for Ray accelerated DAG tests
|
|
|
|
|
-r requirements-adag.txt
|
|
|
|
|
|
2024-06-13 11:22:41 -07:00
|
|
|
# testing
|
|
|
|
|
pytest
|
|
|
|
|
tensorizer>=2.9.0
|
|
|
|
|
pytest-forked
|
|
|
|
|
pytest-asyncio
|
|
|
|
|
pytest-rerunfailures
|
|
|
|
|
pytest-shard
|
|
|
|
|
|
|
|
|
|
# testing utils
|
|
|
|
|
awscli
|
2024-08-18 17:43:46 -06:00
|
|
|
einops # required for MPT and qwen-vl
|
2024-06-13 11:22:41 -07:00
|
|
|
httpx
|
|
|
|
|
peft
|
|
|
|
|
requests
|
|
|
|
|
ray
|
|
|
|
|
sentence-transformers # required for embedding
|
2024-08-14 18:07:37 -04:00
|
|
|
compressed-tensors==0.4.0 # required for compressed-tensors
|
2024-07-29 18:16:30 +08:00
|
|
|
timm # required for internvl test
|
2024-08-18 17:43:46 -06:00
|
|
|
transformers_stream_generator # required for qwen-vl test
|
|
|
|
|
matplotlib # required for qwen-vl test
|
2024-06-13 11:22:41 -07:00
|
|
|
|
2024-08-06 16:55:31 +08:00
|
|
|
# TODO: Add this after fully implementing llava(mantis)
|
|
|
|
|
# git+https://github.com/TIGER-AI-Lab/Mantis.git # required for llava(mantis) test
|
|
|
|
|
|
2024-06-13 11:22:41 -07:00
|
|
|
# Benchmarking
|
|
|
|
|
aiohttp
|
|
|
|
|
|
|
|
|
|
# quantization
|
2024-08-19 19:55:04 -07:00
|
|
|
bitsandbytes==0.42.0
|
|
|
|
|
buildkite-test-collector==0.1.8
|
