diff --git a/tests/v1/sample/test_sampling_params_e2e.py b/tests/v1/sample/test_sampling_params_e2e.py index 915b9957031d..31a1eb27a8b3 100644 --- a/tests/v1/sample/test_sampling_params_e2e.py +++ b/tests/v1/sample/test_sampling_params_e2e.py @@ -11,7 +11,7 @@ PROMPT = "Hello my name is Robert and I" @pytest.fixture(scope="module") def llm() -> LLM: - return LLM(MODEL, enforce_eager=True) + return LLM(MODEL, enforce_eager=True, dtype="half") def test_n_gt_1(llm):