ColossalAI/colossalai/pipeline/middleware/adaptor/__init__.py

3 lines
78 B
Python

from .fx import get_topology as get_fx_topology
__all__ = ['get_fx_topology']