Installation

Installation

You can install the package by either using PyPI

pip install simcse

Or directly install it from our code

git clone https://github.com/princeton-nlp/SimCSE.git
cd SimCSE
python setup.py install

Dependencies

Most dependencies are automatically solved when installing the package. However, there are two special dependencies, PyTorch and Faiss.

If you are using GPUs, you should check the PyTorch official website for instructions for installing correct versions, otherwise there is chance that PyTorch cannot work correctly with your GPU devices.

For fast indexing and retrieval, our package supports the use of Faiss, a library for efficient similarity search and clustering of dense vectors. Note that faiss is not a necessary dependency for our package, but if you install one, simcse automatically takes faiss as the backend.