Skip to content

Commit b4bc64a

Browse files
Add failing cases for lora name parsing
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
1 parent 6a5730d commit b4bc64a

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/lora/test_utils.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,18 @@ def test_parse_fine_tuned_lora_name_valid():
3939
False,
4040
False,
4141
),
42+
(
43+
"language_model.layers.9.mlp.down_proj.lora_A.weight",
44+
"language_model.layers.9.mlp.down_proj",
45+
True,
46+
False,
47+
),
48+
(
49+
"language_model.layers.9.mlp.down_proj.lora_B.weight",
50+
"language_model.layers.9.mlp.down_proj",
51+
False,
52+
False,
53+
),
4254
}
4355
for name, module_name, is_lora_a, is_bias in fixture:
4456
assert (module_name, is_lora_a,

0 commit comments

Comments
 (0)