min-dalle-test/setup.sh

16 lines
519 B
Bash
Raw Normal View History

2022-06-27 15:57:56 +00:00
#!/bin/bash
2022-06-30 01:45:05 +00:00
set -e
2022-06-28 16:16:44 +00:00
pip install -r requirements.txt
2022-06-27 15:57:56 +00:00
2022-06-28 21:01:26 +00:00
mkdir -p pretrained/vqgan
2022-06-27 15:57:56 +00:00
# download vqgan
2022-06-28 21:01:26 +00:00
curl https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384/resolve/main/flax_model.msgpack -L --output ./pretrained/vqgan/flax_model.msgpack
2022-06-27 15:57:56 +00:00
# download dalle-mini and dalle mega
2022-07-01 11:54:19 +00:00
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