From 860b697a705ad28982b20aac1135e4ff35e50647 Mon Sep 17 00:00:00 2001
From: Syler Clayton <2224238+Relys@users.noreply.github.com>
Date: Sat, 15 Apr 2023 09:51:45 -0700
Subject: [PATCH] Update install_requirements.sh

Made parameter case insensitive.
---
 install_requirements.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/install_requirements.sh b/install_requirements.sh
index 7d2c0c2f..e131ad37 100755
--- a/install_requirements.sh
+++ b/install_requirements.sh
@@ -1,12 +1,12 @@
 #!/bin/bash
-if [[ $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" ]]; 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