requirements-common.txt 1.64 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.48.2  # Required for Bamba model and Transformers backend.
9
tokenizers >= 0.19.1  # Required for Llama 3.
10
protobuf # Required by LlamaTokenizer.
11
12
fastapi[standard] >= 0.107.0, < 0.113.0; python_version < '3.9'
fastapi[standard]  >= 0.107.0, != 0.113.*, != 0.114.0; python_version >= '3.9'
13
aiohttp
14
openai >= 1.52.0 # Ensure modern openai package (ensure types module present and max_completion_tokens field support)
15
pydantic >= 2.9
16
prometheus_client >= 0.18.0
17
pillow  # Required for image processing
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.1.11
22
lark == 1.2.2 
23
xgrammar == 0.1.11; 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.9.1 # required for compressed-tensors
37
38
depyf==0.18.0 # required for profiling and debugging with compilation config
cloudpickle # allows pickling lambda functions in model_executor/models/registry.py