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
This commit is contained in:
Brendan Dolan-Gavitt 2022-11-23 21:01:29 -05:00 committed by GitHub
commit 1b2b8ec251
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 31 additions and 1 deletions

View file

@ -0,0 +1,8 @@
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