github/fauxpilot
1
0
Fork 0
mirror of https://github.com/fauxpilot/fauxpilot.git synced 2025-08-19 21:04:08 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Check for docker-compose and use docker compose if not found

Browse source 
Fixes #8.
This commit is contained in:
Brendan Dolan-Gavitt 2022-08-04 17:20:55 -04:00
commit 9e7fb28b65
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
launch.sh
View file

@ -7,7 +7,13 @@ if [ ! -f config.env ]; then
fi fi
source config.env source config.env
# On newer versions, docker-compose is docker compose
DOCKER_COMPOSE=$(command -v docker-compose)
if [ -z "$DOCKER_COMPOSE" ]; then
DOCKER_COMPOSE="docker compose"
fi
export NUM_GPUS=${NUM_GPUS} export NUM_GPUS=${NUM_GPUS}
export MODEL_DIR="${MODEL_DIR}"/"${MODEL}-${NUM_GPUS}gpu" export MODEL_DIR="${MODEL_DIR}"/"${MODEL}-${NUM_GPUS}gpu"
export GPUS=$(seq 0 $(( NUM_GPUS - 1 )) | paste -sd ',') export GPUS=$(seq 0 $(( NUM_GPUS - 1 )) | paste -sd ',')
docker-compose up $DOCKER_COMPOSE up