From 16e8572019e28fbc2cf0a612d39900bd3d7d8501 Mon Sep 17 00:00:00 2001 From: hfadzxy Date: Fri, 5 Dec 2025 10:55:48 +0800 Subject: [PATCH] [Fix] fix llava-1.5-7b-hf & Qwen2-Audio-7B-Instruct accuracy test Signed-off-by: hfadzxy --- tests/e2e/models/configs/Qwen2-Audio-7B-Instruct.yaml | 1 + tests/e2e/models/configs/llava-1.5-7b-hf.yaml | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/e2e/models/configs/Qwen2-Audio-7B-Instruct.yaml b/tests/e2e/models/configs/Qwen2-Audio-7B-Instruct.yaml index 2dd02ab434e..d1d9d421b55 100644 --- a/tests/e2e/models/configs/Qwen2-Audio-7B-Instruct.yaml +++ b/tests/e2e/models/configs/Qwen2-Audio-7B-Instruct.yaml @@ -9,3 +9,4 @@ tasks: value: 0.45 num_fewshot: 5 gpu_memory_utilization: 0.8 +enforce_eager: True diff --git a/tests/e2e/models/configs/llava-1.5-7b-hf.yaml b/tests/e2e/models/configs/llava-1.5-7b-hf.yaml index 7bd69de99f7..95ab52695ea 100644 --- a/tests/e2e/models/configs/llava-1.5-7b-hf.yaml +++ b/tests/e2e/models/configs/llava-1.5-7b-hf.yaml @@ -9,3 +9,4 @@ tasks: trust_remote_code: True gpu_memory_utilization: 0.8 dtype: "bfloat16" +enforce_eager: True