Update deepspeed.yaml

Ascend · Nov 4, 2024 · 4477b0b · 4477b0b
1 parent c3001d1
commit 4477b0b
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/.github/workflows/deepspeed.yaml b/.github/workflows/deepspeed.yaml
@@ -11,11 +11,7 @@ on:
       - 'requirements/**'
   schedule:
     - cron: "0 0 * * *"
-  push:
-    paths:
-      - '.github/workflows/deepspeed.yaml'
 
-
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true
@@ -70,13 +66,18 @@ jobs:
         EOF
 
     - name: Install transformers
-      run: |
-        source /root/.bashrc
-        echo "y" | apt-get install git
-        git clone https://github.com/huggingface/transformers
-        cd transformers
-        git rev-parse --short HEAD
-        pip install .
+      uses: nick-fields/retry@v3
+      with:
+        timeout_minutes: 30
+        max_attempts: 3
+        retry_on: error
+        command: | 
+          source /root/.bashrc
+          echo "y" | apt-get install git
+          git clone https://github.com/huggingface/transformers
+          cd transformers
+          git rev-parse --short HEAD
+          pip install .
              
     - name: Install deepspeed
       uses: nick-fields/retry@v3
@@ -129,4 +130,4 @@ jobs:
 
 
 
- 
+