[CI] Fix UT (#5106)
Fix broken ut introduced by #5053
- vLLM version: v0.12.0
- vLLM main:
ad32e3e19c
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -490,7 +490,7 @@ class TestNPUPlatform(TestBase):
|
||||
dtype="float16",
|
||||
kv_cache_dtype="float16",
|
||||
block_size=64,
|
||||
#use_sfa=False,
|
||||
use_sparse=False,
|
||||
use_mla=True,
|
||||
)
|
||||
self.assertEqual(result,
|
||||
@@ -503,7 +503,7 @@ class TestNPUPlatform(TestBase):
|
||||
dtype="float16",
|
||||
kv_cache_dtype="float16",
|
||||
block_size=64,
|
||||
#use_sfa=False,
|
||||
use_sparse=False,
|
||||
use_mla=False,
|
||||
)
|
||||
self.assertEqual(
|
||||
|
||||
Reference in New Issue
Block a user