Skip to content

Commit

Permalink
Add docker checks for debugging
Browse files Browse the repository at this point in the history
  • Loading branch information
alvarobartt committed Sep 2, 2024
1 parent bd7e210 commit 83e2c95
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions .github/workflows/run-tests-action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,17 @@ jobs:
uv venv --python 3.10
uv pip install -r tests/requirements.txt
- name: Check Docker version
run: docker --version

- name: Run INFERENCE_DLC container
if: inputs.group == 'aws-g4dn-2xlarge'
run: |
docker run --name test-container -d -it --gpus all -p 8080:8080 ${{ inputs.tgi-dlc }} --model-id TinyLlama/TinyLlama-1.1B-Chat-v1.0
sleep 60
docker stop test-container
docker rm test-container
- name: Run Hugging Face DLCs Tests
run: uv run pytest -s tests/
env:
Expand Down

0 comments on commit 83e2c95

Please sign in to comment.