github/fauxpilot
1
0
Fork 0
mirror of https://github.com/fauxpilot/fauxpilot.git synced 2025-07-15 01:23:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Another try, fixing missing env vars

Browse source
This commit is contained in:
Brendan Dolan-Gavitt 2022-11-23 20:49:41 -05:00
parent 2178d9e9de
commit f9a4d4f528
2 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
.github/workflows/python_backend_tests.yaml vendored
View file

@ -14,7 +14,9 @@ jobs:
pip install -r tests/python_backend/requirements.txt
- name: Build container
run: |
docker compose build
cp tests/python_backend/runner.env .env &&
docker compose build &&
rm -f .env
- name: Run tests
run: pytest tests

8
tests/python_backend/runner.env Normal file
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