Skip to content

Commit 83e2c95

Browse files
committed
Add docker checks for debugging
1 parent bd7e210 commit 83e2c95

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

.github/workflows/run-tests-action.yml

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,17 @@ jobs:
4545
uv venv --python 3.10
4646
uv pip install -r tests/requirements.txt
4747
48+
- name: Check Docker version
49+
run: docker --version
50+
51+
- name: Run INFERENCE_DLC container
52+
if: inputs.group == 'aws-g4dn-2xlarge'
53+
run: |
54+
docker run --name test-container -d -it --gpus all -p 8080:8080 ${{ inputs.tgi-dlc }} --model-id TinyLlama/TinyLlama-1.1B-Chat-v1.0
55+
sleep 60
56+
docker stop test-container
57+
docker rm test-container
58+
4859
- name: Run Hugging Face DLCs Tests
4960
run: uv run pytest -s tests/
5061
env:

0 commit comments

Comments
 (0)