Merge branch 'main' of https://github.com/kuprel/min-dalle

2 years ago · d846cab1b6
parent 17c96fe110 0af6b1731f
commit d846cab1b6
2 changed files with 4 additions and 5 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,2 +1,3 @@
 torch
 flax==0.4.2
+wandb
--- a/setup.sh
+++ b/setup.sh
@ -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