From 9a677233d9d94a7d8ff71f2424d9b7acf8cb2a4e Mon Sep 17 00:00:00 2001 From: yaofengchen Date: Thu, 7 May 2026 07:14:54 +0000 Subject: [PATCH] update ci --- .../e2e/test_mixed_prefill_precision.py | 4 +++- tests/test_lmdeploy/utils/pipeline_chat.py | 17 +++++++++++++---- 2 files changed, 16 insertions(+), 5 deletions(-) diff --git a/tests/test_lmdeploy/e2e/test_mixed_prefill_precision.py b/tests/test_lmdeploy/e2e/test_mixed_prefill_precision.py index 60bc8345..5e48d44b 100644 --- a/tests/test_lmdeploy/e2e/test_mixed_prefill_precision.py +++ b/tests/test_lmdeploy/e2e/test_mixed_prefill_precision.py @@ -178,7 +178,9 @@ def test_mixed_prefill_precision(config, eager_mode): pipe = None try: - pipe = pipeline(model_path, backend_config=backend_config) + pipe = pipeline( + model_path, backend_config=backend_config, trust_remote_code=True + ) responses = pipe( prompts, gen_config=gen_config, diff --git a/tests/test_lmdeploy/utils/pipeline_chat.py b/tests/test_lmdeploy/utils/pipeline_chat.py index efa3cb92..c2638dd1 100644 --- a/tests/test_lmdeploy/utils/pipeline_chat.py +++ b/tests/test_lmdeploy/utils/pipeline_chat.py @@ -41,7 +41,7 @@ def run_pipeline_chat_test( block_size=block_size, ) print("backend_config: ", backend_config) - pipe = pipeline(hf_path, backend_config=backend_config) + pipe = pipeline(hf_path, backend_config=backend_config, trust_remote_code=True) # run testcases gen_config = GenerationConfig(top_k=1) @@ -67,7 +67,9 @@ def run_pipeline_chat_test( "reproduce config info:", "engine_config = " + str(backend_config), "gen_config = " + str(gen_config), - 'pipe = pipeline("' + hf_path + '", backend_config=engine_config)', + 'pipe = pipeline("' + + hf_path + + '", backend_config=engine_config, trust_remote_code=True)', 'res = pipe("Hi, pls introduce shanghai", gen_config=gen_config)', ] ) @@ -215,13 +217,20 @@ def run_pipeline_vl_chat_test(config, model_case, device_type, eager_mode=True): ) print("backend_config: ", backend_config) gen_config = GenerationConfig(top_k=1) - pipe = pipeline(hf_path, backend_config=backend_config, gen_config=gen_config) + pipe = pipeline( + hf_path, + backend_config=backend_config, + gen_config=gen_config, + trust_remote_code=True, + ) log_string = "\n".join( [ "reproduce config info:", "engine_config = " + str(backend_config), - 'pipe = pipeline("' + hf_path + '", backend_config=engine_config)', + 'pipe = pipeline("' + + hf_path + + '", backend_config=engine_config, trust_remote_code=True)', ] ) print("log config: ", log_string)