tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Simplified requirements:

Browse Source 
* No wandb login
* wandb install as part of requirements.txt
This commit is contained in:
Andrew Ginns
2022-06-28 21:34:57 +01:00
parent 9d6b6dcc92
commit 3af89066e8
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@@ -1,2 +1,3 @@
torch torch
flax==0.4.2 flax==0.4.2
wandb

3
setup.sh
View File

@@ -9,7 +9,6 @@ git lfs install
git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretrained/vqgan git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretrained/vqgan
# download dalle-mini and dalle mega # download dalle-mini and dalle mega
pip install wandb wandb login --anonymously
wandb login
wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0 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