Skip to content

Commit a782946

Browse files
Disable logits diff test on MPT GPU for now
1 parent 7902951 commit a782946

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/llm/test/run-llm-inference-tests-gpu.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ start=$(date "+%s")
2929
source ${ANALYTICS_ZOO_ROOT}/python/llm/test/run-llm-check-function.sh
3030

3131
pytest_check_error pytest ${LLM_INFERENCE_TEST_DIR}/test_transformers_api.py -v -s
32-
pytest_check_error pytest ${LLM_INFERENCE_TEST_DIR}/test_transformers_api_final_logits.py -v -s
32+
# Disable for now
33+
# pytest_check_error pytest ${LLM_INFERENCE_TEST_DIR}/test_transformers_api_final_logits.py -v -s
3334
pytest_check_error pytest ${LLM_INFERENCE_TEST_DIR}/test_transformers_api_attention.py -v -s
3435
pytest_check_error pytest ${LLM_INFERENCE_TEST_DIR}/test_transformers_api_mlp.py -v -s
3536
pytest_check_error pytest ${LLM_INFERENCE_TEST_DIR}/test_transformers_api_RMSNorm.py -v -s

0 commit comments

Comments
 (0)