common.txt 2.02 KB
Newer Older
1
cachetools
2
3
psutil
sentencepiece  # Required for LLaMA tokenizer.
4
numpy
5
requests >= 2.26.0
6
tqdm
7
blake3
yhu422's avatar
yhu422 committed
8
py-cpuinfo
9
transformers >= 4.50.3
10
tokenizers >= 0.19.1  # Required for Llama 3.
11
protobuf # Required by LlamaTokenizer.
12
fastapi[standard] >= 0.115.0 # Required by FastAPI's form models in the OpenAI API server's audio transcriptions endpoint.
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.11, < 0.11
21
llguidance >= 0.7.9, < 0.8.0; platform_machine == "x86_64" or platform_machine == "arm64" or platform_machine == "aarch64"
22
outlines == 0.1.11
23
lark == 1.2.2
24
xgrammar == 0.1.16; platform_machine == "x86_64" or platform_machine == "aarch64"
25
typing_extensions >= 4.10
26
filelock >= 3.16.1 # need to contain https://github.com/tox-dev/filelock/pull/317
27
partial-json-parser # used for parsing partial JSON outputs
28
pyzmq
29
msgspec
30
gguf == 0.10.0
31
importlib_metadata
Patrick von Platen's avatar
Patrick von Platen committed
32
mistral_common[opencv] >= 1.5.4
33
opencv-python-headless >= 4.11.0    # required for video IO
34
pyyaml
35
six>=1.16.0; python_version > '3.11' # transitive dependency of pandas that needs to be the latest version for python 3.12
36
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
37
einops # Required for Qwen2-VL.
38
compressed-tensors == 0.9.2 # required for compressed-tensors
39
40
depyf==0.18.0 # required for profiling and debugging with compilation config
cloudpickle # allows pickling lambda functions in model_executor/models/registry.py
41
watchfiles # required for http server to monitor the updates of TLS files
42
python-json-logger # Used by logging as per examples/other/logging_configuration.md
43
scipy # Required for phi-4-multimodal-instruct
44
ninja # Required for xgrammar, rocm, tpu, xpu