Installation
Before you start, you’ll need to setup your environment and install the appropriate packages. 🤗 SetFit is tested on Python 3.8+.
pip
The most straightforward way to install 🤗 SetFit is with pip:
pip install setfit
If you have a CUDA-capable graphics card, then it is recommended to install torch
with CUDA support to train and performing inference much more quickly:
pip install torch --index-url https://download.pytorch.org/whl/cu118
Installing from source
Building 🤗 SetFit from source lets you make changes to the code base. To install from the source, clone the repository and install 🤗 SetFit in editable mode with the following commands:
git clone https://github.com/huggingface/setfit.git
cd setfit
pip install -e .
If you just want the bleeding-edge version without making any changes of your own, then install from source by running:
pip install git+https://github.com/huggingface/setfit.git
Conda
If conda is your package management system of choice, then you can install 🤗 SetFit like so:
conda install -c conda-forge setfit