doc-to-lora/install.sh

14 lines
452 B
Bash
Executable file

# install unsloth
# see https://docs.unsloth.ai/get-started/install-update/conda-install
conda create --name unsloth_env \
python=3.10 \
pytorch-cuda=<11.8/12.1> \
pytorch cudatoolkit xformers -c pytorch -c nvidia -c xformers \
-y
conda activate unsloth_env
pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
pip install --no-deps "trl<0.9.0" peft accelerate bitsandbytes
pip install -r requirements.txt