diff --git a/setup.sh b/setup.sh
index ee82889..166e821 100644
--- a/setup.sh
+++ b/setup.sh
@@ -1,5 +1,7 @@
 #!/bin/bash
 
+set -e
+
 pip install -r requirements.txt
 
 mkdir -p pretrained
@@ -12,4 +14,4 @@ git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretraine
 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 
\ No newline at end of file
+wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14