Skip to content

Commit

Permalink
Fixing accelerate missing deps
Browse files Browse the repository at this point in the history
  • Loading branch information
PicoCreator authored Jul 19, 2024
1 parent c5fa774 commit 97f05fd
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions .github/actions/gh-task-runner-composite/action.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -75,10 +75,12 @@ runs:
# Needed for sealion model
python -m pip install sentencepiece
# Needed for RWKV v6
pip install -U git+https://github.com/sustcsonglin/flash-linear-attention
pip install triton==2.2.0
# Some lm eval install issues
pip install peft
python -m pip install flash-attn
python -m pip install -U git+https://github.com/sustcsonglin/flash-linear-attention
python -m pip install triton==2.2.0
# Some huggingface accelerate missing packages issue (WTH)
python -m pip install peft sqlitedict accelerate evaluate sacrebleu scikit-learn pytablewriter
# Setup HF cache
chmod +x ./gh-task-runner/*.sh
Expand Down

0 comments on commit 97f05fd

Please sign in to comment.