ColossalAI/colossalai/inference/kv_cache/__init__.py
Runyu Lu 18d67d0e8e
[Feat]Inference RPC Server Support (#5705)
* rpc support source
* kv cache logical/physical disaggregation
* sampler refactor
* colossalai launch built in
* Unitest
* Rpyc support

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2024-05-14 10:00:55 +08:00

5 lines
164 B
Python

from .block_cache import CacheBlock
from .kvcache_manager import KVCacheManager, RPCKVCacheManager
__all__ = ["CacheBlock", "KVCacheManager", "RPCKVCacheManager"]