From dea5d046abea1dc8c5554479ee561ea199437d3f Mon Sep 17 00:00:00 2001
From: Rachel Hu <goldpiggy@berkeley.edu>
Date: Wed, 20 Sep 2023 10:15:29 -0700
Subject: [PATCH] add installation multiple options

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 336bb3c..2d90e74 100644
--- a/README.md
+++ b/README.md
@@ -69,7 +69,7 @@ conda create -n pykoi python=3.10 -y && source activate pykoi
 
 Then install `pykoi` and [pytorch based on your cuda version](https://pytorch.org/get-started/locally/). You can find your CUDA version via `nvcc -V`.
 ```
-pip3 install pykoi[huggingface]
+pip3 install "pykoi[rag, huggingface]"
 
 pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121  # cu121 means cuda 12.1
 ```
@@ -110,7 +110,7 @@ Then, based the feature you need to develop, run one or more installation option
   ```
 - Option 2: RAG (GPU) 
   ```
-  poetry install --no-root --extras huggingface
+  poetry install --no-root --extras "rag huggingface"
   ```
 - Option 3: RLHF (GPU) 
   ```