CI: set HF caching (#691)

Co-authored-by: Daniel Stancl <46073029+stancld@users.noreply.github.com>
Lightning-AI · Dec 22, 2021 · 63d0d75 · 63d0d75
1 parent 67f78ab
commit 63d0d75
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/.github/workflows/ci_test-base.yml b/.github/workflows/ci_test-base.yml
@@ -17,6 +17,8 @@ jobs:
       matrix:
         os: [ubuntu-20.04, windows-2019, macOS-10.15]
         python-version: [3.7]
+    env:
+      TRANSFORMERS_CACHE: .cache/huggingface/
 
     # Timeout: https://stackoverflow.com/a/59076067/4521646
     timeout-minutes: 20
@@ -64,6 +66,12 @@ jobs:
         pip list
       shell: bash
 
+    - name: HF cache
+      uses: actions/cache@v2
+      with:
+        path: $TRANSFORMERS_CACHE
+        key: cache-transformers
+
     - name: Test Package [only]
       run: |
         # NOTE: run coverage on tests does not propagare faler status for Win, https://github.com/nedbat/coveragepy/issues/1003

diff --git a/.github/workflows/ci_test-conda.yml b/.github/workflows/ci_test-conda.yml
@@ -18,6 +18,8 @@ jobs:
         include:
           - python-version: 3.7
             pytorch-version: '1.3'
+    env:
+      TRANSFORMERS_CACHE: .cache/huggingface/
 
     # Timeout: https://stackoverflow.com/a/59076067/4521646
     timeout-minutes: 55
@@ -60,6 +62,12 @@ jobs:
       # prevent hanging Conda creations
       timeout-minutes: 10
 
+    - name: HF cache
+      uses: actions/cache@v2
+      with:
+        path: $TRANSFORMERS_CACHE
+        key: cache-transformers
+
     - name: Temp COCO env. fix
       if: matrix.pytorch-version == '1.4'
       run: |

diff --git a/.github/workflows/ci_test-full.yml b/.github/workflows/ci_test-full.yml
@@ -24,6 +24,8 @@ jobs:
             requires: 'latest'
           - python-version: 3.9
             requires: 'oldest'
+    env:
+      TRANSFORMERS_CACHE: .cache/huggingface/
 
     # Timeout: https://stackoverflow.com/a/59076067/4521646
     timeout-minutes: 75
@@ -88,6 +90,12 @@ jobs:
         pip list
       shell: bash
 
+    - name: HF cache
+      uses: actions/cache@v2
+      with:
+        path: $TRANSFORMERS_CACHE
+        key: cache-transformers
+
     - name: Tests
       run: |
         phmdoctest README.md --outfile tests/test_readme.py