feat: cache wheel as build artifact

This commit is contained in:
drbh 2024-05-21 22:46:54 +00:00
parent 7765aa6ecd
commit ec8c638d2b

View File

@ -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 }}