This commit is contained in:
Sixian Yi 2025-01-21 17:32:34 -08:00
parent edf56884a7
commit 684de1e6fd

View file

@ -29,6 +29,21 @@ def pytest_addoption(parser):
)
TEXT_MODEL = "meta-llama/Llama-3.1-8B-Instruct"
INFERENCE_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"
if config.getoption("--output"):
config.pluginmanager.register(Report())
def pytest_addoption(parser):
parser.addoption(
"--output",
action="store_false",
help="Knob to determine if we should generate report, e.g. --output=True",
)
TEXT_MODEL = "meta-llama/Llama-3.1-8B-Instruct"
INFERENCE_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"