ColossalAI/colossalai/inference/kv_cache
Yuanheng Zhao f9afe0addd
[hotfix] Fix KV Heads Number Assignment in KVCacheManager (#5695)
- Fix key value number assignment in KVCacheManager, as well as method of accessing
2024-05-07 23:13:14 +08:00
..
__init__.py [Inference] Add CacheBlock and KV-Cache Manager (#5156) 2024-01-11 13:39:29 +00:00
block_cache.py [doc] updated inference readme (#5343) 2024-02-02 14:31:10 +08:00
kvcache_manager.py [hotfix] Fix KV Heads Number Assignment in KVCacheManager (#5695) 2024-05-07 23:13:14 +08:00