Brendan Dolan-Gavitt 1b2b8ec251
Add basic CI that tests the Python backend (#124)
* Basic action that runs pytest

* fix indentation

* debugging actions :(

* pull the images in a separate step so it doesn't take so long to build

* Split out the container build

* Another try, fixing missing env vars
2022-11-23 21:01:29 -05:00

9 lines
201 B
Bash

NUM_GPUS=1
GPUS=0
API_EXTERNAL_PORT=5000
TRITON_HOST=triton
TRITON_PORT=8001
MODEL=py-codegen-350M-mono
MODEL_DIR=${HOME}/models/py-Salesforce-codegen-350M-mono
HF_CACHE_DIR=${HOME}/.cache/huggingface