diff --git a/.github/workflows/tests_latest.yml b/.github/workflows/tests_latest.yml index cbac7bce749..53dcbc90fcd 100644 --- a/.github/workflows/tests_latest.yml +++ b/.github/workflows/tests_latest.yml @@ -16,9 +16,7 @@ jobs: tests: name: Tests latest TRL release with dev dependencies runs-on: - # TODO: move to aws-g6e-4xlarge (L40S) once v1.11 ships; v1.10-release predates the flash-attn2 xfails and - # would fail on Ada hardware - group: aws-g4dn-2xlarge + group: aws-g6e-4xlarge container: image: pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel options: --gpus all