Skip to content

Commit 10ee978

Browse files
JeffwanYard1
andcommitted
Update tests/lora/conftest.py
Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
1 parent a16ec5d commit 10ee978

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/lora/conftest.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,9 @@ def sql_lora_huggingface_id():
168168
# huggingface repo id is used to test lora runtime downloading.
169169
return "yard1/llama-2-7b-sql-lora-test"
170170

171+
@pytest.fixture(scope="session")
172+
def sql_lora_files(sql_lora_huggingface_id):
173+
return snapshot_download(repo_id=sql_lora_huggingface_id)
171174

172175
@pytest.fixture(scope="session")
173176
def mixtral_lora_files():

0 commit comments

Comments
 (0)