[CI/Build] Fix CI LoRA failure (#16270)

Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
2025-04-09 09:13:56 +08:00
parent 2755c34a8f
commit 86c3369eb8
8 changed files with 19 additions and 13 deletions
--- a/tests/lora/test_baichuan.py
+++ b/tests/lora/test_baichuan.py
@@ -73,7 +73,6 @@ def test_baichuan_tensor_parallel_equality(baichuan_lora_files,
                       max_num_seqs=16,
                       max_loras=4,
                       max_lora_rank=64,
-                       tensor_parallel_size=1,
                       trust_remote_code=True,
                       fully_sharded_loras=fully_sharded)
    output_tp1 = do_sample(llm_tp1, baichuan_lora_files, lora_id=1)