From 98dd9dda0cbbe6b60331e424e1deb90c2282a4ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9mi=20Louf?= <remilouf@gmail.com>
Date: Mon, 15 Apr 2024 15:04:10 +0200
Subject: [PATCH] Add `torch` dependency in installation instructions

---
 docs/installation.md | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/docs/installation.md b/docs/installation.md
index e76a2a7c7..12f113d36 100644
--- a/docs/installation.md
+++ b/docs/installation.md
@@ -14,10 +14,11 @@ Outlines supports OpenAI, transformers, Mamba, llama.cpp and exllama2 but **you
 
 ```python
 pip install openai
-pip install transformers datasets accelerate
+pip install transformers datasets accelerate torch
 pip install llama-cpp-python
-pip install exllamav2
-pip install mamba_ssm
+pip install exllamav2 torch
+pip install mamba_ssm torch
+pip install vllm
 ```
 
 If you encounter any problem 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.