Temporarily checkout PR branch in the nv-accelerate-v100 pipeline

deepspeedai · May 30, 2024 · a04de7f · a04de7f
1 parent 8c982d2
commit a04de7f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/nv-accelerate-v100.yml b/.github/workflows/nv-accelerate-v100.yml
@@ -47,6 +47,7 @@ jobs:
           unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
           git clone https://github.com/huggingface/accelerate
           cd accelerate
+          git fetch origin pull/2814/head:pr-2814 && git checkout pr-2814
           git rev-parse --short HEAD
           # installing dependencies
           pip install .[testing]