diff --git a/install_requirements.sh b/install_requirements.sh index 561b1b00..6e37c7e9 100755 --- a/install_requirements.sh +++ b/install_requirements.sh @@ -5,14 +5,14 @@ if [[ $1 = "cuda" || $1 = "CUDA" ]]; then wget -qO- https://micromamba.snakepit.net/api/micromamba/linux-64/latest | tar -xvj bin/micromamba bin/micromamba create -f environments/huggingface.yml -r runtime -n koboldai -y # Weird micromamba bug causes it to fail the first time, running it twice just to be safe, the second time is much faster -# bin/micromamba create -f environments/huggingface.yml -r runtime -n koboldai -y +bin/micromamba create -f environments/huggingface.yml -r runtime -n koboldai -y exit fi if [[ $1 = "rocm" || $1 = "ROCM" ]]; then wget -qO- https://micromamba.snakepit.net/api/micromamba/linux-64/latest | tar -xvj bin/micromamba bin/micromamba create -f environments/rocm.yml -r runtime -n koboldai-rocm -y # Weird micromamba bug causes it to fail the first time, running it twice just to be safe, the second time is much faster -# bin/micromamba create -f environments/rocm.yml -r runtime -n koboldai-rocm -y +bin/micromamba create -f environments/rocm.yml -r runtime -n koboldai-rocm -y exit fi echo Please specify either CUDA or ROCM