2022-06-27 15:57:56 +00:00
|
|
|
#!/bin/bash
|
|
|
|
|
|
2022-06-30 01:45:05 +00:00
|
|
|
set -e
|
|
|
|
|
|
2022-07-01 16:53:55 +00:00
|
|
|
pip3 install -r requirements_flax.txt
|
2022-06-27 15:57:56 +00:00
|
|
|
|
|
|
|
|
# download vqgan
|
2022-07-01 14:17:29 +00:00
|
|
|
mkdir -p pretrained/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
|
|
|
|
2022-07-01 14:17:29 +00:00
|
|
|
# download dalle-mini and dalle-mega
|
2022-07-01 16:21:31 +00:00
|
|
|
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
|
