2023-07-04 02:31:55 +08:00
|
|
|
# formatting
|
|
|
|
|
yapf==0.32.0
|
2023-12-14 16:19:41 +08:00
|
|
|
toml==0.10.2
|
2024-02-22 10:56:01 +08:00
|
|
|
tomli==2.0.1
|
2023-11-21 03:58:01 +08:00
|
|
|
ruff==0.1.5
|
2024-02-22 10:56:01 +08:00
|
|
|
codespell==2.2.6
|
2024-03-25 22:59:47 +08:00
|
|
|
isort==5.13.2
|
2024-05-22 15:18:41 +08:00
|
|
|
clang-format==18.1.5
|
2023-07-04 02:31:55 +08:00
|
|
|
|
|
|
|
|
# type checking
|
2024-04-13 05:35:50 +08:00
|
|
|
mypy==1.9.0
|
2023-07-04 02:31:55 +08:00
|
|
|
types-PyYAML
|
|
|
|
|
types-requests
|
|
|
|
|
types-setuptools
|
|
|
|
|
|
|
|
|
|
# testing
|
2023-05-24 08:58:51 +08:00
|
|
|
pytest
|
2024-05-14 05:57:07 +08:00
|
|
|
tensorizer>=2.9.0
|
2023-09-01 10:19:43 +08:00
|
|
|
pytest-forked
|
2023-09-17 15:29:08 +08:00
|
|
|
pytest-asyncio
|
2024-03-02 04:47:51 +08:00
|
|
|
pytest-rerunfailures
|
2024-03-18 05:56:30 +08:00
|
|
|
pytest-shard
|
2024-05-12 02:30:37 +08:00
|
|
|
|
|
|
|
|
# testing utils
|
|
|
|
|
awscli
|
2024-01-15 04:37:58 +08:00
|
|
|
einops # required for MPT
|
2024-05-12 02:30:37 +08:00
|
|
|
httpx
|
|
|
|
|
peft
|
2024-01-17 13:33:14 +08:00
|
|
|
requests
|
2024-02-23 01:58:29 +08:00
|
|
|
ray
|
2024-05-12 02:30:37 +08:00
|
|
|
sentence-transformers # required for embedding
|
2024-03-07 16:39:28 +08:00
|
|
|
|
|
|
|
|
# Benchmarking
|
|
|
|
|
aiohttp
|
2024-03-26 05:16:30 +08:00
|
|
|
|
2024-06-02 04:51:10 +08:00
|
|
|
# quantization
|
|
|
|
|
bitsandbytes==0.42.0
|