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: | run: |
ls flash-attention/dist ls flash-attention/dist
- name: Install Hugging Face CLI - uses: actions/upload-artifact@v4
run: | with:
pip install huggingface_hub name: flash-attention-wheels
path: flash-attention/dist/${{ env.wheel_name }}
- 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/*