From 3af89066e8103aa5418f7a8481afb5e75d236ca5 Mon Sep 17 00:00:00 2001 From: Andrew Ginns Date: Tue, 28 Jun 2022 21:34:57 +0100 Subject: [PATCH] Simplified requirements: * No wandb login * wandb install as part of requirements.txt --- requirements.txt | 1 + setup.sh | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) 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