Skip to content

Commit 9456bfb

Browse files
committed
remain necessary settings
Signed-off-by: QiuChunshuo <[email protected]>
1 parent b31062b commit 9456bfb

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/distributed/test_context_parallel.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -118,14 +118,10 @@ def iter_params(self, model_id: str):
118118

119119
CP_TEXT_GENERATION_MODELS = {
120120
"deepseek-ai/DeepSeek-V2-Lite-Chat": [
121-
CPTestSettings.detailed(dcp_multipliers=[0.5, 1]),
122-
CPTestSettings.detailed(tp_base=2, dcp_multipliers=[0.5, 1]),
123-
CPTestSettings.detailed(tp_base=2, dcp_multipliers=[0.5, 1], cp_kv_cache_interleave_size=64),
121+
CPTestSettings.detailed(dcp_multipliers=[0.5, 1], cp_kv_cache_interleave_size=64),
124122
],
125123
"Qwen/Qwen2.5-1.5B-Instruct": [
126-
CPTestSettings.detailed(attn_backend="FLASH_ATTN"),
127124
CPTestSettings.detailed(cp_kv_cache_interleave_size=16, attn_backend="FLASH_ATTN"),
128-
CPTestSettings.detailed(attn_backend="FLASHINFER"),
129125
CPTestSettings.detailed(cp_kv_cache_interleave_size=16, attn_backend="FLASHINFER"),
130126
],
131127
}

0 commit comments

Comments
 (0)