[CI/Build] Remove BaiChuanForCausalLM from the LoRA test (#46494)
Signed-off-by: Jee Jee Li <jeejeelee@inferact.ai>
Jee Jee Li committed
Jun 23, 2026 at 22:09 UTC
9f6f2964287ae0a844b55f0c509c04d8fa139e2d
1 file changed
+12
-9
tests/lora/test_lora_checkpoints.py
+12
-9
@@ -6,7 +6,6 @@ import pytest
6
from vllm.lora.lora_model import LoRAModel
7
from vllm.lora.peft_helper import PEFTHelper
8
from vllm.lora.utils import parse_fine_tuned_lora_name
9
-from vllm.model_executor.models.baichuan import BaiChuanBaseForCausalLM
9
from vllm.model_executor.models.gemma4 import Gemma4ForCausalLM
10
from vllm.model_executor.models.utils import WeightsMapper
11
@@ -18,6 +17,14 @@ BAICHUAN_LORA_MODULES = [
17
"down_proj",
18
]
19
20
+MOCK_PACKED_MAPPING = {
21
+ "W_pack": ["W_pack"],
22
+ "gate_up_proj": [
23
+ "gate_proj",
24
+ "up_proj",
25
+ ],
26
+}
27
+
28
29
@pytest.mark.parametrize("lora_name", lora_lst)
30
def test_load_checkpoints(
@@ -27,12 +34,10 @@ def test_load_checkpoints(
34
baichuan_regex_lora_files,
35
chatglm3_lora_files,
36
):
30
- packed_modules_mapping = BaiChuanBaseForCausalLM.packed_modules_mapping
31
-
37
expected_lora_lst: list[str] = []
38
for module in BAICHUAN_LORA_MODULES:
34
- if module in packed_modules_mapping:
35
- expected_lora_lst.extend(packed_modules_mapping[module])
39
+ if module in MOCK_PACKED_MAPPING:
40
+ expected_lora_lst.extend(MOCK_PACKED_MAPPING[module])
41
else:
42
expected_lora_lst.append(module)
43
expected_lora_modules = set(expected_lora_lst)
@@ -98,12 +103,10 @@ def test_load_checkpoints(
103
104
105
def test_lora_weights_mapping(baichuan_lora_files):
101
- packed_modules_mapping = BaiChuanBaseForCausalLM.packed_modules_mapping
102
-
106
expected_lora_lst: list[str] = []
107
for module in BAICHUAN_LORA_MODULES:
105
- if module in packed_modules_mapping:
106
- expected_lora_lst.extend(packed_modules_mapping[module])
108
+ if module in MOCK_PACKED_MAPPING:
109
+ expected_lora_lst.extend(MOCK_PACKED_MAPPING[module])
110
else:
111
expected_lora_lst.append(module)
112
expected_lora_modules = set(expected_lora_lst)