From e4c15fe1f6a2be574eb223e0102cbdb2a11dfe40 Mon Sep 17 00:00:00 2001 From: henk717 Date: Fri, 21 Apr 2023 03:00:52 +0200 Subject: [PATCH] Update install_requirements.sh --- install_requirements.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/install_requirements.sh b/install_requirements.sh index e131ad37..12c83595 100755 --- a/install_requirements.sh +++ b/install_requirements.sh @@ -1,12 +1,12 @@ #!/bin/bash -if [[ $1 = "cuda" ]||[ $1 = "CUDA" ]]; then +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 exit fi -if [[ $1 = "rocm" ]||[ $1 = "ROCM" ]]; then +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