2023-12-05 07:15:37 +00:00
|
|
|
#!/bin/bash
|
2024-03-01 21:48:53 +00:00
|
|
|
set -ex
|
2023-12-05 07:15:37 +00:00
|
|
|
|
|
|
|
export PATH=$PATH:/opt/conda/bin
|
|
|
|
|
2024-03-01 21:48:53 +00:00
|
|
|
# Check if environment exist
|
|
|
|
conda_env_exists(){
|
|
|
|
! conda list --name "${@}" >/dev/null 2>/dev/null
|
|
|
|
}
|
2023-12-05 07:15:37 +00:00
|
|
|
|
2024-03-01 21:48:53 +00:00
|
|
|
if conda_env_exists "exllama" ; then
|
|
|
|
echo "Creating virtual environment..."
|
|
|
|
conda env create --name exllama --file $1
|
|
|
|
echo "Virtual environment created."
|
|
|
|
else
|
|
|
|
echo "Virtual environment already exists."
|
|
|
|
fi
|
2023-12-05 07:15:37 +00:00
|
|
|
|
2024-03-01 21:48:53 +00:00
|
|
|
source activate exllama
|
2023-12-05 07:15:37 +00:00
|
|
|
|
|
|
|
git clone https://github.com/turboderp/exllama $CONDA_PREFIX/exllama && pushd $CONDA_PREFIX/exllama && pip install -r requirements.txt && popd
|
|
|
|
|
2024-01-07 23:37:02 +00:00
|
|
|
cp -rfv $CONDA_PREFIX/exllama/* ./
|
|
|
|
|
|
|
|
if [ "$PIP_CACHE_PURGE" = true ] ; then
|
|
|
|
pip cache purge
|
|
|
|
fi
|