requirements-common.txt 1.55 KB
Newer Older
1
2
psutil
sentencepiece  # Required for LLaMA tokenizer.
3
numpy < 2.0.0
4
requests >= 2.26.0
5
tqdm
6
blake3
yhu422's avatar
yhu422 committed
7
py-cpuinfo
8
transformers >= 4.45.2  # Required for Llama 3.2 and Qwen2-VL.
9
tokenizers >= 0.19.1  # Required for Llama 3.
10
protobuf # Required by LlamaTokenizer.
11
12
fastapi >= 0.107.0, < 0.113.0; python_version < '3.9'
fastapi >= 0.107.0, != 0.113.*, != 0.114.0; python_version >= '3.9'
13
aiohttp
14
openai >= 1.45.0 # Ensure modern openai package (ensure types module present and max_completion_tokens field support)
15
uvicorn[standard]
16
pydantic >= 2.9  # Required for fastapi >= 0.113.0
17
pillow  # Required for image processing
18
prometheus_client >= 0.18.0
19
prometheus-fastapi-instrumentator >= 7.0.0
20
tiktoken >= 0.6.0  # Required for DBRX tokenizer
21
lm-format-enforcer >= 0.10.9, < 0.11
22
outlines == 0.1.11
23
xgrammar >= 0.1.6; platform_machine == "x86_64"
24
typing_extensions >= 4.10
25
filelock >= 3.16.1 # need to contain https://github.com/tox-dev/filelock/pull/317
26
partial-json-parser # used for parsing partial JSON outputs
27
pyzmq
28
msgspec
29
gguf == 0.10.0
30
importlib_metadata
31
mistral_common[opencv] >= 1.5.0
32
pyyaml
33
six>=1.16.0; python_version > '3.11' # transitive dependency of pandas that needs to be the latest version for python 3.12
34
setuptools>=74.1.1; python_version > '3.11' # Setuptools is used by triton, we need to ensure a modern version is installed for 3.12+ so that it does not try to import distutils, which was removed in 3.12
35
einops # Required for Qwen2-VL.
36
compressed-tensors == 0.8.1 # required for compressed-tensors
37
depyf==0.18.0 # required for profiling and debugging torch.compile