| 12345678910111213141516171819202122232425262728 |
- # Needed for Ray accelerated DAG tests
- -r requirements-adag.txt
- # testing
- pytest
- tensorizer>=2.9.0
- pytest-forked
- pytest-asyncio
- pytest-rerunfailures
- pytest-shard
- # testing utils
- awscli
- einops # required for MPT
- httpx
- peft
- requests
- ray
- sentence-transformers # required for embedding
- sparseml==1.8.0 # required for compressed-tensors
- compressed-tensors==0.4.0 # required for compressed-tensors
- timm # required for internvl test
- # Benchmarking
- aiohttp
- # quantization
- bitsandbytes==0.42.0
|
