refactor: move generation.py to llama3

This commit is contained in:
Ashwin Bharambe 2025-03-03 13:38:06 -08:00
parent 02066591b8
commit 816fdf289a
2 changed files with 3 additions and 3 deletions

View file

@ -46,7 +46,6 @@ def pytest_generate_tests(metafunc):
if ("Vision" in cls_name and "Vision" in model) or ("Vision" not in cls_name and "Vision" not in model):
params.append(pytest.param(model, id=model))
print(f"params: {params}")
if not params:
model = metafunc.config.getoption("--inference-model")
params = [pytest.param(model, id=model)]