[Misc][LoRA] Support loading LoRA weights for target_modules in reg format (#9275)

2024-10-11 20:31:21 +08:00
parent e808156f30
commit 36ea79079b
4 changed files with 59 additions and 5 deletions
--- a/tests/lora/conftest.py
+++ b/tests/lora/conftest.py
@ -199,6 +199,11 @@ def baichuan_zero_lora_files():
    return snapshot_download(repo_id="jeeejeee/baichuan7b-zero-init")


+@pytest.fixture(scope="session")
+def baichuan_regex_lora_files():
+    return snapshot_download(repo_id="jeeejeee/baichuan-7b-lora-zero-regex")
+
+
@pytest.fixture(scope="session")
 def minicpmv_lora_files():
    return snapshot_download(repo_id="jeeejeee/minicpmv25-lora-pokemon")