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

4
llama_stack/providers/tests/agents/conftest.py
View file

@ -81,13 +81,13 @@ def pytest_addoption(parser):
parser.addoption(
"--inference-model",
action="store",
default="Llama3.1-8B-Instruct",
default="meta-llama/Llama-3.1-8B-Instruct",
help="Specify the inference model to use for testing",
)
parser.addoption(
"--safety-shield",
action="store",
default="Llama-Guard-3-8B",
default="meta-llama/Llama-Guard-3-8B",
help="Specify the safety shield to use for testing",
)