7041de4384
Co-authored-by: LiuXiaoxuanPKU <llilyliupku@gmail.com>, bong-furiosa <bongwon.jang@furiosa.ai>
22 lines
279 B
Plaintext
22 lines
279 B
Plaintext
# 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
|
|
|
|
# Benchmarking
|
|
aiohttp
|
|
|
|
# quantization
|
|
bitsandbytes==0.42.0 |