Skip to content

Commit

Permalink
fix run_with_both_engines
Browse files Browse the repository at this point in the history
Signed-off-by: Varun Sundar Rabindranath <varun@neuralmagic.com>
  • Loading branch information
Varun Sundar Rabindranath committed Dec 31, 2024
1 parent d016ce5 commit 3200ed4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/lora/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -276,7 +276,7 @@ def llama_2_7b_model_extra_embeddings(llama_2_7b_engine_extra_embeddings):
model_runner.model)


@pytest.fixture(params=[True])
@pytest.fixture(params=[False, True])
def run_with_both_engines_lora(request):
# Automatically runs tests twice, once with V1 and once without
use_v1 = request.param
Expand Down

0 comments on commit 3200ed4

Please sign in to comment.