mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-21 02:15:39 +08:00
use FP1 instead
Signed-off-by: Fanli Lin <fanli.lin@intel.com>
This commit is contained in:
parent
acb1bfa601
commit
0700fdddc7
@ -11,7 +11,7 @@ PROMPT = "Hello my name is Robert and I"
|
|||||||
|
|
||||||
@pytest.fixture(scope="module")
|
@pytest.fixture(scope="module")
|
||||||
def llm() -> LLM:
|
def llm() -> LLM:
|
||||||
return LLM(MODEL, enforce_eager=True)
|
return LLM(MODEL, enforce_eager=True, dtype="half")
|
||||||
|
|
||||||
|
|
||||||
def test_n_gt_1(llm):
|
def test_n_gt_1(llm):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user