mirror of https://github.com/hpcaitech/ColossalAI
![]() * add kv cache memory manager * add stateinfo during inference * format * format * rename file * add kv cache test * revise on BatchInferState * file dir change |
||
---|---|---|
.. | ||
_utils.py | ||
test_kvcache_manager.py | ||
test_llama_infer.py |