Installation
You can install Outlines with pip
:
Outlines supports OpenAI, Transformers, Mamba, llama.cpp, and ExLlamaV2, but you will need to install them manually:
pip install openai
pip install transformers datasets accelerate torch
pip install llama-cpp-python
pip install exllamav2 transformers torch
pip install mamba_ssm transformers torch
pip install vllm
If you encounter any problems using Outlines with these libraries, take a look at their installation instructions. The installation of openai
and transformers
should be straightforward, but other libraries have specific hardware requirements.
Optional Dependencies
Outlines provides multiple optional dependency sets to support different backends and use cases. You can install them as needed using:
pip install "outlines[vllm]"
for vLLM, optimized for high-throughput inference.pip install "outlines[transformers]"
for Hugging Face Transformers.pip install "outlines[mlx]"
for MLX-LM, optimized for Apple silicon.pip install "outlines[openai]"
to use OpenAIโs API.pip install "outlines[llamacpp]"
for llama.cpp, a lightweight LLM inference engine.pip install "outlines[exllamav2]"
for ExLlamaV2, optimized for NVIDIA GPUs.
Bleeding Edge
You can install the latest version of Outlines from the repository's main
branch:
This can be useful, for instance, when a fix has been merged but not yet released.
Installing for Development
See the contributing documentation for instructions on how to install Outlines for development, including an example using the dot-install
method for one of the backends.