Skip to content

Commit

Permalink
flash attn installation in daily testcase (#1272)
Browse files Browse the repository at this point in the history
* Update daily-run-test.yml

* Update daily-run-test.yml

* Update oc_score_baseline.yaml
  • Loading branch information
zhulinJulia24 committed Jun 24, 2024
1 parent e5ee164 commit 26d077b
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
8 changes: 4 additions & 4 deletions .github/scripts/oc_score_baseline.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,9 @@ chatglm3-6b-base-hf:
openbookqa_fact: 92.00

internlm2-7b-hf:
ARC-c: 34.92
ARC-c: 36.27
chid-dev: 55.94
chid-test: 53.70
openai_humaneval: 44.51
openbookqa: 83.00
openbookqa_fact: 83.00
openai_humaneval: 45.12
openbookqa: 80.00
openbookqa_fact: 86.40
3 changes: 2 additions & 1 deletion .github/workflows/daily-run-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,14 @@ jobs:
conda create -y --name ${{env.CONDA_ENV}} python=3.10
conda activate ${{env.CONDA_ENV}}
pip install torch torchvision torchaudio --cache-dir ${{env.PIP_CACHE_PATH}} --index-url https://download.pytorch.org/whl/cu118
pip install /cpfs01/user/qa-llm-cicd/packages/flash_attn-*.whl
conda info --envs
- name: Prepare - Pip install code
run: |
eval "$(conda shell.bash hook)"
conda activate ${{env.CONDA_ENV}}
pip install -e . --cache-dir ${{env.PIP_CACHE_PATH}}
pip install human_eval transformers==4.33.0 protobuf --cache-dir ${{env.PIP_CACHE_PATH}}
pip install human_eval transformers protobuf --cache-dir ${{env.PIP_CACHE_PATH}}
conda info --envs
- name: Prepare - prepare data and hf model
run: |
Expand Down

0 comments on commit 26d077b

Please sign in to comment.