diff --git a/README.md b/README.md index 45dd0b4..0048b31 100644 --- a/README.md +++ b/README.md @@ -113,7 +113,7 @@ There are two ways to load the knowledge base data: If you are running eval, you may install the following packages: ```bash -pip install llm2vec gritlm bm25 +pip install llm2vec gritlm bm25s ``` - Our evaluation requires embed the node documents into `candidate_emb_dict.pt`, which is a dictionary `node_id -> torch.Tensor`. Query embeddings will be automatically generated if not available. You can either run the following the python script to download query embeddings and document embeddings generated by `text-embedding-ada-002`. (We provide them so you can run on our benchmark right away.)