apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-22 16:07:51 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

[Metax][CI] update metax ci files (#6364)

Browse Source
This commit is contained in:
MingkunZhang
2026-02-05 17:16:31 +08:00
committed by GitHub
parent e3fb8796b4
commit 6e28b5ef4f
3 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/run_ci_metax.sh
-1
View File
@@ -91,7 +91,6 @@ METAX_CI_CASELIST=(
"tests/operators/test_update_inputs_v1.py"
"tests/operators/test_set_value_by_flags_and_idx.py"
"tests/operators/test_get_token_penalty_multi_scores.py"
"tests/operators/test_speculate_get_token_penalty_multi_scores.py"
"tests/operators/test_token_penalty.py"
"tests/operators/test_stop_generation_multi_ends.py"
"tests/operators/test_get_padding_offset.py"
tests/metax_ci/test_ernie_21b.py
+2 -2
View File
@@ -29,9 +29,9 @@ class TestErnie21B(unittest.TestCase):
max_model_len=256,
quantization="wint8",
load_choices="default_v1",
enable_prefix_caching=False,
# enable_prefix_caching=False,
disable_custom_all_reduce=True,
graph_optimization_config={"use_cudagraph": False, "graph_opt_level": 0},
# graph_optimization_config={"use_cudagraph": False, "graph_opt_level": 0},
)
cls.sampling_params = fastdeploy.SamplingParams(top_p=0.95, max_tokens=256, temperature=0.6)
tests/metax_ci/test_ernie_28b_vl.py
+2 -2
View File
@@ -36,8 +36,8 @@ class TestErnie28BVL(unittest.TestCase):
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},
# enable_prefix_caching=False,
# 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",