SetFit documentation

Installation

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

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
< > Update on GitHub