interegular
jinja2
lark
nest_asyncio
numpy
cloudpickle
diskcache
pydantic>=2.0
referencing
jsonschema
requests
tqdm
typing_extensions
pycountry
airportsdata
torch
outlines_core==0.1.20

[exllamav2]
exllamav2

[llamacpp]
llama-cpp-python
transformers
datasets
numpy<2

[mlxlm]
mlx-lm
datasets

[openai]
openai

[serve]
vllm>=0.3.0
uvicorn
fastapi
pydantic>=2.0

[test]
pre-commit
pytest
pytest-benchmark
pytest-cov
pytest-mock
coverage[toml]>=5.1
diff-cover
accelerate
beartype<0.16.0
responses
llama-cpp-python
huggingface_hub
openai>=1.0.0
datasets
transformers
pillow
exllamav2
jax

[test:platform_machine == "arm64" and sys_platform == "darwin"]
mlx-lm>=0.19.2

[test:sys_platform != "darwin"]
vllm

[transformers]
transformers
accelerate
datasets
numpy<2

[vllm]
vllm
transformers
numpy2
