first commit

2022-06-27 11:57:56 -04:00
parent 97fe8515f1
commit 9cf9d46dd7
13 changed files with 1514 additions and 0 deletions
@@ -0,0 +1,15 @@
+#!/bin/bash
+
+pip install -r requirements.txt
+
+mkdir -p pretrained
+
+# download vqgan
+git lfs install
+git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretrained/vqgan
+
+# download dalle-mini and dalle mega
+pip install wandb
+wandb login
+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