biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI/Build] Replaced some models on tests for smaller ones (#9570)

Browse Source 
Signed-off-by: Wallas Santos <wallashss@ibm.com>
This commit is contained in:
Wallas Henrique
2024-10-22 01:52:14 -03:00
committed by GitHub
parent 74692421f7
commit c0292211ce
8 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/entrypoints/openai/test_chat.py
View File

@@ -16,9 +16,6 @@ from .test_completion import zephyr_lora_files # noqa: F401
# any model with a chat template should work here
MODEL_NAME = "HuggingFaceH4/zephyr-7b-beta"
# technically this needs Mistral-7B-v0.1 as base, but we're not testing
# generation quality here
LORA_NAME = "typeof/zephyr-7b-beta-lora"
@pytest.fixture(scope="module")