From ee846ba1cfa0bcaf46a4e2df92fdcf3b227fa726 Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Fri, 1 Jul 2022 12:21:31 -0400 Subject: [PATCH] update setup scripts --- setup_flax.sh | 6 +++--- setup_torch.sh | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/setup_flax.sh b/setup_flax.sh index 0376eed..4346768 100644 --- a/setup_flax.sh +++ b/setup_flax.sh @@ -9,6 +9,6 @@ mkdir -p pretrained/vqgan curl https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384/resolve/main/flax_model.msgpack -L --output ./pretrained/vqgan/flax_model.msgpack # download dalle-mini and dalle-mega -python -m wandb login --anonymously -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 +python3 -m wandb login --anonymously +python3 -m wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0 +python3 -m wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14 diff --git a/setup_torch.sh b/setup_torch.sh index b0eab2e..3d4e69f 100644 --- a/setup_torch.sh +++ b/setup_torch.sh @@ -2,7 +2,7 @@ set -e -pip install torch +pip3 install torch mkdir -p ./pretrained/dalle_bart_mega/ curl https://huggingface.co/kuprel/min-dalle/resolve/main/vocab.json -L --output ./pretrained/dalle_bart_mega/vocab.json