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

Merge pull request #19 from jedahan/patch-2

Browse Source 
Support pip/pip3 binary names for setup
This commit is contained in:
kuprel 2022-06-28 11:35:20 -04:00 committed by GitHub
commit 1e2649148b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
setup.sh
View File

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