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

Support pip/pip3 binary names for setup

Browse Source
This commit is contained in:
Jonathan Dahan 2022-06-28 11:20:23 -04:00 committed by GitHub
parent 7478c5ffa3
commit fd0e808454
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
setup.sh
View File

@ -1,6 +1,8 @@
#!/bin/bash
pip install -r requirements.txt
_pip=$(command -v pip pip3)
$_pip install -r requirements.txt
mkdir -p pretrained
@ -9,7 +11,7 @@ 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
$_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