diff --git a/script/run-mlperf-inference-app/customize.py b/script/run-mlperf-inference-app/customize.py index 0eb089abdd..15c54e8bdb 100644 --- a/script/run-mlperf-inference-app/customize.py +++ b/script/run-mlperf-inference-app/customize.py @@ -108,12 +108,12 @@ def preprocess(i): env['OUTPUT_BASE_DIR'] = env.get('CM_MLPERF_INFERENCE_RESULTS_DIR', os.getcwd()) - test_list = ["TEST01", "TEST05"] + test_list = ["TEST01"] if env['CM_MODEL'] in ["resnet50"]: test_list.append("TEST04") if "gpt" in env['CM_MODEL'] or "llama2-70b" in env['CM_MODEL'] or "mixtral-8x7b" in env['CM_MODEL']: test_list.remove("TEST01") - test_list.remove("TEST05") + #test_list.remove("TEST05") if "llama2" in env['CM_MODEL'].lower() or "mixtral-8x7b" in env['CM_MODEL']: test_list.append("TEST06")