Skip to content

Commit 4a0f503

Browse files
fix error info
Signed-off-by: shikang-hangzhou <459956190@qq.com>
1 parent 30aad6e commit 4a0f503

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

tests/multicard/test_offline_inference_distributed.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -138,16 +138,8 @@ def test_models_distributed_DeepSeek_w8a8_ep_dbo():
138138
tensor_parallel_size=4,
139139
enforce_eager=True,
140140
enable_expert_parallel=True,
141-
distributed_executor_backend="mp",
142-
additional_config={
143-
"ascend_scheduler_config": {
144-
"enabled": True,
145-
},
146-
"torchair_graph_config": {
147-
"enable": True,
148-
"enable_multistream_moe": True
149-
}
150-
}) as vllm_model:
141+
distributed_executor_backend="mp"
142+
) as vllm_model:
151143
model_arch = 'DeepseekV2ForCausalLM'
152144
registed_models = ModelRegistry.models
153145
assert registed_models[

0 commit comments

Comments
 (0)