phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-18 16:49:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

temp commit

Browse source
This commit is contained in:
Botao Chen 2024-12-16 21:43:30 -08:00
parent 30f6eb282f
commit 81e1957446
10 changed files with 54 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -69,7 +69,6 @@ def pytest_generate_tests(metafunc):
else:
params = MODEL_PARAMS
# print("params", params)
metafunc.parametrize(
"inference_model",
params,
@ -83,7 +82,5 @@ def pytest_generate_tests(metafunc):
"inference": INFERENCE_FIXTURES,
},
):
# print("I reach here")
fixtures = [stack.values[0]["inference"] for stack in filtered_stacks]
print("fixtures", fixtures)
metafunc.parametrize("inference_stack", fixtures, indirect=True)

4
llama_stack/providers/tests/inference/test_model_registration.py
View file

@ -9,8 +9,8 @@ import pytest
# How to run this test:
#
# pytest -v -s llama_stack/providers/tests/inference/test_model_registration.py
# -m "meta_reference"
# torchrun $CONDA_PREFIX/bin/pytest -v -s -k "meta_reference" --inference-model="Llama3.1-8B-Instruct"
# ./llama_stack/providers/tests/inference/test_model_registration.py
class TestModelRegistration: