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