From ec8c638d2b126cd458af4aafd1964ed493308945 Mon Sep 17 00:00:00 2001
From: drbh <david.richard.holtz@gmail.com>
Date: Tue, 21 May 2024 22:46:54 +0000
Subject: [PATCH] feat: cache wheel as build artifact

---
 .github/workflows/python-packaging.yaml | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/python-packaging.yaml b/.github/workflows/python-packaging.yaml
index dfac565b..422b636a 100644
--- a/.github/workflows/python-packaging.yaml
+++ b/.github/workflows/python-packaging.yaml
@@ -166,11 +166,7 @@ jobs:
         run: |
           ls flash-attention/dist
 
-      - name: Install Hugging Face CLI
-        run: |
-          pip install huggingface_hub
-
-      - name: Upload to Hugging Face Hub
-        run: |
-          export HUGGING_FACE_HUB_TOKEN=${{ secrets.HUGGING_FACE_HUB_TOKEN }}
-          huggingface-cli upload drbh/flash-attention-pre-compile flash-attention/dist/*
+      - uses: actions/upload-artifact@v4
+        with:
+          name: flash-attention-wheels
+          path: flash-attention/dist/${{ env.wheel_name }}