misc(ci):run the tests on GPU instance

This commit is contained in:
Morgan Funtowicz 2025-01-08 23:59:08 +01:00
parent d7698e851b
commit c4abc206f4

View File

@ -37,7 +37,7 @@ jobs:
group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }} group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }}
cancel-in-progress: true cancel-in-progress: true
runs-on: runs-on:
group: aws-highmemory-64-plus-priv group: aws-g6-12xl-plus-priv-cache
container: container:
image: ${{ needs.build-and-push.outputs.docker_image }} image: ${{ needs.build-and-push.outputs.docker_image }}
credentials: credentials: