mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-28 13:05:26 +00:00
b6696beb04
* refactor pipeline into new CaiInferEngine * updata llama modeling forward * merge tp with pp * update docstring * optimize test workflow and example * fix typo * add assert and todo
4 lines
83 B
Python
4 lines
83 B
Python
from .microbatch_manager import MicroBatchManager
|
|
|
|
__all__ = ["MicroBatchManager"]
|