Initial commit

This commit is contained in:
Brendan Dolan-Gavitt 2022-08-02 21:47:27 -04:00
commit a23a1478f7
65 changed files with 157042 additions and 0 deletions

13
launch.sh Executable file
View file

@ -0,0 +1,13 @@
#!/bin/bash
# Read in config.env file; error if not found
if [ ! -f config.env ]; then
echo "config.env not found, please run setup.sh"
exit 1
fi
source config.env
export NUM_GPUS=${NUM_GPUS}
export MODEL_DIR="${MODEL_DIR}"/"${MODEL}-${NUM_GPUS}gpu"
export GPUS=$(seq 0 $(( NUM_GPUS - 1 )) | paste -sd ',')
docker-compose up