mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-20 06:05:02 +08:00
[Bug] fix import and unit test (#25558)
Signed-off-by: Jonas M. Kübler <44084297+jmkuebler@users.noreply.github.com> Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
parent
d5c0f43b86
commit
f98d4d38c0
@ -11,7 +11,7 @@ from vllm.v1.attention.backends.utils import (UBatchSlice,
|
|||||||
slice_query_start_locs,
|
slice_query_start_locs,
|
||||||
split_attn_metadata,
|
split_attn_metadata,
|
||||||
split_decodes_and_prefills)
|
split_decodes_and_prefills)
|
||||||
from vllm.v1.worker.ubatch_utils import create_ubatch_slices
|
from vllm.v1.worker.ubatch_splitting import create_ubatch_slices
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user