diff --git a/setup.sh b/setup.sh index 7df64c6..4793242 100644 --- a/setup.sh +++ b/setup.sh @@ -10,6 +10,6 @@ git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretraine # download dalle-mini and dalle mega 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 +python -m wandb login +python -m wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0 +python -m wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14