Skip to content

Commit b4faffe

Browse files
younesbelkadaglegendre01
andauthoredNov 24, 2023
[Tests] Migrate to AWS runners (huggingface#1185)
* migrate single-gpu runners * Update nightly.yml * Update nightly.yml --------- Co-authored-by: Guillaume LEGENDRE <glegendre01@gmail.com>
1 parent 19145bb commit b4faffe

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed
 

‎.github/workflows/nightly.yml

+6-10
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
run_all_tests_single_gpu:
1818
strategy:
1919
fail-fast: false
20-
runs-on: [self-hosted, docker-gpu, multi-gpu]
20+
runs-on: [self-hosted, single-gpu, nvidia-gpu, t4, ci]
2121
env:
2222
CUDA_VISIBLE_DEVICES: "0"
2323
TEST_TYPE: "single_gpu"
@@ -26,14 +26,12 @@ jobs:
2626
options: --gpus all --shm-size "16gb" -e NVIDIA_DISABLE_REQUIRE=true
2727
defaults:
2828
run:
29-
working-directory: peft/
3029
shell: bash
3130
steps:
32-
- name: Update clone & pip install
31+
- uses: actions/checkout@v3
32+
- name: Pip install
3333
run: |
3434
source activate peft
35-
git config --global --add safe.directory '*'
36-
git fetch && git checkout ${{ github.sha }}
3735
pip install -e . --no-deps
3836
pip install pytest-reportlog
3937
@@ -61,7 +59,7 @@ jobs:
6159
run_all_tests_multi_gpu:
6260
strategy:
6361
fail-fast: false
64-
runs-on: [self-hosted, docker-gpu, multi-gpu]
62+
runs-on: [self-hosted, multi-gpu, nvidia-gpu, t4, ci]
6563
env:
6664
CUDA_VISIBLE_DEVICES: "0,1"
6765
TEST_TYPE: "multi_gpu"
@@ -70,14 +68,12 @@ jobs:
7068
options: --gpus all --shm-size "16gb" -e NVIDIA_DISABLE_REQUIRE=true
7169
defaults:
7270
run:
73-
working-directory: peft/
7471
shell: bash
7572
steps:
76-
- name: Update clone
73+
- uses: actions/checkout@v3
74+
- name: Pip install
7775
run: |
7876
source activate peft
79-
git config --global --add safe.directory '*'
80-
git fetch && git checkout ${{ github.sha }}
8177
pip install -e . --no-deps
8278
pip install pytest-reportlog
8379

0 commit comments

Comments
 (0)
Please sign in to comment.