min-dalle-test/setup.sh

#!/bin/bash

pip install -r requirements.txt

mkdir -p pretrained/vqgan

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

torch.no_grad(), cleanup 2022-06-28 16:16:44 +00:00			`pip install -r requirements.txt`
first commit 2022-06-27 15:57:56 +00:00
Add flax model curl 2022-06-28 21:01:26 +00:00			`mkdir -p pretrained/vqgan`
first commit 2022-06-27 15:57:56 +00:00
			`# download 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
			`# download dalle-mini and dalle mega`
Simplified requirements: * No wandb login * wandb install as part of requirements.txt 2022-06-28 20:34:57 +00:00			`wandb login --anonymously`
first commit 2022-06-27 15:57:56 +00:00			`wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0`
Support pip/pip3 binary names for setup 2022-06-28 15:20:23 +00:00			`wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14`