File size: 2,384 Bytes
9bcb78e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
#!/bin/bash
# converted from windows-setup.bat by github copilot
# check if "venv" subdirectory exists, if not, create one
if [ ! -d "venv" ]; then
python -m venv venv
else
echo "venv directory already exists. If something is broken, delete everything but exl2-quant.py and run this script again."
read -p "Press enter to continue"
exit
fi
# ask if the user has git installed
read -p "Do you have git and wget installed? (y/n) " gitwget
if [ "$gitwget" = "y" ]; then
echo "Setting up environment"
else
echo "Please install git and wget before running this script."
read -p "Press enter to continue"
exit
fi
# if CUDA version 12 install pytorch for 12.1, else if CUDA 11 install pytorch for 11.8. If ROCm, install pytorch for ROCm 5.7
read -p "Please enter your GPU compute version, CUDA 11/12 or AMD ROCm (11, 12, rocm): " pytorch_version
if [ "$pytorch_version" = "11" ]; then
echo "Installing PyTorch for CUDA 11.8"
venv/bin/python -m pip install torch --index-url https://download.pytorch.org/whl/cu118
elif [ "$pytorch_version" = "12" ]; then
echo "Installing PyTorch for CUDA 12.1"
venv/bin/python -m pip install torch
elif [ "$pytorch_version" = "rocm" ]; then
echo "Installing PyTorch for AMD ROCm 5.7"
venv/bin/python -m pip install torch --index-url https://download.pytorch.org/whl/rocm5.7
else
echo "Invalid compute version. Please enter 11, 12, or rocm."
read -p "Press enter to continue"
exit
fi
# download stuff
echo "Downloading files"
git clone https://github.com/turboderp/exllamav2
wget https://raw.githubusercontent.com/oobabooga/text-generation-webui/main/convert-to-safetensors.py
wget https://raw.githubusercontent.com/oobabooga/text-generation-webui/main/download-model.py
echo "Installing pip packages"
venv/bin/python -m pip install -r exllamav2/requirements.txt
venv/bin/python -m pip install huggingface-hub transformers accelerate
venv/bin/python -m pip install ./exllamav2
# create start-quant.sh
echo "#!/bin/bash" > start-quant.sh
echo "venv/bin/python exl2-quant.py" >> start-quant.sh
echo "read -p \"Press enter to continue\""
chmod +x start-quant.sh
echo "If you use ctrl+c to stop, you may need to also use 'pkill python' to stop running scripts."
echo "Environment setup complete. run start-quant.sh to start the quantization process."
read -p "Press enter to exit"
exit |