This commit is contained in:
Brett Kuprel 2022-06-28 21:28:50 -04:00
commit d846cab1b6
2 changed files with 4 additions and 5 deletions

View File

@ -1,2 +1,3 @@
torch
flax==0.4.2
wandb

View File

@ -2,14 +2,12 @@
pip install -r requirements.txt
mkdir -p pretrained
mkdir -p pretrained/vqgan
# download vqgan
git lfs install
git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./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
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