Skip to content

Commit

Permalink
Merge pull request #234 from huggingface/ci-move
Browse files Browse the repository at this point in the history
Ci move
  • Loading branch information
glegendre01 authored Sep 26, 2024
2 parents 97c13b0 + 676351b commit 51ca40b
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion .github/workflows/3d_parallelism_unit_tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,8 @@ on:

jobs:
tests:
runs-on: [multi-gpu, nvidia-gpu, 8-t4, ci]
runs-on:
group: aws-g4dn-metal
container:
image: runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04
ports:
Expand Down
3 changes: 2 additions & 1 deletion .github/workflows/fa2_unit_tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@ on:

jobs:
tests:
runs-on: [single-gpu, nvidia-gpu, a10, ci]
runs-on:
group: aws-g5-4xlarge-plus
container:
image: runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04
ports:
Expand Down

0 comments on commit 51ca40b

Please sign in to comment.