Skip to content

Commit 05585e9

Browse files
author
Vincent Li
committed
update
1 parent f17ac19 commit 05585e9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

.github/ci_config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ platforms:
106106
timeout: 3600
107107
env:
108108
MODEL_LIST: 9g_8b_thinking
109-
TEST_PARAM: ['--attn=default', '--attn=flash-attn --enable-paged-attn --cache-type=paged --block-size=256 --max-new-tokens=256', '--attn=flash-attn --enable-paged-attn --cache-type=paged --block-size=256 --max-new-tokens=256']
109+
TEST_PARAM: ['--attn=default', '--attn=page --enable-paged-attn --cache-type=paged --block-size=256 --max-new-tokens=256', '--attn=page --enable-paged-attn --cache-type=paged --block-size=256 --max-new-tokens=256']
110110
stages:
111111
- name: test
112112
run: python -u "InfiniLM/examples/test_infer.py" --device metax --model=/data-aisoft/mechdancer/models/9g_8b_thinking/ --backend cpp --prompt "Hello" <TEST_PARAM>

0 commit comments

Comments
 (0)