Update install stuff

This commit is contained in:
Joey Yakimowich-Payne 2025-05-13 14:58:44 -06:00
commit 25b12a512d
3 changed files with 21 additions and 12 deletions

View file

@ -3,22 +3,15 @@ conda init
conda create --name prompt-guard python=3.12 -c conda-forge
conda activate prompt-guard
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
rustup install 1.82
rustup default 1.82
pip install fschat
#cuda
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
#or cpu
pip install torch --index-url https://download.pytorch.org/whl/cpu
pip install -U "huggingface_hub[cli]"
# install transformers
pip install transformers
export RUSTFLAGS="-A invalid_reference_casting"
pip install git+https://github.com/llm-attacks/llm-attacks
pip install -U transformers
pip install -U tiktoken
huggingface-cli login
pip install -U transformers