Update tests/lora/conftest.py

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
vllm-project · Jul 16, 2024 · 8431063 · 8431063
1 parent 978688c
commit 8431063
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/lora/conftest.py b/tests/lora/conftest.py
@@ -168,6 +168,9 @@ def sql_lora_huggingface_id():
     # huggingface repo id is used to test lora runtime downloading.
     return "yard1/llama-2-7b-sql-lora-test"
 
+@pytest.fixture(scope="session")
+def sql_lora_files(sql_lora_huggingface_id):
+    return snapshot_download(repo_id=sql_lora_huggingface_id)
 
 @pytest.fixture(scope="session")
 def mixtral_lora_files():