From 3c0c89797743607117cb0d0aa19744a41698c72a Mon Sep 17 00:00:00 2001
From: Andrew Ginns <ginns.aw@gmail.com>
Date: Tue, 28 Jun 2022 22:01:26 +0100
Subject: [PATCH] Add flax model curl

---
 setup.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.sh b/setup.sh
index 91f7483..6eee680 100644
--- a/setup.sh
+++ b/setup.sh
@@ -2,11 +2,13 @@
 
 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
+# download the flax model
+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