|
|
|
@ -1,5 +1,7 @@ |
|
|
|
|
#!/bin/bash |
|
|
|
|
|
|
|
|
|
set -e |
|
|
|
|
|
|
|
|
|
pip install -r requirements.txt |
|
|
|
|
|
|
|
|
|
mkdir -p pretrained |
|
|
|
@ -12,4 +14,4 @@ git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretraine |
|
|
|
|
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 |
|
|
|
|
wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14 |
|
|
|
|