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