[Model] Add Phi-2 LoRA support (#4886)

2024-05-21 13:24:17 +08:00
parent d130b573a0
commit f12c3b5b3d
4 changed files with 99 additions and 6 deletions
--- a/tests/lora/conftest.py
+++ b/tests/lora/conftest.py
@ -165,6 +165,11 @@ def tinyllama_lora_files():
    return snapshot_download(repo_id="jashing/tinyllama-colorist-lora")


+@pytest.fixture(scope="session")
+def phi2_lora_files():
+    return snapshot_download(repo_id="isotr0py/phi-2-test-sql-lora")
+
+
@pytest.fixture(scope="session")
 def long_context_lora_files_16k_1():
    return snapshot_download(repo_id="SangBinCho/long_context_16k_testing_1")