From d6c2a6a81f20a8bae69c1ec1c11fd6223029fce4 Mon Sep 17 00:00:00 2001 From: Yechan Kim <161688079+yechank-nvidia@users.noreply.github.com> Date: Tue, 19 Aug 2025 16:29:32 +0900 Subject: [PATCH] [https://nvbugs/5448579][fix] EXAONE-4.0 accuracy test bugfix (#6888) Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> --- tests/integration/defs/accuracy/test_llm_api_pytorch.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/integration/defs/accuracy/test_llm_api_pytorch.py b/tests/integration/defs/accuracy/test_llm_api_pytorch.py index ec35ca76f0..310e08cabf 100644 --- a/tests/integration/defs/accuracy/test_llm_api_pytorch.py +++ b/tests/integration/defs/accuracy/test_llm_api_pytorch.py @@ -2373,10 +2373,8 @@ class TestPhi4MM(LlmapiAccuracyTestHarness): class TestEXAONE4(LlmapiAccuracyTestHarness): MODEL_NAME = "LGAI-EXAONE/EXAONE-4.0-32B" - kv_cache_config = KvCacheConfig( - enable_block_reuse=False, - enable_partial_reuse=False, - max_attention_window=[4096, 4096, 4096, 131072]) + kv_cache_config = KvCacheConfig(enable_block_reuse=False, + enable_partial_reuse=False) def test_auto_dtype(self): model_path = f"{llm_models_root()}/EXAONE-4.0-32B"