ColossalAI/colossalai/legacy/inference/pipeline/__init__.py

4 lines
83 B
Python

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