安装

在开始之前，您需要设置您的环境并安装适当的软件包。🤗 SetFit 在 Python 3.9+ 版本上测试通过。

pip

安装 🤗 SetFit 最直接的方法是使用 pip

pip install setfit

如果您有支持 CUDA 的显卡，建议安装支持 CUDA 的 torch，以更快地进行训练和推理

pip install torch --index-url https://download.pytorch.org/whl/cu118

从源码安装

从源码构建 🤗 SetFit 可以让您更改代码库。要从源码安装，请克隆存储库并在可编辑模式下使用以下命令安装 🤗 SetFit

git clone https://github.com/huggingface/setfit.git
cd setfit
pip install -e .

如果您只想使用最新版本而无需进行任何更改，请运行以下命令从源码安装

pip install git+https://github.com/huggingface/setfit.git

Conda

如果 conda 是您选择的包管理系统，那么您可以像这样安装 🤗 SetFit

conda install -c conda-forge setfit