Skip to content

Commit 293be13

Browse files
committed
exllama/exllama2: do not install if not using cuda
1 parent a044a18 commit 293be13

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

backend/python/exllama/install.sh

+5
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,11 @@ set -ex
33

44
export PATH=$PATH:/opt/conda/bin
55

6+
if [ "$BUILD_TYPE" != "cublas" ]; then
7+
echo "[exllama] Attention!!! Nvidia GPU is required - skipping installation"
8+
exit 0
9+
fi
10+
611
# Check if environment exist
712
conda_env_exists(){
813
! conda list --name "${@}" >/dev/null 2>/dev/null

backend/python/exllama2/install.sh

+5
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,11 @@ export SHA=c0ddebaaaf8ffd1b3529c2bb654e650bce2f790f
66

77
SKIP_CONDA=${SKIP_CONDA:-0}
88

9+
if [ "$BUILD_TYPE" != "cublas" ]; then
10+
echo "[exllamav2] Attention!!! Nvidia GPU is required - skipping installation"
11+
exit 0
12+
fi
13+
914
if [ $SKIP_CONDA -eq 0 ]; then
1015
export PATH=$PATH:/opt/conda/bin
1116
source activate transformers

0 commit comments

Comments
 (0)