use FP1 instead

Signed-off-by: Fanli Lin <fanli.lin@intel.com>
This commit is contained in:
Fanli Lin 2025-10-17 01:40:45 -07:00
parent acb1bfa601
commit 0700fdddc7

View File

@ -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):