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

4 lines
79 B
Python

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