Skip to content

Commit

Permalink
Add torch dependency in installation instructions
Browse files Browse the repository at this point in the history
  • Loading branch information
rlouf committed Apr 15, 2024
1 parent f2af45a commit 98dd9dd
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions docs/installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 98dd9dd

Please sign in to comment.