From 15ec8c850d7f216288a65ae2baee66c4e32d06bf Mon Sep 17 00:00:00 2001
From: Tryorish <2575550173@qq.com>
Date: Thu, 14 May 2026 10:28:10 +0800
Subject: [PATCH 1/2] [Metax][CI] update metax ci

---
 tests/metax_ci/test_ernie_28b_vl.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/metax_ci/test_ernie_28b_vl.py b/tests/metax_ci/test_ernie_28b_vl.py
index 1aafa77026f..f78422bab9a 100644
--- a/tests/metax_ci/test_ernie_28b_vl.py
+++ b/tests/metax_ci/test_ernie_28b_vl.py
@@ -31,7 +31,7 @@ def setUpClass(cls):
 
         cls.llm = LLM(
             model=cls.model_path,
-            tensor_parallel_size=2,
+            tensor_parallel_size=1,
             engine_worker_queue_port=8899,
             max_model_len=32768,
             quantization="wint8",

From 3e34a28547d709fc102817d8740615ede3bd87b4 Mon Sep 17 00:00:00 2001
From: Tryorish <2575550173@qq.com>
Date: Thu, 14 May 2026 15:07:23 +0800
Subject: [PATCH 2/2] [Metax][CI] update metax ci

---
 tests/metax_ci/test_ernie_28b_vl.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/metax_ci/test_ernie_28b_vl.py b/tests/metax_ci/test_ernie_28b_vl.py
index f78422bab9a..7eabe232ef4 100644
--- a/tests/metax_ci/test_ernie_28b_vl.py
+++ b/tests/metax_ci/test_ernie_28b_vl.py
@@ -31,13 +31,13 @@ def setUpClass(cls):
 
         cls.llm = LLM(
             model=cls.model_path,
-            tensor_parallel_size=1,
+            tensor_parallel_size=2,
             engine_worker_queue_port=8899,
             max_model_len=32768,
             quantization="wint8",
             disable_custom_all_reduce=True,
             # enable_prefix_caching=False,
-            # graph_optimization_config={"use_cudagraph": False, "graph_opt_level": 0},
+            graph_optimization_config={"use_cudagraph": False, "graph_opt_level": 0},
             limit_mm_per_prompt={"image": 100},
             reasoning_parser="ernie-45-vl",
             load_choices="default_v1",