diff --git a/setup.sh b/setup.sh
index 166e821..a2e9b2a 100644
--- a/setup.sh
+++ b/setup.sh
@@ -1,8 +1,7 @@
 #!/bin/bash
 
-set -e
-
-pip install -r requirements.txt
+_pip=$(command -v pip pip3)
+$_pip install -r requirements.txt
 
 mkdir -p pretrained
 
@@ -11,7 +10,7 @@ git lfs install
 git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretrained/vqgan
 
 # download dalle-mini and dalle mega
-pip install wandb
+$_pip install wandb
 wandb login
 wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0
 wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14