ColossalAI/colossalai/inference/kvcache_manager/__init__.py
Xu Kai c6295c3381
[Refactor] remove useless inference code (#5022)
* remove useless code

* fix quant model

* fix test import bug

* mv original inference legacy

* fix chatglm2
2023-11-10 14:47:06 +08:00

3 lines
90 B
Python

from .batch_infer_state import BatchInferState
from .kvcache_manager import MemoryManager