separate setup processes for flax and torch

2022-07-01 11:08:33 -04:00
parent 7bf76deafb
commit 09a0f85b8e
5 changed files with 16 additions and 11 deletions
@@ -1,14 +0,0 @@
-#!/bin/bash
-
-set -e
-
-pip install -r requirements.txt
-
-# 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
-python -m wandb login --anonymously
-python -m wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0
-python -m wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14