summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGuillaume LEGENDRE <glegendre01@gmail.com>2024-09-26 18:48:52 +0200
committerGitHub <noreply@github.com>2024-09-26 18:48:52 +0200
commitc3c392f45c14f60eb4fb8397cc5c1d3891c9656d (patch)
tree96586b918b5c9803b1c6dbb398513caf325e9f43
parent10d47183c088ce449da13d74f07171c8106cd6dd (diff)
parenta0184a4fe43a0961a5977304c840f03783c3c716 (diff)
downloadcandle-c3c392f45c14f60eb4fb8397cc5c1d3891c9656d.tar.gz
candle-c3c392f45c14f60eb4fb8397cc5c1d3891c9656d.tar.bz2
candle-c3c392f45c14f60eb4fb8397cc5c1d3891c9656d.zip
Merge pull request #2507 from huggingface/ci-move
move CI/Cuda runner
-rw-r--r--.github/workflows/ci_cuda.yaml3
1 files changed, 2 insertions, 1 deletions
diff --git a/.github/workflows/ci_cuda.yaml b/.github/workflows/ci_cuda.yaml
index 0c874e24..fc07f112 100644
--- a/.github/workflows/ci_cuda.yaml
+++ b/.github/workflows/ci_cuda.yaml
@@ -9,7 +9,8 @@ jobs:
concurrency:
group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true
- runs-on: [single-gpu, nvidia-gpu, t4, ci]
+ runs-on:
+ group: aws-g4dn-2xlarge
container:
image: nvidia/cuda:12.3.1-devel-ubuntu22.04
options: --gpus 0