Skip to content

Commit

Permalink
Merge pull request #68 from CambioML/install
Browse files Browse the repository at this point in the history
add installation multiple options
  • Loading branch information
goldmermaid authored Sep 20, 2023
2 parents c6be98b + 78abb31 commit 1a8be3a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
```
Expand Down Expand Up @@ -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)
```
Expand Down

0 comments on commit 1a8be3a

Please sign in to comment.