phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

update tests --inference-model to hf id

Browse source
This commit is contained in:
Xi Yan 2024-11-18 17:36:58 -08:00
parent 939056e265
commit 50d539e6d7
5 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/tests/inference/conftest.py
View file

@ -32,8 +32,12 @@ def pytest_configure(config):
MODEL_PARAMS = [
pytest.param("Llama3.1-8B-Instruct", marks=pytest.mark.llama_8b, id="llama_8b"),
pytest.param("Llama3.2-3B-Instruct", marks=pytest.mark.llama_3b, id="llama_3b"),
pytest.param(
"meta-llama/Llama-3.1-8B-Instruct", marks=pytest.mark.llama_8b, id="llama_8b"
),
pytest.param(
"meta-llama/Llama-3.2-3B-Instruct", marks=pytest.mark.llama_3b, id="llama_3b"
),
]
VISION_MODEL_PARAMS = [