fix

wxsIcey · wxsIcey · commit 0a27f033ca92 · 2025-07-29T07:06:34.000Z
Signed-off-by: Icey &lt;1790571317@qq.com&gt;
diff --git a/tests/e2e/singlecard/models/configs/Qwen2.5-VL-7B-Instruct.yaml b/tests/e2e/singlecard/models/configs/Qwen2.5-VL-7B-Instruct.yaml
@@ -6,3 +6,4 @@ tasks:
   - name: "acc,none"
     value: 0.51
 max_model_len: 8192
+gpu_memory_utilization: 0.6
diff --git a/tests/e2e/singlecard/models/test_lm_eval_correctness.py b/tests/e2e/singlecard/models/test_lm_eval_correctness.py
@@ -43,7 +43,7 @@ def build_model_args(eval_config, tp_size):
         "trust_remote_code": trust_remote_code,
         "max_model_len": max_model_len,
     }
-    for s in ["max_images"]:
+    for s in ["max_images", "gpu_memory_utilization"]:
         val = eval_config.get(s, None)
         if val:
             model_args[s] = val

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ def build_model_args(eval_config, tp_size):`
`43`	`43`	`"trust_remote_code": trust_remote_code,`
`44`	`44`	`"max_model_len": max_model_len,`
`45`	`45`	`}`
`46`		`- for s in ["max_images"]:`
	`46`	`+ for s in ["max_images", "gpu_memory_utilization"]:`
`47`	`47`	`val = eval_config.get(s, None)`
`48`	`48`	`if val:`
`49`	`49`	`model_args[s] = val`