Mention in README datasets requirement

dottxt-ai · Nov 21, 2023 · a16a231 · a16a231
1 parent b736708
commit a16a231
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -68,7 +68,7 @@ pip install outlines
 The dependencies needed to use models are not installed by default. You will need to run:
 
 - `pip install openai` to be able to use OpenAI [models](https://platform.openai.com/docs/api-reference).
-- `pip install transformers` to be able to use Hugging Face `transformers` [models](https://huggingface.co/models?pipeline_tag=text-generation).
+- `pip install transformers datasets` to be able to use Hugging Face `transformers` [models](https://huggingface.co/models?pipeline_tag=text-generation).
 
 ## Guided generation
 

diff --git a/docs/get_started.md b/docs/get_started.md
@@ -36,7 +36,7 @@ pip install outlines
 
 ??? info "Using OpenAI and Transformers"
 
-    Outlines :wavy_dash: does not install the `openai` or `transformers` libraries by default. You will have to install these libraries manually.
+    Outlines :wavy_dash: does not install the `openai` or `transformers` libraries by default. You will have to install these libraries manually. To use `transformers` models you will also need to install the `datasets` library.
 
 ## :eyes: Sneak Peek