diff --git a/requirements.txt b/requirements.txt
index 3ff893b..0867d5f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,3 @@
 torch
 flax==0.4.2
+wandb
\ No newline at end of file
diff --git a/setup.sh b/setup.sh
index 7df64c6..91f7483 100644
--- a/setup.sh
+++ b/setup.sh
@@ -9,7 +9,6 @@ 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
-wandb login
+wandb login --anonymously
 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