fight with CI

axolotl-ai-cloud · Dec 21, 2023 · 6f9891c · 6f9891c
1 parent d42881c
commit 6f9891c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -61,6 +61,7 @@ jobs:
           tags: |
             ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
             ${{ (matrix.is_latest) && format('{0}-latest', steps.metadata.outputs.tags) || '' }}
+          cache-to: type=local,dest=/tmp/buildx-cache,mode=max
       - name: Unit Tests
         run: |
           docker run --rm ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }} pytest --ignore=tests/e2e/ /workspace/axolotl/tests/
@@ -74,6 +75,10 @@ jobs:
             ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
             ${{ (matrix.is_latest) && format('{0}-latest', steps.metadata.outputs.tags) || '' }}
           labels: ${{ steps.metadata.outputs.labels }}
+          cache-from: type=local,src=/tmp/buildx-cache
+      - name: delete the cache
+        run: |
+            rm -rf /tmp/buildx-cache
 
   build-axolotl-runpod:
     needs: build-axolotl