"requirements/test/cuda.in" did not exist on "965fe45935473e0a81cc6a0885ae7161b9c8b8cf"
requirements-common.txt 1.73 KB
Newer Older
1
2
psutil
sentencepiece  # Required for LLaMA tokenizer.
3
numpy < 2.0.0
4
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding.
5
requests >= 2.26.0
6
tqdm
7
blake3
yhu422's avatar
yhu422 committed
8
py-cpuinfo
zhuwenwen's avatar
zhuwenwen committed
9
transformers >= 4.49.0  # Required for Bamba model and Transformers backend.
10
tokenizers >= 0.19.1  # Required for Llama 3.
11
protobuf # Required by LlamaTokenizer.
12
13
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'
14
aiohttp
15
openai >= 1.52.0 # Ensure modern openai package (ensure types module present and max_completion_tokens field support)
16
pydantic >= 2.9
17
prometheus_client >= 0.18.0
18
pillow  # Required for image processing
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
lark == 1.2.2 
24
xgrammar == 0.1.11; platform_machine == "x86_64"
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
32
mistral_common[opencv] >= 1.5.0
33
pyyaml
34
six>=1.16.0; python_version > '3.11' # transitive dependency of pandas that needs to be the latest version for python 3.12
35
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
36
einops # Required for Qwen2-VL.
37
compressed-tensors == 0.9.1 # required for compressed-tensors
38
39
depyf==0.18.0 # required for profiling and debugging with compilation config
cloudpickle # allows pickling lambda functions in model_executor/models/registry.py