ColossalAI/colossalai/legacy/inference/pipeline/__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

4 lines
83 B
Python

from .microbatch_manager import MicroBatchManager
__all__ = ["MicroBatchManager"]