min-dalle-test/setup_flax.sh

#!/bin/bash

set -e

pip install torch flax wandb

# download vqgan
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
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
first commit 2022-06-27 15:57:56 +00:00			`#!/bin/bash`

Update setup.sh 2022-06-30 01:45:05 +00:00			`set -e`

fixed setup_flax.sh 2022-07-01 15:32:36 +00:00			`pip install torch flax wandb`
first commit 2022-06-27 15:57:56 +00:00
			`# download vqgan`
save converted detokenizer params 2022-07-01 14:17:29 +00:00			`mkdir -p pretrained/vqgan`
Add flax model curl 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`
first commit 2022-06-27 15:57:56 +00:00
save converted detokenizer params 2022-07-01 14:17:29 +00:00			`# download dalle-mini and dalle-mega`
update setup scripts 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`