From fd0e80845440de7e49f863414be1848c32a1d343 Mon Sep 17 00:00:00 2001 From: Jonathan Dahan Date: Tue, 28 Jun 2022 11:20:23 -0400 Subject: [PATCH] Support pip/pip3 binary names for setup --- setup.sh | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/setup.sh b/setup.sh index ee82889..85af197 100644 --- a/setup.sh +++ b/setup.sh @@ -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 \ No newline at end of file +wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14