diff --git a/tests/v1/core/test_specialized_manager.py b/tests/v1/core/test_specialized_manager.py index 540720cb9b2f..de06da0fc11c 100644 --- a/tests/v1/core/test_specialized_manager.py +++ b/tests/v1/core/test_specialized_manager.py @@ -4,7 +4,7 @@ from vllm.v1.core.block_pool import BlockPool from vllm.v1.core.kv_cache_utils import BlockHashType, KVCacheBlock -from vllm.v1.core.specialized_manager import SlidingWindowManager +from vllm.v1.core.single_type_kv_cache_manager import SlidingWindowManager from vllm.v1.kv_cache_interface import SlidingWindowSpec diff --git a/vllm/v1/core/kv_cache_manager.py b/vllm/v1/core/kv_cache_manager.py index c4ed127ece60..ad8468a89dc5 100644 --- a/vllm/v1/core/kv_cache_manager.py +++ b/vllm/v1/core/kv_cache_manager.py @@ -10,7 +10,8 @@ from vllm.v1.core.block_pool import BlockPool from vllm.v1.core.kv_cache_utils import (BlockHashType, KVCacheBlock, hash_request_tokens) -from vllm.v1.core.specialized_manager import get_manager_for_kv_cache_spec +from vllm.v1.core.single_type_kv_cache_manager import ( + get_manager_for_kv_cache_spec) from vllm.v1.kv_cache_interface import KVCacheConfig from vllm.v1.metrics.stats import PrefixCacheStats from vllm.v1.request import Request, RequestStatus diff --git a/vllm/v1/core/specialized_manager.py b/vllm/v1/core/single_type_kv_cache_manager.py similarity index 100% rename from vllm/v1/core/specialized_manager.py rename to vllm/v1/core/single_type_kv_cache_manager.py